Added tests for content types

2024-03-28 15:51:47 +01:00 · 2024-03-28 15:51:47 +01:00 · b95ce5002e
commit b95ce5002e
parent d7a323262f
1 changed files with 67 additions and 0 deletions
--- a/content/content_test.go
+++ b/content/content_test.go
@ -0,0 +1,67 @@
+package content_test
+
+import (
+	"testing"
+
+	"git.akyoto.dev/go/assert"
+	"git.akyoto.dev/go/web"
+	"git.akyoto.dev/go/web/content"
+)
+
+func TestContentTypes(t *testing.T) {
+	s := web.NewServer()
+
+	s.Get("/css", func(ctx web.Context) error {
+		return content.CSS(ctx, "body{}")
+	})
+
+	s.Get("/csv", func(ctx web.Context) error {
+		return content.CSV(ctx, "ID;Name\n")
+	})
+
+	s.Get("/html", func(ctx web.Context) error {
+		return content.HTML(ctx, "<html></html>")
+	})
+
+	s.Get("/js", func(ctx web.Context) error {
+		return content.JS(ctx, "console.log(42)")
+	})
+
+	s.Get("/json", func(ctx web.Context) error {
+		return content.JSON(ctx, struct{ Name string }{Name: "User 1"})
+	})
+
+	s.Get("/text", func(ctx web.Context) error {
+		return content.Text(ctx, "Hello")
+	})
+
+	s.Get("/xml", func(ctx web.Context) error {
+		return content.XML(ctx, "<xml></xml>")
+	})
+
+	tests := []struct {
+		Method      string
+		URL         string
+		Body        string
+		Status      int
+		Response    string
+		ContentType string
+	}{
+		{Method: "GET", URL: "/css", Status: 200, Response: "body{}", ContentType: "text/css"},
+		{Method: "GET", URL: "/csv", Status: 200, Response: "ID;Name\n", ContentType: "text/csv"},
+		{Method: "GET", URL: "/html", Status: 200, Response: "<html></html>", ContentType: "text/html"},
+		{Method: "GET", URL: "/js", Status: 200, Response: "console.log(42)", ContentType: "text/javascript"},
+		{Method: "GET", URL: "/json", Status: 200, Response: "{\"Name\":\"User 1\"}\n", ContentType: "application/json"},
+		{Method: "GET", URL: "/text", Status: 200, Response: "Hello", ContentType: "text/plain"},
+		{Method: "GET", URL: "/xml", Status: 200, Response: "<xml></xml>", ContentType: "text/xml"},
+	}
+
+	for _, test := range tests {
+		t.Run(test.URL, func(t *testing.T) {
+			response := s.Request(test.Method, "http://example.com"+test.URL, nil, nil)
+			assert.Equal(t, response.Status(), test.Status)
+			assert.Equal(t, response.Header("Content-Type"), test.ContentType)
+			assert.Equal(t, string(response.Body()), test.Response)
+		})
+	}
+}