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'] }}\"", )