@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Table')
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.SchemaField')
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Client')
def test_list_rows(self, mock_client, mock_schema, mock_table):
    self.hook.list_rows(dataset_id=DATASET_ID, table_id=TABLE_ID, max_results=10, selected_fields=['field_1', 'field_2'], page_token='page123', start_index=5, location=LOCATION)
    mock_table.from_api_repr.assert_called_once_with({'tableReference': TABLE_REFERENCE_REPR})
    mock_schema.has_calls([mock.call(x, '') for x in ['field_1', 'field_2']])
    mock_client.return_value.list_rows.assert_called_once_with(table=mock_table.from_api_repr.return_value, max_results=10, selected_fields=mock.ANY, page_token='page123', start_index=5)
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Table')
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.SchemaField')
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.Client')
def test_list_rows(self, mock_client, mock_schema, mock_table):
    self.hook.list_rows(dataset_id=DATASET_ID, table_id=TABLE_ID, max_results=10, selected_fields=['field_1', 'field_2'], page_token='page123', start_index=5, location=LOCATION)
    mock_table.from_api_repr.assert_called_once_with({'tableReference': TABLE_REFERENCE_REPR})
    mock_schema.has_calls([mock.call(x, ) for x in ['field_1', 'field_2']])
    mock_client.return_value.list_rows.assert_called_once_with(table=mock_table.from_api_repr.return_value, max_results=10, selected_fields=mock.ANY, page_token='page123', start_index=5)