diff --git a/octavia/tests/unit/controller/worker/v1/test_controller_worker.py b/octavia/tests/unit/controller/worker/v1/test_controller_worker.py index db516b0d9b..75547388fe 100644 --- a/octavia/tests/unit/controller/worker/v1/test_controller_worker.py +++ b/octavia/tests/unit/controller/worker/v1/test_controller_worker.py @@ -1413,7 +1413,7 @@ class TestControllerWorker(base.TestCase): mock_update.assert_called_with(_db_session, '123', provisioning_status=constants.ACTIVE) - mock_perform.reset + mock_perform.reset_mock() _load_balancer_mock.amphorae = [ _amphora_mock, _amphora_mock2, _amphora_mock3] _amphora_mock2.role = constants.ROLE_BACKUP @@ -1425,7 +1425,7 @@ class TestControllerWorker(base.TestCase): mock_update.assert_called_with(_db_session, '123', provisioning_status=constants.ACTIVE) - mock_perform.reset + mock_perform.reset_mock() mock_perform.side_effect = OverflowError() self.assertRaises(OverflowError, cw.failover_loadbalancer, 123) mock_update.assert_called_with(_db_session, 123, diff --git a/octavia/tests/unit/controller/worker/v2/test_controller_worker.py b/octavia/tests/unit/controller/worker/v2/test_controller_worker.py index b7a2c961c5..48ebb89e14 100644 --- a/octavia/tests/unit/controller/worker/v2/test_controller_worker.py +++ b/octavia/tests/unit/controller/worker/v2/test_controller_worker.py @@ -1466,7 +1466,7 @@ class TestControllerWorker(base.TestCase): mock_update.assert_called_with(_db_session, '123', provisioning_status=constants.ACTIVE) - mock_perform.reset + mock_perform.reset_mock() _load_balancer_mock.amphorae = [ _amphora_mock, _amphora_mock2, _amphora_mock3] _amphora_mock2.role = constants.ROLE_BACKUP @@ -1478,7 +1478,7 @@ class TestControllerWorker(base.TestCase): mock_update.assert_called_with(_db_session, '123', provisioning_status=constants.ACTIVE) - mock_perform.reset + mock_perform.reset_mock() mock_perform.side_effect = OverflowError() self.assertRaises(OverflowError, cw.failover_loadbalancer, 123) mock_update.assert_called_with(_db_session, 123,