def test_async_query(self):
with tests.capture_stdout() as stdout:
main(
self.constants['projectId'],
self.constants['query'],
False,
5,
5)
value = stdout.getvalue().strip().split('\n').pop()
self.assertIsNotNone(
json.loads(value))
def test_async_query_runner(self):
test_project_id = os.environ.get(tests.PROJECT_ID_ENV)
answers = [test_project_id, self.constants["query"], "n", "1", "1"]
with tests.mock_raw_input(answers):
main()