Fix tests

Signed-off-by: Yves Bastide <yves@botify.com>

Author: ybastide

tests/test_swf/models/test_workflow.py

@@ -187,13 +187,13 @@ def test_workflow_type_changes_with_identical_workflow_type(self):
             self.assertLength(diffs, 0)
 
     def test_save_already_existing_type(self):
-        with patch.object(self.wt.connection, 'register_workflow_type') as mock:
+        with patch.object(self.wt, 'custom_register_workflow_type') as mock:
             with self.assertRaises(AlreadyExistsError):
                 mock.side_effect = SWFTypeAlreadyExistsError(400, "mocked exception")
                 self.wt.save()
 
     def test_save_with_response_error(self):
-        with patch.object(self.wt.connection, 'register_workflow_type') as mock:
+        with patch.object(self.wt, 'custom_register_workflow_type') as mock:
             with self.assertRaises(DoesNotExistError):
                 mock.side_effect = SWFResponseError(
                     400,

tests/test_swf/querysets/test_workflow.py

@@ -14,7 +14,7 @@
                                    WorkflowExecutionQuerySet
 from swf.utils import datetime_timestamp, past_day
 
-from ..mocks.workflow import mock_describe_workflow_type,\
+from tests.test_swf.mocks.workflow import mock_describe_workflow_type,\
                              mock_list_workflow_types,\
                              mock_list_open_workflow_executions,\
                              mock_list_closed_workflow_executions
@@ -110,7 +110,7 @@ def test_get_or_create_non_existent_workflow_type(self):
         with patch.object(Layer1, 'describe_workflow_type') as mock:
             mock.side_effect = DoesNotExistError("Mocked exception")
 
-            with patch.object(Layer1, 'register_workflow_type', mock_describe_workflow_type):
+            with patch.object(WorkflowType, 'custom_register_workflow_type', mock_describe_workflow_type):
                 workflow_type = self.wtq.get_or_create("TestDomain", "testversion")
 
                 self.assertIsInstance(workflow_type, WorkflowType)
@@ -145,9 +145,8 @@ def test_filter_with_registered_status(self):
                 self.assertEqual(wt.status, REGISTERED)
 
     def test_create_workflow_type(self):
-        with patch.object(Layer1, 'register_workflow_type'):
+        with patch.object(WorkflowType, 'custom_register_workflow_type'):
             new_wt = self.wtq.create(
-                self.domain,
                 "TestWorkflowType",
                 "0.test",
             )