]> git.ipfire.org Git - thirdparty/rspamd.git/commitdiff
[Test] Fix sockets reusing
authorVsevolod Stakhov <vsevolod@highsecure.ru>
Fri, 31 May 2019 14:19:00 +0000 (15:19 +0100)
committerVsevolod Stakhov <vsevolod@highsecure.ru>
Fri, 31 May 2019 14:19:00 +0000 (15:19 +0100)
test/functional/util/dummy_http.py
test/functional/util/dummy_killer.py
test/functional/util/dummy_udp.py

index dc4cee3544dcccf0dad9af5f283f1bb1ca1ae267..614fced2952e53bcabc3e229c1737a76a458b0e4 100755 (executable)
@@ -30,7 +30,7 @@ class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
 
     def do_GET(self):
         response = "hello world"
-        
+
         """Respond to a GET request."""
         if self.path == "/empty":
             self.finish()
@@ -55,7 +55,7 @@ class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
         conntype = self.headers.get('Connection', "").lower()
         if conntype != 'keep-alive':
             self.close_connection = True
-        
+
         self.log_message("ka:'%s', pv:%s[%s]" % (str(conntype == 'keep-alive'), str(self.protocol_version >= "HTTP/1.1"), self.protocol_version))
 
 
@@ -73,7 +73,7 @@ class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
             self.send_response(403)
         else:
             self.send_response(200)
-            
+
         if self.path == "/content-length":
             self.send_header("Content-Length", str(len(response)))
 
@@ -85,10 +85,10 @@ class MyHandler(BaseHTTPServer.BaseHTTPRequestHandler):
 class ThreadingSimpleServer(SocketServer.ThreadingMixIn,
                    BaseHTTPServer.HTTPServer):
     def __init__(self):
-        BaseHTTPServer.HTTPServer.__init__(self, (HOST_NAME, PORT), MyHandler)
         self.allow_reuse_address = True
         self.timeout = 1
-        
+        BaseHTTPServer.HTTPServer.__init__(self, (HOST_NAME, PORT), MyHandler)
+
     def run(self):
         dummy_killer.write_pid(PID)
         try:
index 723e6ef4ac5433b00fb3064348dcc309307d7328..85b3d01ba9f9e389b7d46ec76dbb67c7bdc3d444 100644 (file)
@@ -5,7 +5,7 @@ import atexit
 def setup_killer(server, method = None):
     def default_method():
         server.server_close()
-    
+
     if method is None:
         method = default_method
 
@@ -14,7 +14,7 @@ def setup_killer(server, method = None):
 
     signal.signal(signal.SIGALRM, alarm_handler)
     signal.signal(signal.SIGTERM, alarm_handler)
-    signal.alarm(10)
+    signal.alarm(30)
 
 
 def write_pid(path):
index 4d904e3b120835a7125654b5a5f4e2bdf56d4420..02ee6dce842a7573a6896531c4e68fede58bf00c 100755 (executable)
@@ -15,6 +15,7 @@ if __name__ == "__main__":
         port = 5005
     sock = socket.socket(socket.AF_INET, # Internet
                          socket.SOCK_DGRAM) # UDP
+    sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
     sock.bind((UDP_IP, port))
     dummy_killer.write_pid(PID)