service/static_test.go

package service

import (
	"net/http"
	"net/http/httptest"
	"os"
	"path/filepath"
	"testing"
)

func writeStaticFixture(t *testing.T, content string) http.FileSystem {
	t.Helper()
	dir := t.TempDir()
	if err := os.WriteFile(filepath.Join(dir, "index.html"), []byte(content), 0644); err != nil {
		t.Fatal(err)
	}
	return http.Dir(dir)
}

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{ws: DefaultServer}

	// 测试成功访问
	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{ws: DefaultServer}

	// 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)
	}
}

func TestStaticFSHostCandidatesAndRange(t *testing.T) {
	ws := NewWebServer()
	ws.ReplaceStaticFS("*", map[string]http.FileSystem{"/": writeStaticFixture(t, "default")})
	ws.ReplaceStaticFS(":8081", map[string]http.FileSystem{"/": writeStaticFixture(t, "port")})
	ws.ReplaceStaticFS("aaa.com", map[string]http.FileSystem{"/": writeStaticFixture(t, "host")})
	ws.ReplaceStaticFS("aaa.com:8081", map[string]http.FileSystem{"/": writeStaticFixture(t, "exact")})
	handler := &RouteHandler{ws: ws}

	for _, test := range []struct {
		host string
		want string
	}{
		{"aaa.com:8081", "exact"},
		{"aaa.com:9090", "host"},
		{"other.com:8081", "port"},
		{"other.com:9090", "default"},
	} {
		req := httptest.NewRequest(http.MethodGet, "/", nil)
		req.Host = test.host
		out := httptest.NewRecorder()
		handler.ServeHTTP(out, req)
		if out.Code != http.StatusOK || out.Body.String() != test.want {
			t.Fatalf("host %s: got %d %q, want 200 %q", test.host, out.Code, out.Body.String(), test.want)
		}
	}

	req := httptest.NewRequest(http.MethodGet, "/", nil)
	req.Host = "aaa.com:8081"
	req.Header.Set("Range", "bytes=1-2")
	out := httptest.NewRecorder()
	handler.ServeHTTP(out, req)
	if out.Code != http.StatusPartialContent || out.Body.String() != "xa" {
		t.Fatalf("range response = %d %q", out.Code, out.Body.String())
	}
}

func TestStaticFSExclusiveStopsFileFallback(t *testing.T) {
	ws := NewWebServer()
	ws.ReplaceStaticFS("*", map[string]http.FileSystem{"/": writeStaticFixture(t, "default")})
	ws.ReplaceStaticFSExclusive("isolated.example", map[string]http.FileSystem{"/": writeStaticFixture(t, "isolated")})
	handler := &RouteHandler{ws: ws}

	request := httptest.NewRequest(http.MethodGet, "/missing.txt", nil)
	request.Host = "isolated.example"
	response := httptest.NewRecorder()
	handler.ServeHTTP(response, request)
	if response.Code != http.StatusNotFound {
		t.Fatalf("exclusive host missing file returned %d, want 404", response.Code)
	}

	request = httptest.NewRequest(http.MethodGet, "/", nil)
	request.Host = "unknown.example"
	response = httptest.NewRecorder()
	handler.ServeHTTP(response, request)
	if response.Code != http.StatusOK || response.Body.String() != "default" {
		t.Fatalf("unmatched host = %d %q, want default", response.Code, response.Body.String())
	}
}