From 40129eadd3c5caa71fcebb1c8bff1145910e6009 Mon Sep 17 00:00:00 2001
From: Bob Halley <halley@dnspython.org>
Date: Fri, 19 Jun 2020 07:53:00 -0700
Subject: more coverage improvements

---
 dns/_asyncio_backend.py | 6 +++---
 tests/test_async.py     | 4 +++-
 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
-- 
cgit v1.2.1