[Cosmos] create new async user agent for async client

sdk/cosmos/azure-cosmos/CHANGELOG.md

@@ -5,6 +5,9 @@
 This version and all future versions will require Python 3.6+. Python 2.7 is no longer supported.
 We will also be removing support for Python 3.6 and will only support Python 3.7+ starting December 2022.
 
+#### Other Changes
+- Added async user agent for async client
+
 ### 4.3.0b1 (2021-12-14)
 **New features**
 - Added language native async i/o client

sdk/cosmos/azure-cosmos/azure/cosmos/_cosmos_client_connection.py

@@ -182,10 +182,12 @@ def __init__(
             proxy = host if url.port else host + ":" + str(self.connection_policy.ProxyConfiguration.Port)
             proxies.update({url.scheme: proxy})
 
+        self._user_agent = _utils.get_user_agent()
+
         policies = [
             HeadersPolicy(**kwargs),
             ProxyPolicy(proxies=proxies),
-            UserAgentPolicy(base_user_agent=_utils.get_user_agent(), **kwargs),
+            UserAgentPolicy(base_user_agent=self._user_agent, **kwargs),
             ContentDecodePolicy(),
             retry_policy,
             CustomHookPolicy(**kwargs),

sdk/cosmos/azure-cosmos/azure/cosmos/_utils.py

@@ -34,6 +34,13 @@ def get_user_agent():
     return user_agent
 
 
+def get_user_agent_async():
+    os_name = safe_user_agent_header(platform.platform())
+    python_version = safe_user_agent_header(platform.python_version())
+    user_agent = "azsdk-python-cosmos-async/{} Python/{} ({})".format(VERSION, python_version, os_name)
+    return user_agent
+
+
 def safe_user_agent_header(s):
     if s is None:
         s = "unknown"

sdk/cosmos/azure-cosmos/azure/cosmos/aio/_cosmos_client_connection_async.py

@@ -178,12 +178,14 @@ def __init__(
             host = self.connection_policy.ProxyConfiguration.Host
             url = urlparse(host)
             proxy = host if url.port else host + ":" + str(self.connection_policy.ProxyConfiguration.Port)
-            proxies.update({url.scheme : proxy})
+            proxies.update({url.scheme: proxy})
+
+        self._user_agent = _utils.get_user_agent_async()
 
         policies = [
             HeadersPolicy(**kwargs),
             ProxyPolicy(proxies=proxies),
-            UserAgentPolicy(base_user_agent=_utils.get_user_agent(), **kwargs),
+            UserAgentPolicy(base_user_agent=self._user_agent, **kwargs),
             ContentDecodePolicy(),
             retry_policy,
             CustomHookPolicy(**kwargs),

sdk/cosmos/azure-cosmos/test/test_client_user_agent.py

@@ -0,0 +1,53 @@
+# The MIT License (MIT)
+# Copyright (c) 2021 Microsoft Corporation
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+import unittest
+
+import azure.cosmos.cosmos_client as sync_client
+import azure.cosmos.aio.cosmos_client as async_client
+import pytest
+import asyncio
+from test_config import _test_config
+
+# This test class serves to test user-configurable options and verify they are
+# properly set and saved into the different object instances that use these
+# user-configurable settings.
+
+pytestmark = pytest.mark.cosmosEmulator
+
+
+@pytest.mark.usefixtures("teardown")
+class TestClientUserAgent(unittest.TestCase):
+
+    async def test_client_user_agent(self):
+        client_sync = sync_client.CosmosClient(url=_test_config.host, credential=_test_config.masterKey)
+        client_async = async_client.CosmosClient(url=_test_config.host, credential=_test_config.masterKey)
+
+        self.assertTrue(client_sync.client_connection._user_agent.startswith("azsdk-python-cosmos/"))
+        self.assertTrue(client_async.client_connection._user_agent.startswith("azsdk-python-cosmos-async/"))
+        self.assertTrue(client_async.client_connection._user_agent != client_sync.client_connection._user_agent)
+
+        await client_async.close()
+
+
+if __name__ == "__main__":
+    event_loop = asyncio.get_event_loop()
+    event_loop.run_until_complete(unittest.main())