Fix unittest rpc session leak

Qi Xiao · Qi Xiao · commit 08f4128c6158 · 2019-08-07T12:31:00.000+08:00
diff --git a/rpc/mux/wrapper_test.go b/rpc/mux/wrapper_test.go
@@ -58,6 +58,7 @@ func TestCaller_CallNode(t *testing.T) {
 	log.Debugf("GConf: %#v", conf.GConf)
 	// reset the once
 	route.Once.Reset()
+	_ = GetSessionPoolInstance().Close()
 	route.InitKMS(publicKeyStore)
 
 	addr := conf.GConf.ListenAddr // see ../test/node_standalone/config.yaml
@@ -189,6 +190,7 @@ func TestNewPersistentCaller(t *testing.T) {
 	log.Debugf("GConf: %#v", conf.GConf)
 	// reset the once
 	route.Once.Reset()
+	_ = GetSessionPoolInstance().Close()
 	route.InitKMS(publicKeyStore)
 
 	addr := conf.GConf.ListenAddr // see ../test/node_standalone/config.yaml
@@ -335,6 +337,7 @@ func BenchmarkPersistentCaller_CallKayakLog(b *testing.B) {
 	log.Debugf("GConf: %#v", conf.GConf)
 	// reset the once
 	route.Once.Reset()
+	_ = GetSessionPoolInstance().Close()
 	route.InitKMS(publicKeyStore)
 
 	addr := conf.GConf.ListenAddr
@@ -411,6 +414,7 @@ func BenchmarkPersistentCaller_Call(b *testing.B) {
 	log.Debugf("GConf: %#v", conf.GConf)
 	// reset the once
 	route.Once.Reset()
+	_ = GetSessionPoolInstance().Close()
 	route.InitKMS(publicKeyStore)
 
 	addr := conf.GConf.ListenAddr // see ../test/node_standalone/config.yaml
