feat: add comprehensive HTTP e2e tests covering connector, TLS, probeResist, idleTimeout, and UDP relay

Adds 12 new HTTP handler test scenarios: HTTP connector (no-auth, auth, TLS upstream), CONNECT tunnel (no-sniffing, bypass-403), probeResist (host, web, file, knock), idleTimeout, and UDP relay over HTTP. Introduces RunGostContainerWithFiles helper for mounting extra files into test containers.
2026-06-26 22:00:35 +08:00
parent 3c87a8b96e
commit 69b561f944
23 changed files with 1531 additions and 10 deletions
@@ -0,0 +1,60 @@
+import socket
+import sys
+import time
+
+
+def main():
+    host = sys.argv[1] if len(sys.argv) > 1 else "127.0.0.1"
+    port = int(sys.argv[2]) if len(sys.argv) > 2 else 8080
+    idle_timeout = int(sys.argv[3]) if len(sys.argv) > 3 else 3
+
+    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    s.settimeout(10)
+    s.connect((host, port))
+
+    # CONNECT to echo server
+    req = b"CONNECT tcp-echo:5678 HTTP/1.1\r\nHost: tcp-echo:5678\r\n\r\n"
+    s.sendall(req)
+
+    # Read 200 OK response
+    resp = b""
+    while b"\r\n\r\n" not in resp:
+        chunk = s.recv(4096)
+        if not chunk:
+            break
+        resp += chunk
+
+    if b"200" not in resp:
+        print(f"FAIL: expected 200, got {resp.decode(errors='replace')}")
+        sys.exit(1)
+
+    # Send request through the CONNECT tunnel
+    s.sendall(b"ping-1\n")
+    data = s.recv(4096)
+    if b"ping-1" not in data:
+        print(f"FAIL: expected ping-1 echo, got {data!r}")
+        sys.exit(1)
+    print(f"PASS: first request through tunnel succeeded")
+
+    # Wait longer than idleTimeout
+    wait = idle_timeout + 2
+    print(f"Waiting {wait}s for idle timeout...")
+    time.sleep(wait)
+
+    # Try to send more data — idle timeout should have closed the pipe
+    try:
+        s.sendall(b"ping-2\n")
+        data = s.recv(4096)
+        if not data:
+            print("PASS: connection closed after idle timeout (empty recv)")
+            sys.exit(0)
+        # Got data means the pipe is still alive
+        print(f"FAIL: connection still alive after idle timeout, got {data!r}")
+        sys.exit(1)
+    except (socket.timeout, ConnectionResetError, BrokenPipeError, OSError) as e:
+        print(f"PASS: connection closed after idle timeout: {e}")
+        sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()