test(handler/http): add 25+ tests for probe resistance, MITM, bypass, and edge cases
+698/-3 across auth_test, connect_test, handler_test, metadata_test, proxy_test, and websocket_test. Covers probeResistanceResponse (web/file/ host/code types), handleRequest (bypass/UDP-disabled/CONNECT edge cases), setupTrafficLimiter with observer, observeStats lifecycle, MITM metadata, HTTP/1.0 keep-alive proxyRoundTrip, websocket body recording, and sniff timeout handling. Fixes SetDeadline signatures from any→time.Time.
This commit is contained in:
@@ -6,8 +6,10 @@ import (
|
||||
"net"
|
||||
"net/http"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/go-gost/core/handler"
|
||||
stats_util "github.com/go-gost/x/internal/util/stats"
|
||||
xrecorder "github.com/go-gost/x/recorder"
|
||||
)
|
||||
|
||||
@@ -157,3 +159,77 @@ func TestDial_HostHash(t *testing.T) {
|
||||
t.Error("expected error from dial with nil router")
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetupTrafficLimiter_WithObserver(t *testing.T) {
|
||||
h := &httpHandler{
|
||||
options: handler.Options{
|
||||
Logger: &testLogger{},
|
||||
Service: "test",
|
||||
Observer: &testObserver{},
|
||||
},
|
||||
}
|
||||
h.stats = stats_util.NewHandlerStats("test", false)
|
||||
|
||||
client, _ := net.Pipe()
|
||||
defer client.Close()
|
||||
|
||||
result, done := h.setupTrafficLimiter(client, "test-client", "tcp", "example.com:80")
|
||||
if result == nil {
|
||||
t.Error("expected non-nil connection")
|
||||
}
|
||||
if done == nil {
|
||||
t.Error("expected non-nil cleanup function when observer is set")
|
||||
}
|
||||
done()
|
||||
}
|
||||
|
||||
func TestSniffAndHandle_Timeout(t *testing.T) {
|
||||
h := &httpHandler{
|
||||
options: handler.Options{
|
||||
Logger: &testLogger{},
|
||||
},
|
||||
}
|
||||
h.md.sniffing = true
|
||||
h.md.sniffingTimeout = 100 * time.Millisecond
|
||||
h.sniffer = &SnifferBuilder{}
|
||||
|
||||
client, server := net.Pipe()
|
||||
defer client.Close()
|
||||
defer server.Close()
|
||||
|
||||
go func() {
|
||||
// Write non-HTTP, non-TLS bytes slowly
|
||||
client.Write([]byte("NOT_HTTP_OR_TLS_DATA"))
|
||||
}()
|
||||
|
||||
cc, _ := net.Pipe()
|
||||
defer cc.Close()
|
||||
|
||||
handled, err := h.sniffAndHandle(context.Background(), server, cc, &xrecorder.HandlerRecorderObject{}, &testLogger{})
|
||||
if err != nil {
|
||||
t.Logf("sniff error: %v", err)
|
||||
}
|
||||
if handled {
|
||||
t.Error("expected not handled for unknown protocol")
|
||||
}
|
||||
}
|
||||
|
||||
func TestHandleConnect_WriteError(t *testing.T) {
|
||||
h := &httpHandler{
|
||||
options: handler.Options{
|
||||
Logger: &testLogger{},
|
||||
},
|
||||
}
|
||||
h.md.proxyAgent = defaultProxyAgent
|
||||
h.md.readTimeout = 15
|
||||
|
||||
client, server := net.Pipe()
|
||||
// Immediately close client so write fails
|
||||
client.Close()
|
||||
server.Close()
|
||||
|
||||
resp := testConnectResp()
|
||||
err := h.handleConnect(context.Background(), server, &xrecorder.HandlerRecorderObject{}, &testLogger{}, "example.com:443", resp)
|
||||
// Error from reading the closed conn
|
||||
t.Logf("handleConnect with closed conn: %v", err)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user