run cancel in defer

management/client/client_test.go

@@ -259,6 +259,7 @@ func TestClient_Sync(t *testing.T) {
 	ch := make(chan *mgmtProto.SyncResponse, 5)
 
 	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
 	go func() {
 		err = client.Sync(ctx, info, func(msg *mgmtProto.SyncResponse) error {
 			ch <- msg
@@ -291,7 +292,6 @@ func TestClient_Sync(t *testing.T) {
 		t.Error("timeout waiting for test to finish")
 	}
 
-	cancel()
 }
 
 func Test_SystemMetaDataFromClient(t *testing.T) {