Revert "Allow the TestController to be used outside of our module."

This reverts commit e1010ba0
This commit is contained in:
Todd Knight committed 2020-05-18 16:26:25 -07:00
1 parent e1010ba055
commit ff705bf3b2
2 files changed
+4 -112

No files matched your search

+4 -11
View File
@@ -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))
}