Function TestIsDirectRequest

handler/direct_test.go:36–87

(t *testing.T)

}

func TestIsDirectRequest(t *testing.T) {
	tests := []struct {
		name string
		req  *http.Request
		want bool
	}{
		{
			name: "origin form get",
			req: &http.Request{
				Method: http.MethodGet,
				URL:    &url.URL{Path: "/"},
				Host:   "web.nacl.one",
			},
			want: true,
		},
		{
			name: "absolute form get",
			req: &http.Request{
				Method: http.MethodGet,
				URL:    &url.URL{Scheme: "http", Host: "openrouter.ai", Path: "/"},
				Host:   "openrouter.ai",
			},
			want: false,
		},
		{
			name: "connect",
			req: &http.Request{
				Method: http.MethodConnect,
				URL:    &url.URL{Host: "openrouter.ai:443"},
				Host:   "openrouter.ai:443",
			},
			want: false,
		},
		{
			name: "trust tunnel random",
			req: &http.Request{
				Method: "GETRANDOM",
				URL:    &url.URL{Path: "/32"},
				Host:   "web.nacl.one",
			},
			want: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if got := isDirectRequest(tt.req); got != tt.want {
				t.Fatalf("isDirectRequest() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestDirectResponse(t *testing.T) {
	proxy := NewProxyHandler(&Config{