Bump timeout on UDP with fallback.

rthalley · Mar 1, 2024 · 51db0f6 · 51db0f6
1 parent aba268a
commit 51db0f6
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/test_query.py b/tests/test_query.py
@@ -210,7 +210,7 @@ def testQueryUDPFallback(self):
         for address in query_addresses:
             qname = dns.name.from_text(".")
             q = dns.message.make_query(qname, dns.rdatatype.DNSKEY)
-            (_, tcp) = dns.query.udp_with_fallback(q, address, timeout=2)
+            (_, tcp) = dns.query.udp_with_fallback(q, address, timeout=4)
             self.assertTrue(tcp)
 
     def testQueryUDPFallbackWithSocket(self):
@@ -226,7 +226,7 @@ def testQueryUDPFallbackWithSocket(self):
                     qname = dns.name.from_text(".")
                     q = dns.message.make_query(qname, dns.rdatatype.DNSKEY)
                     (_, tcp) = dns.query.udp_with_fallback(
-                        q, address, udp_sock=udp_s, tcp_sock=tcp_s, timeout=2
+                        q, address, udp_sock=udp_s, tcp_sock=tcp_s, timeout=4
                     )
                     self.assertTrue(tcp)
 
@@ -825,8 +825,11 @@ def test_wrong_id_wire_with_truncation_flag_and_truncation_raise(self):
         bad_r.flags |= dns.flags.TC
         bad_r_wire = bad_r.to_wire()
         self.mock_receive(
-            bad_r_wire, ("127.0.0.1", 53), self.good_r_wire, ("127.0.0.1", 53),
-            raise_on_truncation=True
+            bad_r_wire,
+            ("127.0.0.1", 53),
+            self.good_r_wire,
+            ("127.0.0.1", 53),
+            raise_on_truncation=True,
         )
 
     def test_bad_wire_not_ignored(self):