From 43ceedc68fe92ffaeb2f5f3bb41087744018e110 Mon Sep 17 00:00:00 2001
From: Anton Yuzhaninov <citrin+git@citrin.ru>
Date: Wed, 24 Oct 2018 21:08:31 -0400
Subject: TCP Connect robot keyword

* reduce timeout to a reasonable value
* explicitly close socket, don't wait for GC
---
 test/functional/lib/rspamd.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'test/functional')

diff --git a/test/functional/lib/rspamd.py b/test/functional/lib/rspamd.py
index e4462c18c..6fc1c0e67 100644
--- a/test/functional/lib/rspamd.py
+++ b/test/functional/lib/rspamd.py
@@ -160,8 +160,15 @@ def spamc(addr, port, filename):
     return r.decode('utf-8')
 
 def TCP_Connect(addr, port):
+    """Attempts to open a TCP connection to specified address:port
+
+    Example:
+    | Wait Until Keyword Succeeds | 5s | 10ms | TCP Connect | localhost | 8080 |
+    """
     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.settimeout(5) # seconds
     s.connect((addr, port))
+    s.close()
 
 def update_dictionary(a, b):
     a.update(b)
-- 
cgit v1.2.3