ghostream/web/handler_test.go

41 lines
1.0 KiB
Go

package web
import (
"net/http"
"net/http/httptest"
"testing"
)
func TestViewerPageGET(t *testing.T) {
// Load templates
if err := loadTemplates(); err != nil {
t.Errorf("Failed to load templates: %v", err)
}
cfg = &Options{}
// Test GET request
r, _ := http.NewRequest("GET", "/", nil)
w := httptest.NewRecorder()
http.HandlerFunc(viewerHandler).ServeHTTP(w, r)
if w.Code != http.StatusOK {
t.Errorf("Viewer page returned %v != %v on GET", w.Code, http.StatusOK)
}
// Test GET request on not found page
r, _ = http.NewRequest("GET", "", nil)
w = httptest.NewRecorder()
http.HandlerFunc(viewerHandler).ServeHTTP(w, r)
if w.Code != http.StatusNotFound {
t.Errorf("Viewer page returned %v != %v on GET", w.Code, http.StatusOK)
}
// Test GET request on statistics page
r, _ = http.NewRequest("GET", "/_stats/demo/", nil)
w = httptest.NewRecorder()
http.HandlerFunc(statisticsHandler).ServeHTTP(w, r)
if w.Code != http.StatusOK {
t.Errorf("Viewer page returned %v != %v on GET", w.Code, http.StatusOK)
}
}