1 files changed, 54 insertions, 0 deletions
diff --git a/tests/pytests/test_random_close.py b/tests/pytests/test_random_close.py
new file mode 100644
index 0000000..cadd8ef
--- /dev/null
+++ b/tests/pytests/test_random_close.py
@@ -0,0 +1,54 @@
+# SPDX-License-Identifier: GPL-3.0-or-later
+"""TLS test when forward target closes connection after one second
+
+Test utilizes TLS proxy, which forwards queries to configured
+resolver, but closes the connection 1s after establishing.
+
+Kresd must stay alive and be able to answer queries.
+"""
+
+import random
+import string
+import time
+
+from proxy import HINTS, kresd_tls_client, resolve_hint, TLSProxy
+import utils
+
+
+QPS = 500
+
+
+def random_string(size=32, chars=string.ascii_lowercase + string.digits):
+    return ''.join(random.choice(chars) for x in range(size))
+
+
+def rsa_cannon(sock, duration, domain='test.', qps=QPS):
+    end_time = time.time() + duration
+
+    while time.time() < end_time:
+        next_time = time.time() + 1/qps
+        buff, _ = utils.get_msgbuff('{}.{}'.format(random_string(), domain))
+        sock.sendall(buff)
+        time_left = next_time - time.time()
+        if time_left > 0:
+            time.sleep(time_left)
+
+
+def test_proxy_random_close(tmpdir):
+    proxy = TLSProxy(close=1000)
+
+    kresd_tls_client_kwargs = {
+        'verbose': False,
+        'policy_test_pass': True
+        }
+    kresd_fwd_target_kwargs = {
+        'verbose': False
+        }
+    with kresd_tls_client(str(tmpdir), proxy, kresd_tls_client_kwargs, kresd_fwd_target_kwargs) \
+            as kresd:
+        sock2 = kresd.ip_tcp_socket()
+        rsa_cannon(sock2, 20)
+        sock3 = kresd.ip_tcp_socket()
+        for hint in HINTS:
+            resolve_hint(sock3, hint)
+            time.sleep(0.1)