Spanner: add client_info support to client.

spanner/google/cloud/spanner_v1/client.py

@@ -95,9 +95,17 @@ class Client(ClientWithProject):
                         client. If not provided, defaults to the Google
                         Application Default Credentials.
 
+    :type client_info: :class:`google.api_core.client_info.ClientInfo`
+    :param client_info:
+        (Optional) The client info used to send a user-agent string along with
+        API requests. If ``None``, then default info will be used. Generally,
+        you only need to set this if you're developing your own library or
+        partner tool.
+
     :type user_agent: str
-    :param user_agent: (Optional) The user agent to be used with API request.
-                       Defaults to :const:`DEFAULT_USER_AGENT`.
+    :param user_agent:
+        (Deprecated) The user agent to be used with API request.
+        Not used.  Defaults to :const:`DEFAULT_USER_AGENT`.
 
     :raises: :class:`ValueError <exceptions.ValueError>` if both ``read_only``
              and ``admin`` are :data:`True`
@@ -110,13 +118,20 @@ class Client(ClientWithProject):
     SCOPE = (SPANNER_ADMIN_SCOPE,)
     """The scopes required for Google Cloud Spanner."""
 
-    def __init__(self, project=None, credentials=None, user_agent=DEFAULT_USER_AGENT):
+    def __init__(
+        self,
+        project=None,
+        credentials=None,
+        client_info=client_info,
+        user_agent=DEFAULT_USER_AGENT,
+    ):
         # NOTE: This API has no use for the _http argument, but sending it
         #       will have no impact since the _http() @property only lazily
         #       creates a working HTTP object.
         super(Client, self).__init__(
             project=project, credentials=credentials, _http=None
         )
+        self._client_info = client_info
         self.user_agent = user_agent
 
     @property
@@ -153,7 +168,7 @@ def instance_admin_api(self):
         """Helper for session-related API calls."""
         if self._instance_admin_api is None:
             self._instance_admin_api = InstanceAdminClient(
-                credentials=self.credentials, client_info=_CLIENT_INFO
+                credentials=self.credentials, client_info=self._client_info
             )
         return self._instance_admin_api
 
@@ -162,7 +177,7 @@ def database_admin_api(self):
         """Helper for session-related API calls."""
         if self._database_admin_api is None:
             self._database_admin_api = DatabaseAdminClient(
-                credentials=self.credentials, client_info=_CLIENT_INFO
+                credentials=self.credentials, client_info=self._client_info
             )
         return self._database_admin_api
 

spanner/google/cloud/spanner_v1/database.py

@@ -19,14 +19,12 @@
 import re
 import threading
 
-from google.api_core.gapic_v1 import client_info
 import google.auth.credentials
 from google.protobuf.struct_pb2 import Struct
 from google.cloud.exceptions import NotFound
 import six
 
 # pylint: disable=ungrouped-imports
-from google.cloud.spanner_v1 import __version__
 from google.cloud.spanner_v1._helpers import _make_value_pb
 from google.cloud.spanner_v1._helpers import _metadata_with_prefix
 from google.cloud.spanner_v1.batch import Batch
@@ -46,7 +44,6 @@
 # pylint: enable=ungrouped-imports
 
 
-_CLIENT_INFO = client_info.ClientInfo(client_library_version=__version__)
 SPANNER_DATA_SCOPE = "https://www.googleapis.com/auth/spanner.data"
 
 
@@ -179,8 +176,9 @@ def spanner_api(self):
             credentials = self._instance._client.credentials
             if isinstance(credentials, google.auth.credentials.Scoped):
                 credentials = credentials.with_scopes((SPANNER_DATA_SCOPE,))
+            client_info = self._instance._client._client_info
             self._spanner_api = SpannerClient(
-                credentials=credentials, client_info=_CLIENT_INFO
+                credentials=credentials, client_info=client_info
             )
         return self._spanner_api
 

spanner/tests/unit/test_client.py

@@ -49,13 +49,22 @@ def _make_one(self, *args, **kwargs):
         return self._get_target_class()(*args, **kwargs)
 
     def _constructor_test_helper(
-        self, expected_scopes, creds, user_agent=None, expected_creds=None
+        self,
+        expected_scopes,
+        creds,
+        user_agent=None,
+        expected_creds=None,
+        client_info=None,
     ):
         from google.cloud.spanner_v1 import client as MUT
 
         user_agent = user_agent or MUT.DEFAULT_USER_AGENT
+        client_info = client_info or MUT._CLIENT_INFO
         client = self._make_one(
-            project=self.PROJECT, credentials=creds, user_agent=user_agent
+            project=self.PROJECT,
+            credentials=creds,
+            user_agent=user_agent,
+            client_info=client_info,
         )
 
         expected_creds = expected_creds or creds.with_scopes.return_value
@@ -67,6 +76,7 @@ def _constructor_test_helper(
 
         self.assertEqual(client.project, self.PROJECT)
         self.assertEqual(client.user_agent, user_agent)
+        self.assertIs(client._client_info, client_info)
 
     def test_constructor_default_scopes(self):
         from google.cloud.spanner_v1 import client as MUT
@@ -85,6 +95,14 @@ def test_constructor_custom_user_agent_and_timeout(self):
             expected_scopes, creds, user_agent=CUSTOM_USER_AGENT
         )
 
