def test_interrupt_experiment(organizations, default_org_id, default_team_id):
execution_id = str(uuid.uuid4())
experiment_id = str(uuid.uuid4())
with requests_mock.mock() as m:
url = urls.full(urls.base(ENDPOINT),
default_org_id,
default_team_id,
experiment_id,
execution_id=execution_id,
with_safeguards=True)
m.get(url,
status_code=200,
json={
"allowed": False,
"policies": [{
"name": "godzilla says stop"
}]
})
with client_session(ENDPOINT, organizations) as s:
with pytest.raises(InterruptExecution):
is_allowed_to_continue(s, [{
"name": "chaosiq",
"execution_id": execution_id,
"experiment_id": experiment_id
}])
def test_store_safeguards_to_journal(organizations, default_org_id,
default_team_id):
execution_id = str(uuid.uuid4())
experiment_id = str(uuid.uuid4())
safeguards = [{"name": "godzilla says stop"}]
with requests_mock.mock() as m:
url = urls.full(urls.base(ENDPOINT),
default_org_id,
default_team_id,
experiment_id,
execution_id=execution_id,
with_safeguards=True)
m.get(url,
status_code=200,
json={
"allowed": False,
"policies": safeguards
})
extensions = [{
"name": "chaosiq",
"execution_id": execution_id,
"experiment_id": experiment_id
}]
journal = {}
with client_session(ENDPOINT, organizations) as s:
with pytest.raises(InterruptExecution):
is_allowed_to_continue(s, extensions)
set_applied_safeguards_for_execution(extensions, journal)
assert journal["extensions"][0]["safeguards"] == safeguards
def test_no_check_when_execution_is_not_found(organizations, default_org_id,
default_team_id):
execution_id = str(uuid.uuid4())
experiment_id = str(uuid.uuid4())
with requests_mock.mock() as m:
url = urls.full(urls.base(ENDPOINT),
default_org_id,
default_team_id,
experiment_id,
execution_id=execution_id,
with_safeguards=True)
m.get(url, status_code=404)
try:
with client_session(ENDPOINT, organizations) as s:
is_allowed_to_continue(s, [{
"name": "chaosiq",
"execution_id": execution_id,
"experiment_id": experiment_id
}])
except InterruptExecution:
pytest.fail(
"Missing execution identifier in experiment should not "
"lead to execution interruption")
def test_continue_when_experiment_id_not_in_experiment():
assert is_allowed_to_continue(None, []) is None