diff --git a/pool/pool_test.go b/pool/pool_test.go index 2b6b3503..c549c253 100644 --- a/pool/pool_test.go +++ b/pool/pool_test.go @@ -152,6 +152,7 @@ func TestOneNode(t *testing.T) { clientBuilder := func(opts ...client.Option) (client.Client, error) { mockClient := NewMockClient(ctrl) mockClient.EXPECT().CreateSession(gomock.Any(), gomock.Any()).Return(tok, nil) + mockClient.EXPECT().EndpointInfo(gomock.Any(), gomock.Any()).Return(&client.EndpointInfo{}, nil).AnyTimes() return mockClient, nil } @@ -189,6 +190,7 @@ func TestTwoNodes(t *testing.T) { tokens = append(tokens, tok) return tok, err }) + mockClient.EXPECT().EndpointInfo(gomock.Any(), gomock.Any()).Return(&client.EndpointInfo{}, nil).AnyTimes() return mockClient, nil }