Carlos Florêncio
84 lines
2.5 KiB
Go
84 lines
2.5 KiB
Go
package webapp
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestDevHandlerInjectsBootPayloadIntoViteIndexForSPARoutes(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
vite := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/" {
|
|
t.Fatalf("vite received path %q, want /", r.URL.Path)
|
|
}
|
|
w.Header().Set("Content-Type", "text/html")
|
|
_, _ = w.Write([]byte(`<html><head></head><body><script type="module" src="/src/main.tsx"></script></body></html>`))
|
|
}))
|
|
t.Cleanup(vite.Close)
|
|
|
|
handler, err := NewDevHandler(vite.URL, WithPayloadBuilder(func(_ *http.Request, route RouteClassification) BootPayload {
|
|
payload := NewBootPayload(route, RuntimeConfig{APIPrefix: "/api/v1", WebSocketPath: "/ws"}, nil)
|
|
payload.RouteData = map[string]any{"from": "go"}
|
|
return payload
|
|
}))
|
|
if err != nil {
|
|
t.Fatalf("NewDevHandler: %v", err)
|
|
}
|
|
|
|
resp := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, "/tasks", nil)
|
|
req.Header.Set("Accept", "text/html")
|
|
handler.ServeHTTP(resp, req)
|
|
|
|
body := resp.Body.String()
|
|
if resp.Code != http.StatusOK {
|
|
t.Fatalf("status = %d, want %d body=%s", resp.Code, http.StatusOK, body)
|
|
}
|
|
if !strings.Contains(body, bootPayloadGlobal) {
|
|
t.Fatalf("body missing boot payload: %s", body)
|
|
}
|
|
if !strings.Contains(body, `"route":"tasks"`) {
|
|
t.Fatalf("body missing route metadata: %s", body)
|
|
}
|
|
if !strings.Contains(body, `"from":"go"`) {
|
|
t.Fatalf("body missing route data: %s", body)
|
|
}
|
|
if got := resp.Header().Get("Cache-Control"); got != "no-store" {
|
|
t.Fatalf("Cache-Control = %q, want no-store", got)
|
|
}
|
|
}
|
|
|
|
func TestDevHandlerProxiesViteModuleRequests(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
vite := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path != "/src/main.tsx" {
|
|
t.Fatalf("vite received path %q, want /src/main.tsx", r.URL.Path)
|
|
}
|
|
w.Header().Set("Content-Type", "application/javascript")
|
|
_, _ = w.Write([]byte(`console.log("vite module")`))
|
|
}))
|
|
t.Cleanup(vite.Close)
|
|
|
|
handler, err := NewDevHandler(vite.URL)
|
|
if err != nil {
|
|
t.Fatalf("NewDevHandler: %v", err)
|
|
}
|
|
|
|
resp := httptest.NewRecorder()
|
|
req := httptest.NewRequest(http.MethodGet, "/src/main.tsx", nil)
|
|
req.Header.Set("Accept", "*/*")
|
|
handler.ServeHTTP(resp, req)
|
|
|
|
body := resp.Body.String()
|
|
if resp.Code != http.StatusOK {
|
|
t.Fatalf("status = %d, want %d body=%s", resp.Code, http.StatusOK, body)
|
|
}
|
|
if got := strings.TrimSpace(body); got != `console.log("vite module")` {
|
|
t.Fatalf("body = %q, want proxied module", got)
|
|
}
|
|
}
|