def test_should_return_to_initial_state_dev(mock_get_imports,
looker_client: LookerClient):
# Set up starting branch and workspace
looker_client.update_workspace("dev")
manager = LookerBranchManager(looker_client, LOOKER_PROJECT)
assert manager.init_state.workspace == "dev"
with manager():
assert looker_client.get_workspace() == "production"
assert looker_client.get_workspace() == "dev"
def test_manage_prod_with_advanced_deploy(mock_get_imports,
looker_client: LookerClient):
# Set up starting branch and workspace
project = "spectacles-advanced-deploy"
looker_client.update_workspace("production")
commit = looker_client.get_active_branch(project)["ref"]
manager = LookerBranchManager(looker_client, project)
assert manager.init_state.workspace == "production"
assert manager.init_state.commit == commit
with manager():
assert looker_client.get_workspace() == "production"
