Update test_client_ws.py

tests/test_client_ws.py

@@ -2,7 +2,7 @@
 import base64
 import hashlib
 import os
-from typing import Mapping
+from typing import Mapping, Type
 from unittest import mock
 
 import pytest
@@ -535,8 +535,9 @@ async def test_close_exc2(
                     await resp.close()
 
 
+@pytest.mark.parametrize("exc", (ClientConnectionResetError, ConnectionResetError))
 async def test_send_data_after_close(
-    ws_key: bytes, key_data: bytes, loop: asyncio.AbstractEventLoop
+    exc: Type[Exception], ws_key: bytes, key_data: bytes, loop: asyncio.AbstractEventLoop
 ) -> None:
     mresp = mock.Mock()
     mresp.status = 101
@@ -562,7 +563,7 @@ async def test_send_data_after_close(
                 (resp.send_bytes, (b"b",)),
                 (resp.send_json, ({},)),
             ):
-                with pytest.raises(ConnectionResetError):
+                with pytest.raises(exc):  # Verify exc can be caught with both classes
                     await meth(*args)