fix cluster test to run within delay context

Author: kristjanvalur

tests/test_asyncio/test_cwe_404.py

@@ -219,15 +219,18 @@ def set_delay(delay: int = 0):
         await r.set("foo", "foo")
         await r.set("bar", "bar")
 
+        async def op(r):
+            with set_delay(delay=delay):
+                return await r.get("foo")
+
         all_clear()
-        with set_delay(delay=delay):
-            t = asyncio.create_task(r.get("foo"))
-            # One of the proxies will handle our request, wait for it to send
-            await any_wait()
-            await asyncio.sleep(delay)
-            t.cancel()
-            with pytest.raises(asyncio.CancelledError):
-                await t
+        t = asyncio.create_task(op(r))
+        # One of the proxies will handle our request, wait for it to send
+        await any_wait()
+        await asyncio.sleep(delay)
+        t.cancel()
+        with pytest.raises(asyncio.CancelledError):
+            await t
 
         # try a number of requests to excercise all the connections
         async def doit():