@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.get_client')
def test_poll_job_complete(self, mock_client):
    self.hook.poll_job_complete(job_id=JOB_ID, location=LOCATION, project_id=PROJECT_ID)
    mock_client.assert_called_once_with(location=LOCATION, project_id=PROJECT_ID)
    mock_client.return_value.get_job.assert_called_once_with(job_id=JOB_ID)
    mock_client.return_value.get_job.return_value.done.assert_called_once_with(retry=DEFAULT_RETRY)
@mock.patch('airflow.providers.google.cloud.hooks.bigquery.BigQueryHook.get_client')
def test_poll_job_complete(self, mock_client):
    self.hook.poll_job_complete(job_id=JOB_ID, location=LOCATION, project_id=PROJECT_ID)
    mock_client.assert_called_once_with(location=LOCATION, project_id=PROJECT_ID)
    mock_client.return_value.get_job.assert_called_once_with(job_id=JOB_ID)
    mock_client.return_value.get_job.return_value.done.assert_called_once_with(retry=DEFAULT_RETRY)