Dedupe

Signed-off-by: Anuraag Agrawal <[email protected]>

Author: anuraaga

test/test_lifespan.py

@@ -17,16 +17,17 @@
 from .haberdasher_pb2 import Hat, Size
 
 
-@pytest.mark.asyncio
-async def test_lifespan() -> None:
-    class CountingHaberdasher(Haberdasher):
-        def __init__(self, counter: Counter) -> None:
-            self._counter = counter
+class CountingHaberdasher(Haberdasher):
+    def __init__(self, counter: Counter) -> None:
+        self._counter = counter
+
+    async def make_hat(self, request, ctx):
+        self._counter["requests"] += 1
+        return Hat(size=request.inches, color="blue")
 
-        async def make_hat(self, request, ctx):
-            self._counter["requests"] += 1
-            return Hat(size=request.inches, color="blue")
 
+@pytest.mark.asyncio
+async def test_lifespan() -> None:
     final_count = None
 
     async def counting_haberdasher():
@@ -69,14 +70,6 @@ async def counting_haberdasher():
 
 @pytest.mark.asyncio
 async def test_lifespan_startup_error() -> None:
-    class CountingHaberdasher(Haberdasher):
-        def __init__(self, counter: Counter) -> None:
-            self._counter = counter
-
-        async def make_hat(self, request, ctx):
-            self._counter["requests"] += 1
-            return Hat(size=request.inches, color="blue")
-
     final_count = None
 
     async def counting_haberdasher():
@@ -112,14 +105,6 @@ async def counting_haberdasher():
 
 @pytest.mark.asyncio
 async def test_lifespan_shutdown_error() -> None:
-    class CountingHaberdasher(Haberdasher):
-        def __init__(self, counter: Counter) -> None:
-            self._counter = counter
-
-        async def make_hat(self, request, ctx):
-            self._counter["requests"] += 1
-            return Hat(size=request.inches, color="blue")
-
     async def counting_haberdasher():
         counter = Counter()
         try:
@@ -167,14 +152,6 @@ async def counting_haberdasher():
 
 @pytest.mark.asyncio
 async def test_lifespan_not_supported() -> None:
-    class CountingHaberdasher(Haberdasher):
-        def __init__(self, counter: Counter) -> None:
-            self._counter = counter
-
-        async def make_hat(self, request, ctx):
-            self._counter["requests"] += 1
-            return Hat(size=request.inches, color="blue")
-
     final_count = None
 
     async def counting_haberdasher():