add more tests

2020-10-10 13:47:55 +08:00
parent adc275872d
commit 3f8b080882
3 changed files with 4 additions and 46 deletions
--- 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()