def test_execute(self, mock_hook):
ignore_if_missing = True
deletion_dataset_table = '{}.{}'.format(TEST_DATASET, TEST_TABLE_ID)
operator = BigQueryDeleteTableOperator(
task_id=TASK_ID,
deletion_dataset_table=deletion_dataset_table,
ignore_if_missing=ignore_if_missing)
operator.execute(None)
mock_hook.return_value \
.run_table_delete \
.assert_called_once_with(
deletion_dataset_table=deletion_dataset_table,
ignore_if_missing=ignore_if_missing
)
task_id="create_view",
dataset_id=LOCATION_DATASET_NAME,
table_id="test_view",
view={
"query": "SELECT * FROM `{}.test_table`".format(DATASET_NAME),
"useLegacySql": False
})
get_empty_dataset_tables = BigQueryGetDatasetTablesOperator(
task_id="get_empty_dataset_tables", dataset_id=DATASET_NAME)
get_dataset_tables = BigQueryGetDatasetTablesOperator(
task_id="get_dataset_tables", dataset_id=DATASET_NAME)
delete_view = BigQueryDeleteTableOperator(
task_id="delete_view",
deletion_dataset_table="{}.test_view".format(DATASET_NAME))
delete_table = BigQueryDeleteTableOperator(
task_id="delete_table",
deletion_dataset_table="{}.test_table".format(DATASET_NAME))
get_dataset = BigQueryGetDatasetOperator(task_id="get-dataset",
dataset_id=DATASET_NAME)
get_dataset_result = BashOperator(
task_id="get_dataset_result",
bash_command=
"echo \"{{ task_instance.xcom_pull('get-dataset')['id'] }}\"",
)