Renamed content to send

2024-04-02 15:23:44 +02:00
parent b28a9f12ce
commit f0cb179b8b
2 changed files with 10 additions and 10 deletions
--- a/send/send.go
+++ b/send/send.go
@ -0,0 +1,49 @@
+package send
+
+import (
+	"encoding/json"
+
+	"git.akyoto.dev/go/web"
+)
+
+// CSS sends the body with the content type set to `text/css`.
+func CSS(ctx web.Context, body string) error {
+	ctx.Response().SetHeader("Content-Type", "text/css")
+	return ctx.String(body)
+}
+
+// CSV sends the body with the content type set to `text/csv`.
+func CSV(ctx web.Context, body string) error {
+	ctx.Response().SetHeader("Content-Type", "text/csv")
+	return ctx.String(body)
+}
+
+// HTML sends the body with the content type set to `text/html`.
+func HTML(ctx web.Context, body string) error {
+	ctx.Response().SetHeader("Content-Type", "text/html")
+	return ctx.String(body)
+}
+
+// JS sends the body with the content type set to `text/javascript`.
+func JS(ctx web.Context, body string) error {
+	ctx.Response().SetHeader("Content-Type", "text/javascript")
+	return ctx.String(body)
+}
+
+// JSON encodes the object in JSON format and sends it with the content type set to `application/json`.
+func JSON(ctx web.Context, object any) error {
+	ctx.Response().SetHeader("Content-Type", "application/json")
+	return json.NewEncoder(ctx.Response()).Encode(object)
+}
+
+// Text sends the body with the content type set to `text/plain`.
+func Text(ctx web.Context, body string) error {
+	ctx.Response().SetHeader("Content-Type", "text/plain")
+	return ctx.String(body)
+}
+
+// XML sends the body with the content type set to `text/xml`.
+func XML(ctx web.Context, body string) error {
+	ctx.Response().SetHeader("Content-Type", "text/xml")
+	return ctx.String(body)
+}
--- a/send/send_test.go
+++ b/send/send_test.go
@ -0,0 +1,67 @@
+package send_test
+
+import (
+	"testing"
+
+	"git.akyoto.dev/go/assert"
+	"git.akyoto.dev/go/web"
+	"git.akyoto.dev/go/web/send"
+)
+
+func TestContentTypes(t *testing.T) {
+	s := web.NewServer()
+
+	s.Get("/css", func(ctx web.Context) error {
+		return send.CSS(ctx, "body{}")
+	})
+
+	s.Get("/csv", func(ctx web.Context) error {
+		return send.CSV(ctx, "ID;Name\n")
+	})
+
+	s.Get("/html", func(ctx web.Context) error {
+		return send.HTML(ctx, "<html></html>")
+	})
+
+	s.Get("/js", func(ctx web.Context) error {
+		return send.JS(ctx, "console.log(42)")
+	})
+
+	s.Get("/json", func(ctx web.Context) error {
+		return send.JSON(ctx, struct{ Name string }{Name: "User 1"})
+	})
+
+	s.Get("/text", func(ctx web.Context) error {
+		return send.Text(ctx, "Hello")
+	})
+
+	s.Get("/xml", func(ctx web.Context) error {
+		return send.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)
+		})
+	}
+}