add more tests

flamingstar · Oct 10, 2020 · 3f8b080 · 3f8b080
1 parent adc2758
commit 3f8b080
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 46 deletions.
diff --git a/core/discov/internal/listener.go b/core/discov/internal/listener.go
@@ -1,4 +1,3 @@
-//go:generate mockgen -package internal -destination listener_mock.go -source listener.go Listener
 package internal
 
 type Listener interface {

diff --git a/core/discov/internal/listener_mock.go b/core/discov/internal/listener_mock.go
diff --git a/core/discov/publisher_test.go b/core/discov/publisher_test.go
@@ -111,6 +111,10 @@ func TestPublisher_keepAliveAsyncQuit(t *testing.T) {
 	defer ctrl.Finish()
 	const id clientv3.LeaseID = 1
 	cli := internal.NewMockEtcdClient(ctrl)
+	cli.EXPECT().ActiveConnection()
+	cli.EXPECT().Close()
+	defer cli.Close()
+	cli.ActiveConnection()
 	restore := setMockClient(cli)
 	defer restore()
 	cli.EXPECT().Ctx().AnyTimes()