From f333e7f11646b22f6ca0a4964dcf485e587a9388 Mon Sep 17 00:00:00 2001 From: Bob Halley Date: Tue, 1 Sep 2020 06:38:18 -0700 Subject: [PATCH] more test coverage for dns.query._matches_destination() --- tests/test_query.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) diff --git a/tests/test_query.py b/tests/test_query.py index a13833ef..7a1ec718 100644 --- a/tests/test_query.py +++ b/tests/test_query.py @@ -540,3 +540,32 @@ class LowLevelWaitTests(unittest.TestCase): finally: l.close() r.close() + + +class MiscTests(unittest.TestCase): + def test_matches_destination(self): + self.assertTrue(dns.query._matches_destination(socket.AF_INET, + ('10.0.0.1', 1234), + ('10.0.0.1', 1234), + True)) + self.assertTrue(dns.query._matches_destination(socket.AF_INET6, + ('1::2', 1234), + ('0001::2', 1234), + True)) + self.assertTrue(dns.query._matches_destination(socket.AF_INET, + ('10.0.0.1', 1234), + None, + True)) + self.assertFalse(dns.query._matches_destination(socket.AF_INET, + ('10.0.0.1', 1234), + ('10.0.0.2', 1234), + True)) + self.assertFalse(dns.query._matches_destination(socket.AF_INET, + ('10.0.0.1', 1234), + ('10.0.0.1', 1235), + True)) + with self.assertRaises(dns.query.UnexpectedSource): + dns.query._matches_destination(socket.AF_INET, + ('10.0.0.1', 1234), + ('10.0.0.1', 1235), + False) -- 2.47.3