81 lines
2.0 KiB
Go
81 lines
2.0 KiB
Go
package service
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestStaticService(t *testing.T) {
|
|
// 创建临时测试目录和文件
|
|
tempDir, _ := os.MkdirTemp("", "static_test")
|
|
defer os.RemoveAll(tempDir)
|
|
|
|
testFile := filepath.Join(tempDir, "index.html")
|
|
os.WriteFile(testFile, []byte("<h1>Static Page</h1>"), 0644)
|
|
|
|
// 注册静态目录
|
|
Static("/ui", tempDir)
|
|
|
|
rh := &RouteHandler{}
|
|
|
|
// 测试成功访问
|
|
req := httptest.NewRequest("GET", "/ui/index.html", nil)
|
|
w := httptest.NewRecorder()
|
|
rh.ServeHTTP(w, req)
|
|
|
|
if w.Code != http.StatusOK {
|
|
t.Errorf("Expected 200, got %d", w.Code)
|
|
}
|
|
if body := w.Body.String(); body != "<h1>Static Page</h1>" {
|
|
t.Errorf("Content mismatch: %s", body)
|
|
}
|
|
|
|
// 测试 404
|
|
req404 := httptest.NewRequest("GET", "/ui/notfound.html", nil)
|
|
w404 := httptest.NewRecorder()
|
|
rh.ServeHTTP(w404, req404)
|
|
if w404.Code != http.StatusNotFound {
|
|
t.Errorf("Expected 404 for missing file, got %d", w404.Code)
|
|
}
|
|
}
|
|
|
|
func TestHostStaticService(t *testing.T) {
|
|
tempDir, _ := os.MkdirTemp("", "host_static_test")
|
|
defer os.RemoveAll(tempDir)
|
|
|
|
testFile := filepath.Join(tempDir, "index.html")
|
|
_ = os.WriteFile(testFile, []byte("<h1>Host Static Page</h1>"), 0644)
|
|
|
|
// 注册域名特定静态文件服务
|
|
Host("example.com").Static("/host-ui", tempDir)
|
|
|
|
rh := &RouteHandler{}
|
|
|
|
// 1. 匹配域名访问
|
|
req1 := httptest.NewRequest("GET", "/host-ui/index.html", nil)
|
|
req1.Host = "example.com"
|
|
w1 := httptest.NewRecorder()
|
|
rh.ServeHTTP(w1, req1)
|
|
|
|
if w1.Code != http.StatusOK {
|
|
t.Errorf("Expected 200, got %d", w1.Code)
|
|
}
|
|
if body := w1.Body.String(); body != "<h1>Host Static Page</h1>" {
|
|
t.Errorf("Content mismatch: %s", body)
|
|
}
|
|
|
|
// 2. 不匹配域名访问 (应该 404)
|
|
req2 := httptest.NewRequest("GET", "/host-ui/index.html", nil)
|
|
req2.Host = "other.com"
|
|
w2 := httptest.NewRecorder()
|
|
rh.ServeHTTP(w2, req2)
|
|
|
|
if w2.Code != http.StatusNotFound {
|
|
t.Errorf("Expected 404 for mismatched host, got %d", w2.Code)
|
|
}
|
|
}
|
|
|