+    def test_constructor_custom_client_info(self):
+        from google.cloud.spanner_v1 import client as MUT
+
+        client_info = mock.Mock()
+        expected_scopes = (MUT.SPANNER_ADMIN_SCOPE,)
+        creds = _make_credentials()
+        self._constructor_test_helper(expected_scopes, creds, client_info=client_info)
+
     def test_constructor_implicit_credentials(self):
         creds = _make_credentials()
 
@@ -102,10 +120,13 @@ def test_constructor_credentials_wo_create_scoped(self):
         self._constructor_test_helper(expected_scopes, creds)
 
     def test_instance_admin_api(self):
-        from google.cloud.spanner_v1.client import _CLIENT_INFO, SPANNER_ADMIN_SCOPE
+        from google.cloud.spanner_v1.client import SPANNER_ADMIN_SCOPE
 
         credentials = _make_credentials()
-        client = self._make_one(project=self.PROJECT, credentials=credentials)
+        client_info = mock.Mock()
+        client = self._make_one(
+            project=self.PROJECT, credentials=credentials, client_info=client_info
+        )
         expected_scopes = (SPANNER_ADMIN_SCOPE,)
 
         inst_module = "google.cloud.spanner_v1.client.InstanceAdminClient"
@@ -119,16 +140,19 @@ def test_instance_admin_api(self):
         self.assertIs(again, api)
 
         instance_admin_client.assert_called_once_with(
-            credentials=credentials.with_scopes.return_value, client_info=_CLIENT_INFO
+            credentials=credentials.with_scopes.return_value, client_info=client_info
         )
 
         credentials.with_scopes.assert_called_once_with(expected_scopes)
 
     def test_database_admin_api(self):
-        from google.cloud.spanner_v1.client import _CLIENT_INFO, SPANNER_ADMIN_SCOPE
+        from google.cloud.spanner_v1.client import SPANNER_ADMIN_SCOPE
 
         credentials = _make_credentials()
-        client = self._make_one(project=self.PROJECT, credentials=credentials)
+        client_info = mock.Mock()
+        client = self._make_one(
+            project=self.PROJECT, credentials=credentials, client_info=client_info
+        )
         expected_scopes = (SPANNER_ADMIN_SCOPE,)
 
         db_module = "google.cloud.spanner_v1.client.DatabaseAdminClient"
@@ -142,7 +166,7 @@ def test_database_admin_api(self):
         self.assertIs(again, api)
 
         database_admin_client.assert_called_once_with(
-            credentials=credentials.with_scopes.return_value, client_info=_CLIENT_INFO
+            credentials=credentials.with_scopes.return_value, client_info=client_info
         )
 
         credentials.with_scopes.assert_called_once_with(expected_scopes)

spanner/tests/unit/test_database.py

@@ -231,9 +231,8 @@ def test_name_property(self):
         self.assertEqual(database.name, expected_name)
 
     def test_spanner_api_property_w_scopeless_creds(self):
-        from google.cloud.spanner_v1.database import _CLIENT_INFO
-
         client = _Client()
+        client_info = client._client_info = mock.Mock()
         credentials = client.credentials = object()
         instance = _Instance(self.INSTANCE_NAME, client=client)
         pool = _Pool()
@@ -251,12 +250,12 @@ def test_spanner_api_property_w_scopeless_creds(self):
         self.assertIs(again, api)
 
         spanner_client.assert_called_once_with(
-            credentials=credentials, client_info=_CLIENT_INFO
+            credentials=credentials, client_info=client_info
         )
 
     def test_spanner_api_w_scoped_creds(self):
         import google.auth.credentials
-        from google.cloud.spanner_v1.database import _CLIENT_INFO, SPANNER_DATA_SCOPE
+        from google.cloud.spanner_v1.database import SPANNER_DATA_SCOPE
 
         class _CredentialsWithScopes(google.auth.credentials.Scoped):
             def __init__(self, scopes=(), source=None):
@@ -271,6 +270,7 @@ def with_scopes(self, scopes):
 
         expected_scopes = (SPANNER_DATA_SCOPE,)
         client = _Client()
+        client_info = client._client_info = mock.Mock()
         credentials = client.credentials = _CredentialsWithScopes()
         instance = _Instance(self.INSTANCE_NAME, client=client)
         pool = _Pool()
@@ -290,7 +290,7 @@ def with_scopes(self, scopes):
         self.assertEqual(len(spanner_client.call_args_list), 1)
         called_args, called_kw = spanner_client.call_args
         self.assertEqual(called_args, ())
-        self.assertEqual(called_kw["client_info"], _CLIENT_INFO)
+        self.assertEqual(called_kw["client_info"], client_info)
         scoped = called_kw["credentials"]
         self.assertEqual(scoped._scopes, expected_scopes)
         self.assertIs(scoped._source, credentials)