improve testing

Author: sanderegg

packages/service-library/tests/redis/test_semaphore.py

@@ -281,10 +281,16 @@ async def delayed_release() -> None:
     await semaphore2.release()
 
 
+@pytest.mark.parametrize(
+    "exception",
+    [RuntimeError, asyncio.CancelledError],
+    ids=str,
+)
 async def test_semaphore_context_manager_with_exception(
     redis_client_sdk: RedisClientSDK,
     semaphore_name: str,
     semaphore_capacity: int,
+    exception: type[Exception | asyncio.CancelledError],
 ):
     captured_semaphore: DistributedSemaphore | None = None
 
@@ -296,10 +302,9 @@ async def _raising_context():
         ) as sem:
             nonlocal captured_semaphore
             captured_semaphore = sem
-            msg = "Test exception"
-            raise RuntimeError(msg)
+            raise exception("Test")
 
-    with pytest.raises(RuntimeError, match="Test exception"):
+    with pytest.raises(exception, match="Test"):
         await _raising_context()
 
     # Should be released even after exception