def test_flush_cursor(self):
bq_cursor = hook.BigQueryCursor("test", "test")
bq_cursor.page_token = '456dcea9-fcbf-4f02-b570-83f5297c685e'
bq_cursor.job_id = 'c0a79ae4-0e72-4593-a0d0-7dbbf726f193'
bq_cursor.all_pages_loaded = True
bq_cursor.buffer = [('a', 100, 200), ('b', 200, 300)]
bq_cursor.flush_results()
self.assertIsNone(bq_cursor.page_token)
self.assertIsNone(bq_cursor.job_id)
self.assertFalse(bq_cursor.all_pages_loaded)
self.assertListEqual(bq_cursor.buffer, [])
def test_execute_with_parameters(self, mocked_rwc):
hook.BigQueryCursor("test", "test").execute(
"SELECT %(foo)s", {"foo": "bar"})
assert mocked_rwc.call_count == 1
def test_execute_with_parameters(self, mocked_rwc):
hook.BigQueryCursor("test", "test").execute("SELECT %(foo)s",
{"foo": "bar"})
mocked_rwc.assert_called_once()
def test_flush_cursor_in_execute(self, _, mocked_fr):
hook.BigQueryCursor("test", "test").execute(
"SELECT %(foo)s", {"foo": "bar"})
assert mocked_fr.call_count == 1
