def mock_put_object(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs/abcdefg/status"
assert body["status"] == "CANCELLED"
_was_called = True
return MockResponse({"org_key":"Z100", "name":"FoobieBletch",
"id":"abcdefg", "status":"CANCELLED"})
def mock_delete_object(url):
nonlocal _was_called
if _was_called:
pytest.fail("delete should not be called twice!")
assert url == "/livequery/v1/orgs/Z100/runs/abcdefg"
_was_called = True
return MockResponse(None)
def mock_post_object(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs"
assert body["sql"] == "select * from whatever;"
_was_called = True
return MockResponse({
"org_key": "Z100",
"name": "FoobieBletch",
"id": "abcdefg"
})
def mock_post_object(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs/_search"
assert body["query"] == "xyzzy"
_was_called = True
run1 = {"org_key": "Z100", "name": "FoobieBletch", "id": "abcdefg"}
run2 = {"org_key": "Z100", "name": "Aoxomoxoa", "id": "cdefghi"}
run3 = {"org_key": "Z100", "name": "Read_Me", "id": "efghijk"}
return MockResponse({
"org_key": "Z100",
"num_found": 3,
"results": [run1, run2, run3]
})
def mock_post_object(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs/abcdefg/results/device_summaries/_search"
assert body["query"] == "foo"
t = body["criteria"]
assert t["device_name"] == ["AxCx", "A7X"]
t = body["sort"][0]
assert t["field"] == "device_name"
assert t["order"] == "ASC"
_was_called = True
metrics = [{"key":"aaa", "value":0.0}, {"key":"bbb", "value":0.0}]
res1 = {"id":"ghijklm", "total_results":2, "device_id":314159, "metrics":metrics}
res2 = {"id":"mnopqrs", "total_results":3, "device_id":271828, "metrics":metrics}
return MockResponse({"org_key":"Z100", "num_found":2, "results":[res1, res2]})
def mock_post_object(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs"
assert body["sql"] == "select * from whatever;"
assert body["name"] == "AmyWasHere"
assert body["notify_on_finish"] == True
df = body["device_filter"]
assert df["device_ids"] == [1, 2, 3]
assert df["device_types"] == ["Alpha", "Bravo", "Charlie"]
assert df["policy_ids"] == [16, 27, 38]
_was_called = True
return MockResponse({
"org_key": "Z100",
"name": "FoobieBletch",
"id": "abcdefg"
})
def mock_post_object(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs/abcdefg/results/device_summaries/_facet"
assert body["query"] == "xyzzy"
t = body["criteria"]
assert t["device_name"] == ["AxCx", "A7X"]
t = body["terms"]
assert t["fields"] == ["alpha", "bravo", "charlie"]
_was_called = True
v1 = {"total":1, "id":"alpha1", "name":"alpha1"}
v2 = {"total":2, "id":"alpha2", "name":"alpha2"}
term1 = {"field":"alpha", "values":[v1, v2]}
v1 = {"total":1, "id":"bravo1", "name":"bravo1"}
v2 = {"total":2, "id":"bravo2", "name":"bravo2"}
term2 = {"field":"bravo", "values":[v1, v2]}
v1 = {"total":1, "id":"charlie1", "name":"charlie1"}
v2 = {"total":2, "id":"charlie2", "name":"charlie2"}
term3 = {"field":"charlie", "values":[v1, v2]}
return MockResponse({"terms":[term1, term2, term3]})
def mock_put_object(url, body, **kwargs):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs/abcdefg/status"
assert body["status"] == "CANCELLED"
_was_called = True
return MockResponse({"error_message":"The query is not presently running."}, 409)
def mock_delete_object(url):
nonlocal _was_called
assert url == "/livequery/v1/orgs/Z100/runs/abcdefg"
_was_called = True
return MockResponse(None, 403)