@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Client')
def test_insert_all_succeed(self, mock_client):
    rows = [{'json': {'a_key': 'a_value_0'}}]
    self.hook.insert_all(project_id=PROJECT_ID, dataset_id=DATASET_ID, table_id=TABLE_ID, rows=rows, ignore_unknown_values=True, skip_invalid_rows=True)
    mock_client.return_value.get_table.assert_called_once_with(TABLE_REFERENCE)
    mock_client.return_value.insert_rows.assert_called_once_with(table=mock_client.return_value.get_table.return_value, rows=rows, ignore_unknown_values=True, skip_invalid_rows=True)
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Client')
def test_insert_all_succeed(self, mock_client):
    rows = [{'json': {'a_key': 'a_value_0'}}]
    self.hook.insert_all(project_id=PROJECT_ID, dataset_id=DATASET_ID, table_id=TABLE_ID, rows=rows, ignore_unknown_values=True, skip_invalid_rows=True)
    mock_client.return_value.get_table.assert_called_once_with(TABLE_REFERENCE)
    mock_client.return_value.insert_rows.assert_called_once_with(table=mock_client.return_value.get_table.return_value, rows=rows, ignore_unknown_values=True, skip_invalid_rows=True)