Revert "Allow the TestController to be used outside of our module."
This reverts commit e1010ba0
This commit is contained in:
1 parent
e1010ba055
commit
ff705bf3b2
2 files changed
+4
-112
No files matched your search
@@ -19,7 +19,6 @@ type TestController struct {
|
||||
b *base.Server
|
||||
c *Controller
|
||||
t *testing.T
|
||||
addr string // The address the Controller API is listening on
|
||||
client *api.Client
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
@@ -42,10 +41,7 @@ func (tc *TestController) Cancel() {
|
||||
tc.cancel()
|
||||
}
|
||||
|
||||
func (tc *TestController) ApiAddress() string {
|
||||
if tc.addr != "" {
|
||||
return tc.addr
|
||||
}
|
||||
func (tc *TestController) buildClient() {
|
||||
var apiLn *base.ServerListener
|
||||
for _, listener := range tc.b.Listeners {
|
||||
if listener.Config.Purpose[0] == "api" {
|
||||
@@ -56,20 +52,17 @@ func (tc *TestController) ApiAddress() string {
|
||||
if apiLn == nil {
|
||||
tc.t.Fatal("could not find api listener")
|
||||
}
|
||||
|
||||
tcpAddr, ok := apiLn.Mux.Addr().(*net.TCPAddr)
|
||||
if !ok {
|
||||
tc.t.Fatal("could not parse address as a TCP addr")
|
||||
}
|
||||
tc.addr = fmt.Sprintf("http://%s:%d", tcpAddr.IP.String(), tcpAddr.Port)
|
||||
return tc.addr
|
||||
}
|
||||
|
||||
func (tc *TestController) buildClient() {
|
||||
addr := fmt.Sprintf("http://%s:%d", tcpAddr.IP.String(), tcpAddr.Port)
|
||||
client, err := api.NewClient(nil)
|
||||
if err != nil {
|
||||
tc.t.Fatal(fmt.Errorf("error creating client: %w", err))
|
||||
}
|
||||
if err := client.SetAddr(tc.ApiAddress()); err != nil {
|
||||
if err := client.SetAddr(addr); err != nil {
|
||||
tc.t.Fatal(fmt.Errorf("error setting client address: %w", err))
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user