diff --git a/goblin/driver/connection.py b/goblin/driver/connection.py index 1653088e31bdc20cf4644f4a8b1461c61c6b8659..1f249703ccc369991aed18f67b5c111eed2fe88b 100644 --- a/goblin/driver/connection.py +++ b/goblin/driver/connection.py @@ -16,9 +16,10 @@ Message = collections.namedtuple( class AsyncResponseIter: - def __init__(self, response_queue, loop: + def __init__(self, response_queue, loop, conn): self._response_queue = response_queue self._loop = loop + self._conn = conn async def __aiter__(self): return self @@ -134,9 +135,7 @@ class Connection(AbstractConnection): response_queue = asyncio.Queue(loop=self._loop) self.response_queues[request_id] = response_queue self._ws.send_bytes(message) - return AsyncResponseIter(response_queue, self._loop, self, - self._username, self._password, - processor, session) + return AsyncResponseIter(response_queue, self._loop, self) async def close(self): await self._ws.close()