WiP

cloudbase · Sep 5, 2024 · 3e49932 · 3e49932
1 parent ec7f2d9
commit 3e49932
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/coriolis/tests/conductor/rpc/test_server.py b/coriolis/tests/conductor/rpc/test_server.py
@@ -3770,8 +3770,6 @@ def test_delete_migration(
         )
         mock_delete_migration.assert_not_called()
 
-    @mock.patch.object(server.ConductorServerEndpoint,
-                       '_check_delete_reservation_for_transfer')
     @mock.patch.object(server.ConductorServerEndpoint,
                        '_cancel_tasks_execution')
     @mock.patch.object(lockutils, 'lock')
@@ -3783,7 +3781,6 @@ def test_cancel_migration(
         mock_get_migration,
         mock_lock,
         mock_cancel_tasks_execution,
-        mock_check_delete_reservation_for_transfer,
         config,
         raises_exception
     ):
@@ -3822,8 +3819,6 @@ def test_cancel_migration(
                 execution,
                 force=force
             )
-            mock_check_delete_reservation_for_transfer.assert_called_once_with(
-                migration)
 
         mock_get_migration.assert_called_once_with(
             mock.sentinel.context,
@@ -5018,6 +5013,9 @@ def test_cancel_execution_for_osmorphing_debugging(
 
     @mock.patch.object(server.ConductorServerEndpoint,
                        "_advance_execution_state")
+    @mock.patch.object(server.ConductorServerEndpoint,
+                       "_check_delete_reservation_for_transfer")
+    @mock.patch.object(db_api, "get_action")
     @mock.patch.object(db_api, "get_tasks_execution")
     @mock.patch.object(db_api, "set_task_status")
     @mock.patch.object(db_api, "get_task")
@@ -5028,6 +5026,8 @@ def test_confirm_task_cancellation(
         mock_get_task,
         mock_set_task_status,
         mock_get_tasks_execution,
+        mock_get_action,
+        mock_check_delete_reservation,
         mock_advance_execution_state,
         task_status,
         expected_final_status,