def test_execute(self, mock_hook): operator = BigQueryCreateEmptyTableOperator(task_id=TASK_ID, dataset_id=TEST_DATASET, project_id=TEST_PROJECT_ID, table_id=TEST_TABLE_ID) operator.execute(None) mock_hook.return_value \ .get_conn() \ .cursor() \ .create_empty_table \ .assert_called_once_with( dataset_id=TEST_DATASET, project_id=TEST_PROJECT_ID, table_id=TEST_TABLE_ID, schema_fields=None, time_partitioning={} )
def test_execute(self, mock_hook): operator = BigQueryCreateEmptyTableOperator(task_id=TASK_ID, dataset_id=TEST_DATASET, project_id=TEST_PROJECT_ID, table_id=TEST_TABLE_ID) operator.execute(None) mock_hook.return_value \ .get_conn() \ .cursor() \ .create_empty_table \ .assert_called_once_with( dataset_id=TEST_DATASET, project_id=TEST_PROJECT_ID, table_id=TEST_TABLE_ID, schema_fields=None, time_partitioning={} )
def execute(self, context): BigQueryDropTableOperator.execute(self, context) BigQueryCreateEmptyTableOperator.execute(self, context)