Correct unit tests for authentication and has measurements

racker · Mar 29, 2017 · 827ab85 · 827ab85
1 parent 286d2e9
commit 827ab85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/poller/connection_stream_test.go b/poller/connection_stream_test.go
@@ -70,7 +70,7 @@ func TestConnectionStream_Connect(t *testing.T) {
 
 			conn := NewMockConnection(ctrl)
 			conn.EXPECT().Connect(gomock.Any(), gomock.Any(), gomock.Any())
-			conn.EXPECT().Done().Return(done)
+			conn.EXPECT().Done().AnyTimes().Return(done)
 			conn.EXPECT().GetLogPrefix().AnyTimes().Return("1234")
 			conn.EXPECT().Close().AnyTimes().Do(func() {
 				t.Log("Mock conn is closing")