Last changes, Firegex 2.1.2

tests/ph_test.py

@@ -49,7 +49,7 @@ time.sleep(0.5)
 def checkData(should_work):
     res = None
     try: res = server.sendCheckData(secrets.token_bytes(432))
-    except ConnectionRefusedError:  res = False
+    except (ConnectionRefusedError, TimeoutError): res = None
     if res:
         if should_work: puts(f"Successfully received data ✔", color=colors.green)
         else: puts("Test Failed: Connection wasn't blocked ✗", color=colors.red); exit_test(1)

tests/utils/tcpserver.py

@@ -26,7 +26,8 @@ class TcpServer:
 
     def sendCheckData(self,data):
         s = socket.socket(socket.AF_INET6 if self.ipv6 else socket.AF_INET, socket.SOCK_STREAM)
-        s.connect(('::1' if self.ipv6 else '127.0.0.1', self.proxy_port if self.proxy_port else self.port))
+        s.settimeout(1)
+        s.connect(('::1' if self.ipv6 else '127.0.0.1', self.proxy_port if self.proxy_port else self.port), )
         s.sendall(data)
         received_data = s.recv(4096)
         s.close()