fix: mock zk registry

LaurenceLiZhixin · LaurenceLiZhixin · commit 57ea7a6f11bd · 2021-07-29T23:51:02.000+08:00
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,7 @@
 *.so
 *.dylib
 *.jar
+*.log
 
 # Test binary, build with `go test -c`
 *.test
diff --git a/registry/zookeeper/registry.go b/registry/zookeeper/registry.go
@@ -108,6 +108,7 @@ func newMockZkRegistry(url *common.URL, opts ...gxzookeeper.Option) (*zk.TestClu
 	if err != nil {
 		return nil, nil, err
 	}
+	r.WaitGroup().Add(1)
 	go zookeeper.HandleClientRestart(r)
 	r.InitListeners()
 	return c, r, nil

-Original file line number
+Diff line change
 *.so
 *.dylib
 *.jar
 +*.log
 # Test binary, build with `go test -c`
 *.test
Original file line number	Diff line number	Diff line change
`@@ -108,6 +108,7 @@ func newMockZkRegistry(url common.URL, opts ...gxzookeeper.Option) (zk.TestClu`
`108`	`108`	`if err != nil {`
`109`	`109`	`return nil, nil, err`
`110`	`110`	`}`
	`111`	`+ r.WaitGroup().Add(1)`
`111`	`112`	`go zookeeper.HandleClientRestart(r)`
`112`	`113`	`r.InitListeners()`
`113`	`114`	`return c, r, nil`