def test_GbqConnector_get_client_w_new_bq(mock_bigquery_client): gbq._test_google_api_imports() pytest.importorskip("google.api_core.client_info") connector = _make_connector() connector.get_client() _, kwargs = mock_bigquery_client.call_args assert kwargs["client_info"].user_agent == "pandas-{}".format(pandas.__version__)
def test_GbqConnector_get_client_w_old_bq(monkeypatch, mock_bigquery_client): gbq._test_google_api_imports() connector = _make_connector() monkeypatch.setattr(gbq, "HAS_CLIENT_INFO", False) connector.get_client() # No client_info argument. mock_bigquery_client.assert_called_with(credentials=mock.ANY, project=mock.ANY)
def test_GbqConnector_get_client_w_new_bq(mock_bigquery_client): gbq._test_google_api_imports() if not FEATURES.bigquery_has_client_info: pytest.skip("google-cloud-bigquery missing client_info feature") pytest.importorskip("google.api_core.client_info") connector = _make_connector() connector.get_client() _, kwargs = mock_bigquery_client.call_args assert kwargs["client_info"].user_agent == "pandas-{}".format( pandas.__version__)
def test_GbqConnector_get_client_w_old_bq(monkeypatch, mock_bigquery_client): gbq._test_google_api_imports() connector = _make_connector() monkeypatch.setattr( type(FEATURES), "bigquery_has_client_info", mock.PropertyMock(return_value=False), ) connector.get_client() # No client_info argument. mock_bigquery_client.assert_called_with(credentials=mock.ANY, project=mock.ANY)
def test_imports(): gbq._test_google_api_imports()