Stop server after queries

sonic-net · Dec 19, 2024 · dd006d6 · dd006d6
1 parent a3a12e6
commit dd006d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gnmi_server/server_test.go b/gnmi_server/server_test.go
@@ -245,6 +245,7 @@ func createKeepAliveServer(t *testing.T, port int64) *Server {
 func TestPFCWDErrors(t *testing.T) {
 	s := createServer(t, 8081)
 	go runServer(t, s)
+	defer s.ForceStop()
 
 	mock := gomonkey.ApplyFunc(sdc.GetPfcwdMap, func() (map[string]map[string]string, error)  {
 		return nil, fmt.Errorf("Mock error")
@@ -334,7 +335,6 @@ func TestPFCWDErrors(t *testing.T) {
 			c.Close()
 		})
 	}
-	s.s.Stop()
 }