client: do cleanup after UT stop mock server

Signed-off-by: NingBo <[email protected]>
kata-containers · Nov 29, 2018 · e614630 · e614630
1 parent 0ff3006
commit e614630
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/protocols/client/client_test.go b/protocols/client/client_test.go
@@ -34,7 +34,11 @@ const (
 )
 
 func startMockServer(t *testing.T, enableYamux bool) (*grpc.Server, chan error, error) {
-	os.Remove(mockSockAddr)
+	err := os.RemoveAll(mockSockAddr)
+	assert.Nil(t, err, "Remove %s failed", mockSockAddr)
+	if err != nil {
+		assert.FailNow(t, err.Error(), "Failed to start mock server")
+	}
 
 	l, err := net.Listen("unix", mockSockAddr)
 	assert.Nil(t, err, "Listen on %s failed: %s", mockSockAddr, err)
@@ -132,6 +136,7 @@ func TestNewAgentClient(t *testing.T) {
 	// wait mock server to stop
 	mock.Stop()
 	<-waitCh
+	os.Remove(mockSockAddr)
 }
 
 func TestNewAgentClientWithYamux(t *testing.T) {
@@ -153,4 +158,5 @@ func TestNewAgentClientWithYamux(t *testing.T) {
 	// wait mock server to stop
 	mock.Stop()
 	<-waitCh
+	os.Remove(mockSockAddr)
 }