more coverage improvements

2 files changed, 6 insertions, 4 deletions

diff --git a/dns/_asyncio_backend.py b/dns/_asyncio_backend.py
index 4120624..ba7c2e7 100644
--- a/dns/_asyncio_backend.py
+++ b/dns/_asyncio_backend.py
@@ -12,7 +12,7 @@ import dns.exception
 def _get_running_loop():
     try:
         return asyncio.get_running_loop()
-    except AttributeError:
+    except AttributeError:  # pragma: no cover
         return asyncio.get_event_loop()
 
 
@@ -29,7 +29,7 @@ class _DatagramProtocol:
             self.recvfrom.set_result((data, addr))
             self.recvfrom = None
 
-    def error_received(self, exc):
+    def error_received(self, exc):  # pragma: no cover
         if self.recvfrom:
             self.recvfrom.set_exception(exc)
 
@@ -96,7 +96,7 @@ class StreamSocket(dns._asyncbackend.DatagramSocket):
         self.writer.close()
         try:
             await self.writer.wait_closed()
-        except AttributeError:
+        except AttributeError:  # pragma: no cover
             pass
 
     async def getpeername(self):
diff --git a/tests/test_async.py b/tests/test_async.py
index 0e1913e..c36a8f4 100644
--- a/tests/test_async.py
+++ b/tests/test_async.py
@@ -17,6 +17,7 @@
 
 import asyncio
 import socket
+import time
 import unittest
 
 import dns.asyncbackend
@@ -360,9 +361,10 @@ class AsyncTests(unittest.TestCase):
     def testSleep(self):
         async def run():
             before = time.time()
-            self.backend.sleep(0.1)
+            await self.backend.sleep(0.1)
             after = time.time()
             self.assertTrue(after - before >= 0.1)
+        self.async_run(run)
 
 try:
     import trio