From: Bob Halley <halley@dnspython.org>
Date: Fri, 19 Jun 2020 14:15:08 +0000 (-0700)
Subject: improve async coverage
X-Git-Tag: v2.0.0rc1~30
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=117349d3a58533e21258d312c11ca8b5aeefc82c;p=thirdparty%2Fdnspython.git

improve async coverage
---

diff --git a/tests/test_async.py b/tests/test_async.py
index 42f4c69a..0e1913e5 100644
--- a/tests/test_async.py
+++ b/tests/test_async.py
@@ -342,6 +342,28 @@ class AsyncTests(unittest.TestCase):
             (_, tcp) = self.async_run(run)
             self.assertFalse(tcp)
 
+    def testUDPReceiveTimeout(self):
+        async def arun():
+            async with await self.backend.make_socket(socket.AF_INET,
+                                                      socket.SOCK_DGRAM) as s:
+                try:
+                    # for basic coverage
+                    await s.getpeername()
+                except Exception:
+                    # we expect failure as we haven't connected the socket
+                    pass
+                await s.recvfrom(1000, 0.05)
+        def run():
+            self.async_run(arun)
+        self.assertRaises(dns.exception.Timeout, run)
+
+    def testSleep(self):
+        async def run():
+            before = time.time()
+            self.backend.sleep(0.1)
+            after = time.time()
+            self.assertTrue(after - before >= 0.1)
+
 try:
     import trio
     import sniffio