From d1476edf917107b0533e8d766694250c91cdb070 Mon Sep 17 00:00:00 2001
From: yusing <yusing@6uo.me>
Date: Sun, 14 Sep 2025 00:05:05 +0800
Subject: [PATCH] test(middleware): update bypass and rule tests

---
 internal/net/gphttp/middleware/bypass_test.go | 23 ++++++++++++++++---
 internal/route/rules/do_test.go               |  7 +++++-
 2 files changed, 26 insertions(+), 4 deletions(-)

diff --git a/internal/net/gphttp/middleware/bypass_test.go b/internal/net/gphttp/middleware/bypass_test.go
index e8d8567d..86a6ab9b 100644
--- a/internal/net/gphttp/middleware/bypass_test.go
+++ b/internal/net/gphttp/middleware/bypass_test.go
@@ -2,8 +2,11 @@ package middleware_test
 
 import (
 	"io"
+	"net"
 	"net/http"
 	"net/http/httptest"
+	"net/url"
+	"strconv"
 	"strings"
 	"testing"
 
@@ -135,17 +138,31 @@ func TestReverseProxyBypass(t *testing.T) {
 }
 
 func TestEntrypointBypassRoute(t *testing.T) {
-	go http.ListenAndServe(":8080", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		w.Write([]byte("test"))
 	}))
+	defer srv.Close()
+
+	url, err := url.Parse(srv.URL)
+	expect.NoError(t, err)
+
+	host, port, err := net.SplitHostPort(url.Host)
+	expect.NoError(t, err)
+
+	portInt, err := strconv.Atoi(port)
+	expect.NoError(t, err)
+
+	expect.NoError(t, err)
 	entry := entrypoint.NewEntrypoint()
 	r := &route.Route{
 		Alias: "test-route",
+		Host:  host,
 		Port: routeTypes.Port{
-			Proxy: 8080,
+			Proxy: portInt,
 		},
 	}
-	err := entry.SetMiddlewares([]map[string]any{
+
+	err = entry.SetMiddlewares([]map[string]any{
 		{
 			"use":    "redirectHTTP",
 			"bypass": []string{"route test-route"},
diff --git a/internal/route/rules/do_test.go b/internal/route/rules/do_test.go
index 5b1d69bc..0de802c9 100644
--- a/internal/route/rules/do_test.go
+++ b/internal/route/rules/do_test.go
@@ -42,7 +42,7 @@ func TestParseCommands(t *testing.T) {
 		// serve tests
 		{
 			name:    "serve_valid",
-			input:   "serve /var/www",
+			input:   "serve /",
 			wantErr: nil,
 		},
 		{
@@ -50,6 +50,11 @@ func TestParseCommands(t *testing.T) {
 			input:   "serve ",
 			wantErr: ErrInvalidArguments,
 		},
+		{
+			name:    "serve_non_exist_path",
+			input:   "serve /non-exist-path",
+			wantErr: ErrInvalidArguments,
+		},
 		{
 			name:    "serve_too_many_args",
 			input:   "serve / / /",