fork from d7z-project/caddy-gitea-pages

pages/fake_resp.go

@@ -0,0 +1,53 @@
+package pages
+
+import (
+	"fmt"
+	"mime"
+	"net/http"
+	"path/filepath"
+	"strconv"
+)
+
+type FakeResponse struct {
+	*http.Response
+}
+
+func (r *FakeResponse) Length(len int) {
+	r.ContentLength = int64(len)
+	r.Header.Set("Content-Length", strconv.Itoa(len))
+}
+
+func (r *FakeResponse) CacheModeIgnore() {
+	r.CacheMode("SKIP")
+}
+func (r *FakeResponse) CacheModeMiss() {
+	r.CacheMode("MISS")
+}
+func (r *FakeResponse) CacheModeHit() {
+	r.CacheMode("HIT")
+}
+func (r *FakeResponse) CacheMode(mode string) {
+	r.SetHeader("Pages-Server-Cache", mode)
+}
+func (r *FakeResponse) ContentTypeExt(path string) {
+	r.ContentType(mime.TypeByExtension(filepath.Ext(path)))
+}
+func (r *FakeResponse) ContentType(types string) {
+	r.Header.Set("Content-Type", types)
+}
+func (r *FakeResponse) ETag(tag string) {
+	r.Header.Set("ETag", fmt.Sprintf("\"%s\"", tag))
+}
+
+func NewFakeResponse() *FakeResponse {
+	return &FakeResponse{
+		&http.Response{
+			StatusCode: http.StatusOK,
+			Header:     make(http.Header),
+		},
+	}
+}
+
+func (r *FakeResponse) SetHeader(key string, value string) {
+	r.Header.Set(key, value)
+}