Clean up async socket compatibility code

Author: robsdedude

neo4j/_async_compat/network/bolt_socket.py

@@ -69,6 +69,9 @@ def __init__(self, reader, protocol, writer):  # , loop):
         self._timeout = None
 
     async def _wait_for(self, fut):
+        if self._timeout <= 0:
+            # give the io-operation time for one loop cycle to do its thing
+            await asyncio.sleep(0)
         try:
             return await asyncio.wait_for(fut, self._timeout)
         except asyncio.TimeoutError:
@@ -99,23 +102,10 @@ def settimeout(self, timeout):
             self._timeout = timeout
 
     async def recv(self, n):
-        # fut = self._loop.sock_recv(self._socket, n)
-
-        if self._timeout == 0:
-            # TODO: is this ok? It's a nonblocking socket, after all...
-            return self._socket.recv(n)
         fut = self._reader.read(n)
         return await self._wait_for(fut)
 
     async def recv_into(self, buffer, nbytes):
-        # if n < len(buff):
-        #     fut = self._loop.sock_recv_into(self._socket,
-        #                                     memoryview(buff[:n]))
-        # else:
-        #     fut = self._loop.sock_recv_into(self._socket, buff)
-
-        if self._timeout == 0:
-            return self._socket.recv_into(buffer, nbytes)
         # FIXME: not particularly memory or time efficient
         fut = self._reader.read(nbytes)
         res = await self._wait_for(fut)