test: add tests for response.go

response_test.go

@@ -0,0 +1,102 @@
+package htmx
+
+import (
+	"html/template"
+	"net/http"
+	"testing"
+)
+
+func TestWrite(t *testing.T) {
+	w := newMockResponseWriter()
+
+	err := NewResponse().
+		StatusCode(StatusStopPolling).
+		Location("/profiles").
+		Redirect("/pull").
+		PushURL("/push").
+		Refresh(true).
+		ReplaceURL("/water").
+		Retarget("#world").
+		Reselect("#hello").
+		AddTrigger(Trigger("myEvent")).
+		Reswap(SwapInnerHTML.ShowOn("#swappy", Top)).
+		Write(w)
+	if err != nil {
+		t.Errorf("an error occurred writing a response: %v", err)
+	}
+
+	if w.statusCode != StatusStopPolling {
+		t.Errorf("wrong error code. want=%v, got=%v", StatusStopPolling, w.statusCode)
+	}
+
+	expectedHeaders := map[string]string{
+		HeaderTrigger:    "myEvent",
+		HeaderLocation:   "/profiles",
+		HeaderRedirect:   "/pull",
+		HeaderPushURL:    "/push",
+		HeaderRefresh:    "true",
+		HeaderReplaceUrl: "/water",
+		HeaderRetarget:   "#world",
+		HeaderReselect:   "#hello",
+		HeaderReswap:     "innerHTML show:#swappy:top",
+	}
+
+	for k, v := range expectedHeaders {
+		got := w.header.Get(k)
+		if got != v {
+			t.Errorf("wrong value for header %q. got=%q, want=%q", k, got, v)
+		}
+	}
+}
+
+func TestRenderHTML(t *testing.T) {
+	text := `hello world!`
+
+	w := newMockResponseWriter()
+
+	_, err := NewResponse().Location("/conversation/message").RenderHTML(w, template.HTML(text))
+	if err != nil {
+		t.Errorf("an error occurred writing HTML: %v", err)
+	}
+
+	if got, want := w.Header().Get(HeaderLocation), "/conversation/message"; got != want {
+		t.Errorf("wrong value for header %q. got=%q, want=%q", HeaderLocation, got, want)
+	}
+
+	if string(w.body) != text {
+		t.Errorf("wrong response body. got=%q, want=%q", string(w.body), text)
+	}
+}
+
+func TestMustRenderHTML(t *testing.T) {
+	text := `hello world!`
+
+	w := newMockResponseWriter()
+
+	NewResponse().MustRenderHTML(w, template.HTML(text))
+}
+
+type mockResponseWriter struct {
+	body       []byte
+	statusCode int
+	header     http.Header
+}
+
+func newMockResponseWriter() *mockResponseWriter {
+	return &mockResponseWriter{
+		header: http.Header{},
+	}
+}
+
+func (mrw *mockResponseWriter) Header() http.Header {
+	return mrw.header
+}
+
+func (mrw *mockResponseWriter) Write(b []byte) (int, error) {
+	mrw.body = append(mrw.body, b...)
+	return 0, nil
+}
+
+func (mrw *mockResponseWriter) WriteHeader(statusCode int) {
+	mrw.statusCode = statusCode
+}