Update drivers repr

unmade · Mar 9, 2020 · a79a151 · a79a151
1 parent dd1a5a8
commit a79a151
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 7 deletions.
diff --git a/src/apiwrappers/drivers/aiohttp.py b/src/apiwrappers/drivers/aiohttp.py
@@ -43,7 +43,8 @@ def __repr__(self) -> str:
             f"{self.__class__.__name__}("
             f"{', '.join(middleware)}"
             f"timeout={self.timeout}, "
-            f"verify={self.verify}"
+            f"verify={repr(self.verify)}, "
+            f"cert={repr(self.cert)}"
             ")"
         )
 

diff --git a/src/apiwrappers/drivers/requests.py b/src/apiwrappers/drivers/requests.py
@@ -38,8 +38,9 @@ def __repr__(self) -> str:
         return (
             f"{self.__class__.__name__}("
             f"{', '.join(middleware)}"
-            f"timeout={self.timeout}, "
-            f"verify={self.verify}"
+            f"timeout={repr(self.timeout)}, "
+            f"verify={repr(self.verify)}, "
+            f"cert={repr(self.cert)}"
             ")"
         )
 

diff --git a/tests/test_drivers/test_aiohttp.py b/tests/test_drivers/test_aiohttp.py
@@ -86,15 +86,25 @@ async def echo(request: Request):
 async def test_representation() -> None:
     driver = aiohttp_driver()
     setattr(driver, "_middleware", [])
-    assert repr(driver) == "AioHttpDriver(timeout=300, verify=True)"
+    assert repr(driver) == "AioHttpDriver(timeout=300, verify=True, cert=None)"
 
 
 async def test_representation_with_middleware() -> None:
     driver = aiohttp_driver(RequestMiddleware, ResponseMiddleware)
     assert repr(driver) == (
         "AioHttpDriver("
         "Authentication, RequestMiddleware, ResponseMiddleware, "
-        "timeout=300, verify=True"
+        "timeout=300, verify=True, cert=None"
+        ")"
+    )
+
+
+async def test_representation_with_verify_and_cert() -> None:
+    driver = aiohttp_driver(verify=CA_BUNDLE, cert=CLIENT_CERT_PAIR)
+    assert repr(driver) == (
+        "AioHttpDriver("
+        "Authentication, "
+        f"timeout=300, verify='{CA_BUNDLE}', cert={CLIENT_CERT_PAIR}"
         ")"
     )
 

diff --git a/tests/test_drivers/test_requests.py b/tests/test_drivers/test_requests.py
@@ -76,15 +76,25 @@ def echo(request: PreparedRequest):
 def test_representation() -> None:
     driver = requests_driver()
     setattr(driver, "_middleware", [])
-    assert repr(driver) == "RequestsDriver(timeout=300, verify=True)"
+    assert repr(driver) == "RequestsDriver(timeout=300, verify=True, cert=None)"
 
 
 def test_representation_with_middleware() -> None:
     driver = requests_driver(RequestMiddleware, ResponseMiddleware)
     assert repr(driver) == (
         "RequestsDriver("
         "Authentication, RequestMiddleware, ResponseMiddleware, "
-        "timeout=300, verify=True"
+        "timeout=300, verify=True, cert=None"
+        ")"
+    )
+
+
+def test_representation_with_verify_and_cert() -> None:
+    driver = requests_driver(verify=CA_BUNDLE, cert=CLIENT_CERT_PAIR)
+    assert repr(driver) == (
+        "RequestsDriver("
+        "Authentication, "
+        f"timeout=300, verify='{CA_BUNDLE}', cert={CLIENT_CERT_PAIR}"
         ")"
     )