Move connection to redis to DelayProxy.start()

Author: kristjanvalur

tests/test_asyncio/test_cwe_404.py

@@ -6,6 +6,7 @@
 
 from redis.asyncio import Redis
 from redis.asyncio.cluster import RedisCluster
+from redis.asyncio.connection import async_timeout
 
 
 @pytest.fixture
@@ -40,8 +41,13 @@ def __init__(self, addr, redis_addr, delay: float):
         self.redis_addr = redis_addr
         self.delay = delay
         self.send_event = asyncio.Event()
+        self.redis_streams = None
 
     async def start(self):
+        # establish connection to redis
+        with async_timeout(2):
+            self.redis_streams = await asyncio.open_connection(*self.redis_addr)
+        # start local server
         self.server = await asyncio.start_server(self.handle, *self.addr)
         self.ROUTINE = asyncio.create_task(self.server.serve_forever())
 
@@ -59,8 +65,7 @@ def override(self, delay: float = 0.0):
             self.delay = old
 
     async def handle(self, reader, writer):
-        # establish connection to redis
-        redis_reader, redis_writer = await asyncio.open_connection(*self.redis_addr)
+        redis_reader, redis_writer = self.redis_streams
         pipe1 = asyncio.create_task(
             pipe(reader, redis_writer, self, "to redis:", self.send_event)
         )