Update exception tests.

Author: mattgd

tests/test_async_http_client.py

@@ -5,7 +5,12 @@
 from unittest.mock import AsyncMock
 
 from tests.test_sync_http_client import STATUS_CODE_TO_EXCEPTION_MAPPING
-from workos.exceptions import BadRequestException, BaseRequestException, ServerException
+from workos.exceptions import (
+    BadRequestException,
+    BaseRequestException,
+    ConflictException,
+    ServerException,
+)
 from workos.utils.http_client import AsyncHTTPClient
 
 
@@ -186,8 +191,6 @@ async def test_request_exceptions_include_expected_request_data(
         except expected_exception as ex:  # type: ignore
             assert ex.message == response_message
             assert ex.request_id == request_id
-        except Exception as ex:
-            # This'll fail for sure here but... just using the nice error that'd come up
             assert ex.__class__ == expected_exception
 
     async def test_bad_request_exceptions_include_expected_request_data(self):
@@ -210,7 +213,6 @@ async def test_bad_request_exceptions_include_expected_request_data(self):
                 str(ex)
                 == "(message=No message, request_id=request-123, error=example_error, error_description=Example error description)"
             )
-        except Exception as ex:
             assert ex.__class__ == BadRequestException
 
     async def test_bad_request_exceptions_exclude_expected_request_data(self):
@@ -228,7 +230,6 @@ async def test_bad_request_exceptions_exclude_expected_request_data(self):
             await self.http_client.request("bad_place")
         except BadRequestException as ex:
             assert str(ex) == "(message=No message, request_id=request-123, foo=bar)"
-        except Exception as ex:
             assert ex.__class__ == BadRequestException
 
     async def test_request_bad_body_raises_expected_exception_with_request_data(self):
@@ -247,10 +248,24 @@ async def test_request_bad_body_raises_expected_exception_with_request_data(self
         except ServerException as ex:
             assert ex.message == None
             assert ex.request_id == request_id
-        except Exception as ex:
-            # This'll fail for sure here but... just using the nice error that'd come up
             assert ex.__class__ == ServerException
 
+    async def test_conflict_exception(self):
+        request_id = "request-123"
+
+        self.http_client._client.request = AsyncMock(
+            return_value=httpx.Response(
+                status_code=409,
+                headers={"X-Request-ID": request_id},
+            ),
+        )
+
+        try:
+            await self.http_client.request("bad_place")
+        except ConflictException as ex:
+            assert str(ex) == "(message=No message, request_id=request-123)"
+            assert ex.__class__ == ConflictException
+
     async def test_request_includes_base_headers(
         self, capture_and_mock_http_client_request
     ):

tests/test_sync_http_client.py

@@ -201,8 +201,6 @@ def test_request_exceptions_include_expected_request_data(
         except expected_exception as ex:  # type: ignore
             assert ex.message == response_message
             assert ex.request_id == request_id
-        except Exception as ex:
-            # This'll fail for sure here but... just using the nice error that'd come up
             assert ex.__class__ == expected_exception
 
     def test_bad_request_exceptions_include_request_data(self):
@@ -229,7 +227,6 @@ def test_bad_request_exceptions_include_request_data(self):
                 str(ex)
                 == "(message=No message, request_id=request-123, error=example_error, error_description=Example error description, foo=bar)"
             )
-        except Exception as ex:
             assert ex.__class__ == BadRequestException
 
     def test_request_bad_body_raises_expected_exception_with_request_data(self):
@@ -248,8 +245,6 @@ def test_request_bad_body_raises_expected_exception_with_request_data(self):
         except ServerException as ex:
             assert ex.message == None
             assert ex.request_id == request_id
-        except Exception as ex:
-            # This'll fail for sure here but... just using the nice error that'd come up
             assert ex.__class__ == ServerException
 
     def test_conflict_exception(self):
@@ -266,7 +261,6 @@ def test_conflict_exception(self):
             self.http_client.request("bad_place")
         except ConflictException as ex:
             assert str(ex) == "(message=No message, request_id=request-123)"
-        except Exception as ex:
             assert ex.__class__ == ConflictException
 
     def test_request_includes_base_headers(self, capture_and_mock_http_client_request):