Fix tests

Author: polyaxon-ci

cli/tests/test_client/test_organization_client.py

@@ -331,24 +331,30 @@ def test_skip_runs(self, mock_skip):
     @mock.patch("polyaxon._sdk.api.OrganizationsV1Api.tag_organization_runs")
     def test_tag_runs(self, mock_tag):
         """Test batch tagging runs"""
-        data = {"uuids": [self.uuid1, self.uuid2], "tags": ["production", "validated"]}
+        uuids = [self.uuid1, self.uuid2]
+        tags = ["production", "validated"]
 
         client = OrganizationClient(owner=self.owner)
-        client.tag_runs(data)
+        client.tag_runs(uuids, tags)
 
         assert mock_tag.call_count == 1
-        mock_tag.assert_called_with(self.owner, body=data)
+        call_args = mock_tag.call_args
+        assert call_args[1]["body"].uuids == uuids
+        assert call_args[1]["body"].tags == tags
 
     @mock.patch("polyaxon._sdk.api.OrganizationsV1Api.transfer_organization_runs")
     def test_transfer_runs(self, mock_transfer):
         """Test transferring runs to different project"""
-        data = {"uuids": [self.uuid1, self.uuid2], "project": "destination-project"}
+        uuids = [self.uuid1, self.uuid2]
+        to_project = "destination-project"
 
         client = OrganizationClient(owner=self.owner)
-        client.transfer_runs(data)
+        client.transfer_runs(uuids, to_project)
 
         assert mock_transfer.call_count == 1
-        mock_transfer.assert_called_with(self.owner, body=data)
+        call_args = mock_transfer.call_args
+        assert call_args[1]["body"].uuids == uuids
+        assert call_args[1]["body"].project == to_project
 
     # Version Management Tests
     def test_validate_kind(self):

cli/tests/test_client/test_project_client.py

@@ -258,13 +258,16 @@ def test_bookmark_runs(self, mock_bookmark):
     @mock.patch("polyaxon._sdk.api.RunsV1Api.tag_runs")
     def test_tag_runs(self, mock_tag):
         """Test batch tagging runs"""
-        data = {"uuids": [self.uuid1, self.uuid2], "tags": ["production", "validated"]}
+        uuids = [self.uuid1, self.uuid2]
+        tags = ["production", "validated"]
 
         client = ProjectClient(owner=self.owner, project=self.project)
-        client.tag_runs(data)
+        client.tag_runs(uuids, tags)
 
         assert mock_tag.call_count == 1
-        mock_tag.assert_called_with(self.owner, self.project, body=data)
+        call_args = mock_tag.call_args
+        assert call_args[1]["body"].uuids == uuids
+        assert call_args[1]["body"].tags == tags
 
     # Version Management Tests - Validation
     def test_validate_kind(self):

cli/tests/test_deploy/test_email.py

@@ -17,7 +17,7 @@ def test_email_config(self):
             EmailConfig.from_dict(config_dict)
 
         config_dict = {
-            "emailFrom host": "dsf",
+            "emailFromhost": "dsf",
             "port": "sdf",
             "useTls": True,
             "hostUser": "sdf",