Skip to content
Snippets Groups Projects
Commit b56c609e authored by Nathan Hertz's avatar Nathan Hertz
Browse files

Fix moved pause/unpause tests

parent 3ee21dfa
No related branches found
No related tags found
1 merge request!551Fix moved pause/unpause tests
Pipeline #3139 passed
import warnings
from sqlalchemy import exc as sa_exc
from unittest.mock import MagicMock, patch
import pytest
from sqlalchemy import exc as sa_exc
from workspaces.capability.schema import Capability
from workspaces.capability.services.capability_info import CapabilityInfo
pytest_plugins = ["testing.utils.conftest"]
......@@ -63,7 +65,6 @@ def test_can_delete_capability_request(mock_capability_info: CapabilityInfo):
assert mock_capability_info.can_delete_capability_request(request.id) is False
@pytest.mark.skip("This is currently broken")
def test_pause_capability(mock_capability_info: CapabilityInfo):
# Sub lookup_capability with mock
orig_lookup = mock_capability_info.lookup_capability
......@@ -76,25 +77,21 @@ def test_pause_capability(mock_capability_info: CapabilityInfo):
mock_capability_info.save_entity.assert_called_with(fake_capability)
# Put original method back in place
mock_capability_info.capability_info.lookup_capability = orig_lookup
mock_capability_info.lookup_capability = orig_lookup
@pytest.mark.skip("This is currently broken")
def test_unpause_capability(mock_capability_info: CapabilityInfo):
# Sub lookup_capability with mock
orig_lookup = mock_capability_info.lookup_capability
fake_capability = Capability(name="test", steps="test", max_jobs=2, paused=True)
mock_capability_info.lookup_capability = MagicMock(return_value=fake_capability)
orig_startup = mock_capability_info.execution_manager.queue_manager.startup
mock_capability_info.execution_manager.queue_manager.startup = MagicMock()
mock_capability_info.unpause_capability("test")
mock_capability_info.lookup_capability.assert_called_with("test")
assert fake_capability.paused is False
mock_capability_info.save_entity.assert_called_with(fake_capability)
mock_capability_info.execution_manager.queue_manager.startup.assert_called()
with patch("workspaces.capability.services.capability_info.transaction.commit") as mock_t_commit:
mock_capability_info.unpause_capability("test")
mock_capability_info.lookup_capability.assert_called_with("test")
assert fake_capability.paused is False
mock_capability_info.save_entity.assert_called_with(fake_capability)
mock_t_commit.assert_called()
# Put original methods back in place
mock_capability_info.lookup_capability = orig_lookup
mock_capability_info.execution_manager.queue_manager.startup = orig_startup
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment