def test_audit_remediation(monkeypatch):
_was_called = False
def _run_query(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs"
assert body == {"sql": "select * from whatever;", "device_filter": {}}
_was_called = True
return StubResponse({"org_key": "Z100", "name": "FoobieBletch", "id": "abcdefg"})
api = CBCloudAPI(url="https://example.com", token="ABCD/1234", org_key="Z100", ssl_verify=True)
patch_cbapi(monkeypatch, api, POST=_run_query)
query = api.audit_remediation("select * from whatever;")
assert isinstance(query, RunQuery)
run = query.submit()
assert _was_called
assert run.org_key == "Z100"
assert run.name == "FoobieBletch"
assert run.id == "abcdefg"
def test_audit_remediation_with_everything(monkeypatch):
_was_called = False
def _run_query(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs"
assert body == {"sql": "select * from whatever;", "name": "AmyWasHere", "notify_on_finish": True,
"device_filter": {"device_id": [1, 2, 3], "os": ["Alpha", "Bravo", "Charlie"],
"policy_id": [16]}}
_was_called = True
return StubResponse({"org_key": "Z100", "name": "FoobieBletch", "id": "abcdefg"})
api = CBCloudAPI(url="https://example.com", token="ABCD/1234", org_key="Z100", ssl_verify=True)
patch_cbapi(monkeypatch, api, POST=_run_query)
query = api.audit_remediation("select * from whatever;").device_ids([1, 2, 3]).device_types(["Alpha", "Bravo", "Charlie"]) \
.policy_id(16).name("AmyWasHere").notify_on_finish()
assert isinstance(query, RunQuery)
run = query.submit()
assert _was_called
assert run.org_key == "Z100"
assert run.name == "FoobieBletch"
assert run.id == "abcdefg"
def test_audit_remediation_device_types_broken():
"""Test submitting a bad device type to a query."""
api = CBCloudAPI(url="https://example.com", token="ABCD/1234", org_key="Z100", ssl_verify=True)
query = api.audit_remediation("select * from whatever;")
with pytest.raises(ApiError):
query = query.device_types([420])
def test_audit_remediation_policy_id_broken():
"""Test submitting a bad policy ID to a query."""
api = CBCloudAPI(url="https://example.com", token="ABCD/1234", org_key="Z100", ssl_verify=True)
query = api.audit_remediation("select * from whatever;")
with pytest.raises(ApiError):
query = query.policy_id(["Bogus"])
def test_audit_remediation_device_ids_broken():
api = CBCloudAPI(url="https://example.com", token="ABCD/1234", org_key="Z100", ssl_verify=True)
query = api.audit_remediation("select * from whatever;")
with pytest.raises(ApiError):
query = query.device_ids(["Bogus"])