]> git.ipfire.org Git - thirdparty/knot-resolver.git/commitdiff
pytests: adjust test_slow_lorris_attack
authorTomas Krizek <tomas.krizek@nic.cz>
Wed, 14 Nov 2018 14:50:54 +0000 (15:50 +0100)
committerTomas Krizek <tomas.krizek@nic.cz>
Tue, 4 Dec 2018 16:13:42 +0000 (17:13 +0100)
tests/pytests/test_conn_mgmt.py

index 6658d9388dda9a808e2c8d6497de9c7734dfcc3a..8a6f1d468b3cd2b228f4fa317d230a867aa590a5 100644 (file)
@@ -74,8 +74,14 @@ def test_slow_lorris_attack(kresd_sock):
     """
     buff, _ = utils.get_msgbuff()
 
+    time.sleep(3)
+    utils.ping_alive(kresd_sock)  # to reset internal kresd timer
+    end_time = time.time() + utils.MAX_TIMEOUT
+
     with utils.expect_kresd_close():
         for i in range(len(buff)):
             b = buff[i:i+1]
             kresd_sock.send(b)
+            if time.time() > end_time:
+                break
             time.sleep(1)