def test_collect_query_info(self, requests_url_mock, requests_get_mock,
mkdir_mock, open_mock,
req_json_mock, json_dumps_mock):
query_id = "1234_abcd"
query_info_file_name = path.join(TMP_PRESTO_DEBUG,
"query_info_" + query_id + ".json")
file_obj = open_mock.return_value.__enter__.return_value
requests_get_mock.return_value.json.return_value = req_json_mock
requests_get_mock.return_value.status_code = requests.codes.ok
env.host = "myhost"
env.roledefs["coordinator"] = ["myhost"]
collect.query_info(query_id)
mkdir_mock.assert_called_with(TMP_PRESTO_DEBUG)
open_mock.assert_called_with(query_info_file_name, "w")
json_dumps_mock.assert_called_with(req_json_mock, indent=4)
file_obj.write.assert_called_with(json_dumps_mock.return_value)
def test_collect_query_info(self, requests_url_mock, requests_get_mock,
mkdir_mock, open_mock,
req_json_mock, json_dumps_mock):
query_id = "1234_abcd"
query_info_file_name = path.join(TMP_PRESTO_DEBUG,
"query_info_" + query_id + ".json")
file_obj = open_mock.return_value.__enter__.return_value
requests_get_mock.return_value.json.return_value = req_json_mock
requests_get_mock.return_value.status_code = requests.codes.ok
env.host = "myhost"
env.roledefs["coordinator"] = ["myhost"]
collect.query_info(query_id)
mkdir_mock.assert_called_with(TMP_PRESTO_DEBUG)
open_mock.assert_called_with(query_info_file_name, "w")
json_dumps_mock.assert_called_with(req_json_mock, indent=4)
file_obj.write.assert_called_with(json_dumps_mock.return_value)
def test_query_info_not_run_on_workers(self, req_get_mock):
env.host = ["worker1"]
env.roledefs["worker"] = ["worker1"]
collect.query_info("any_query_id")
assert not req_get_mock.called
def test_query_info_not_run_on_workers(self, req_get_mock):
env.host = ["worker1"]
env.roledefs["worker"] = ["worker1"]
collect.query_info("any_query_id")
assert not req_get_mock.called