def test_execute(self, mock_hook): max_results = '100' selected_fields = 'DATE' operator = BigQueryGetDataOperator( task_id=TASK_ID, dataset_id=TEST_DATASET, table_id=TEST_TABLE_ID, max_results=max_results, selected_fields=selected_fields, ) operator.execute(None) mock_hook.return_value \ .get_tabledata \ .assert_called_once_with( dataset_id=TEST_DATASET, table_id=TEST_TABLE_ID, max_results=max_results, selected_fields=selected_fields, )
destination_dataset_table="{}.save_query_result".format(DATASET_NAME), query_params=[{ "name": "to_address", "parameterType": { "type": "STRING" }, "parameterValue": { "value": WALLET_ADDRESS }, }], ) get_data = BigQueryGetDataOperator( task_id="get_data", dataset_id=DATASET_NAME, table_id="save_query_result", max_results="10", selected_fields="value,to_address", ) get_data_result = BashOperator( task_id="get_data_result", bash_command="echo \"{{ task_instance.xcom_pull('get-data') }}\"") create_external_table = BigQueryCreateExternalTableOperator( task_id="create_external_table", bucket=DATA_SAMPLE_GCS_BUCKET_NAME, source_objects=[DATA_SAMPLE_GCS_OBJECT_NAME], destination_project_dataset_table="{}.external_table".format( DATASET_NAME), skip_leading_rows=1,