fix mocked-tests

digitalocean · May 15, 2024 · c622aaf · c622aaf
1 parent 634157b
commit c622aaf
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/mocked/test_client_customizations.py b/tests/mocked/test_client_customizations.py
@@ -17,15 +17,15 @@ def test_custom_headers():
     client = Client("", headers=custom_headers)
 
     # pylint: disable=protected-access
-    assert client._client._config.headers_policy.headers == custom_headers
+    assert client._config.headers_policy.headers == custom_headers
 
 
 def test_custom_timeout():
     timeout = 300
     client = Client("", timeout=timeout)
 
     # pylint: disable=protected-access
-    assert client._client._config.retry_policy.timeout == timeout
+    assert client._config.retry_policy.timeout == timeout
 
 
 def test_custom_endpoint():
@@ -42,7 +42,7 @@ def test_custom_logger():
     client = Client("", logger=logger)
 
     # pylint: disable=protected-access
-    assert client._client._config.http_logging_policy.logger.name == name
+    assert client._config.http_logging_policy.logger.name == name
 
 
 @responses.activate
@@ -59,7 +59,7 @@ def test_custom_user_agent():
     full_user_agent_pattern = r"^test azsdk-python-pydo\/.+Python\/.+\(.+\)$"
 
     # pylint: disable=protected-access
-    got_user_agent = client._client._config.user_agent_policy.user_agent
+    got_user_agent = client._config.user_agent_policy.user_agent
     match = re.match(full_user_agent_pattern, got_user_agent)
     assert match is not None