Просмотр исходного кода

Update unit tests

Updates some conductor server unit tests
Daniel Vincze 1 год назад
Родитель
Сommit
59e5eda89e
1 измененных файлов с 2 добавлено и 5 удалено
  1. 2 5
      coriolis/tests/conductor/rpc/test_server.py

+ 2 - 5
coriolis/tests/conductor/rpc/test_server.py

@@ -2362,8 +2362,6 @@ class ConductorServerEndpointTestCase(test_base.CoriolisBaseTestCase):
             mock.sentinel.replica_id,
             include_task_info=True,
         )
-        mock_check_reservation_for_replica.assert_called_once_with(
-            mock_get_replica.return_value)
         mock_check_replica_running_executions.assert_called_once_with(
             mock.sentinel.context,
             mock_get_replica.return_value
@@ -2409,6 +2407,8 @@ class ConductorServerEndpointTestCase(test_base.CoriolisBaseTestCase):
             mock.sentinel.context,
             mock_migration.return_value
         )
+        mock_check_reservation_for_replica.assert_called_once_with(
+            mock_get_replica.return_value)
 
         self.assertEqual(
             mock_tasks_execution.return_value.status,
@@ -5649,6 +5649,3 @@ class ConductorServerEndpointTestCase(test_base.CoriolisBaseTestCase):
             mock.sentinel.task_id,
             mock.sentinel.exception_details,
         )
-        mock_check_delete_reservation_for_transfer.assert_called_once_with(
-            mock_get_action.return_value,
-        )