def test_switch_to_provisioning( workspace: Workspace, monkeypatch: _pytest.monkeypatch.MonkeyPatch) -> None: workspace.gr = MagicMock(name='dummy-guest-request') workspace.new_pool = MagicMock(name='dummy-pool') patch(monkeypatch, workspace, 'update_guest_state') assert workspace.switch_to_provisioning() is workspace cast(MagicMock, workspace.update_guest_state).assert_called_once_with( tft.artemis.guest.GuestState.PROVISIONING, current_state=tft.artemis.guest.GuestState.ROUTING, set_values={'poolname': workspace.new_pool.poolname}, pool=workspace.new_pool.poolname)
def test_query_policies(workspace: Workspace, session: sqlalchemy.orm.session.Session, monkeypatch: _pytest.monkeypatch.MonkeyPatch) -> None: workspace.gr = MagicMock(name='workspace.gr<mock>') workspace.pools = [] mock_ruling = tft.artemis.routing_policies.PolicyRuling() patch(monkeypatch, workspace, 'run_hook').return_value = mock_ruling assert workspace.query_policies() is workspace assert workspace.result is None assert workspace.ruling is mock_ruling cast(MagicMock, workspace.run_hook).assert_called_once_with( 'ROUTE', session=session, guest_request=workspace.gr, pools=workspace.pools)