service/handler_test.go

package service

import (
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
)

func TestServeHTTP(t *testing.T) {
	// 注册服务
	handler := func(in struct{ Name string }) string {
		return "Hello " + in.Name
	}
	Host("*").POST("/hello", handler).Auth(0).Memo("say hello")

	rh := &RouteHandler{ws: DefaultServer}

	// 模拟请求
	req := httptest.NewRequest("POST", "/hello", strings.NewReader(`{"name":"Star"}`))
	req.Header.Set("Content-Type", "application/json")
	w := httptest.NewRecorder()

	rh.ServeHTTP(w, req)

	if w.Code != http.StatusOK {
		t.Errorf("Expected status 200, got %d", w.Code)
	}

	body := w.Body.String()
	if body != "Hello Star" {
		t.Errorf("Expected 'Hello Star', got '%s'", body)
	}
}

func TestServeHTTP_404(t *testing.T) {
	rh := &RouteHandler{ws: DefaultServer}
	req := httptest.NewRequest("GET", "/notfound", nil)
	w := httptest.NewRecorder()

	rh.ServeHTTP(w, req)

	if w.Code != http.StatusNotFound {
		t.Errorf("Expected status 404, got %d", w.Code)
	}
}

func TestServeHTTP_VerifyFailed(t *testing.T) {
	type ValidIn struct {
		Age int `verify:"between:18-100"`
	}
	handler := func(in ValidIn) string {
		return "ok"
	}
	Host("*").POST("/verify", handler).Auth(0).Memo("test verify")

	rh := &RouteHandler{ws: DefaultServer}
	req := httptest.NewRequest("POST", "/verify", strings.NewReader(`{"age":10}`))
	req.Header.Set("Content-Type", "application/json")
	w := httptest.NewRecorder()

	rh.ServeHTTP(w, req)

	if w.Code != http.StatusBadRequest {
		t.Errorf("Expected status 400, got %d", w.Code)
	}
}

func TestServeHTTP_Panic(t *testing.T) {
	Host("*").GET("/panic", func() string {
		panic("intentional panic")
	})

	rh := &RouteHandler{ws: DefaultServer}
	req := httptest.NewRequest("GET", "/panic", nil)
	w := httptest.NewRecorder()

	rh.ServeHTTP(w, req)

	if w.Code != http.StatusInternalServerError {
		t.Errorf("Expected status 500, got %d", w.Code)
	}
}

func TestFallbackRunsOnlyAfterRegisteredRoutesMiss(t *testing.T) {
	ws := NewWebServer()
	ws.Host("*").GET("/registered", func() string { return "registered" })
	ws.SetFallback(func(request *Request) string { return "fallback:" + request.URL.Path })
	handler := &RouteHandler{ws: ws}

	for _, test := range []struct {
		path string
		want string
	}{
		{"/registered", "registered"},
		{"/missing", "fallback:/missing"},
	} {
		request := httptest.NewRequest(http.MethodGet, test.path, nil)
		response := httptest.NewRecorder()
		handler.ServeHTTP(response, request)
		if response.Code != http.StatusOK || response.Body.String() != test.want {
			t.Fatalf("%s: got %d %q, want 200 %q", test.path, response.Code, response.Body.String(), test.want)
		}
	}
}