package formats_test
import (
"testing"
"gh.tarampamp.am/error-pages/v4/internal/formats"
"gh.tarampamp.am/error-pages/v4/internal/testutil/assert"
)
func TestFormat_ContentType(t *testing.T) {
t.Parallel()
for name, tt := range map[string]struct {
give formats.Format
want string
}{
"plain text": {give: formats.PlainTextFormat, want: "text/plain; charset=utf-8"},
"html": {give: formats.HTMLFormat, want: "text/html; charset=utf-8"},
"json": {give: formats.JSONFormat, want: "application/json; charset=utf-8"},
"xml": {give: formats.XMLFormat, want: "application/xml; charset=utf-8"},
"unknown": {give: formats.Format(255), want: ""},
} {
t.Run(name, func(t *testing.T) {
t.Parallel()
assert.Equal(t, tt.want, tt.give.ContentType())
})
}
}
func TestFormat_FormatError(t *testing.T) {
t.Parallel()
for name, tt := range map[string]struct {
giveFormat formats.Format
giveErr string
want string
}{
"plain/simple": {
giveFormat: formats.PlainTextFormat,
giveErr: "page not found",
want: "page not found",
},
"plain/empty": {
giveFormat: formats.PlainTextFormat,
giveErr: "",
want: "",
},
"plain/special chars passthrough": {
giveFormat: formats.PlainTextFormat,
giveErr: `error & "more"`,
want: `error & "more"`,
},
"json/simple": {
giveFormat: formats.JSONFormat,
giveErr: "page not found",
want: `{"error":"page not found"}`,
},
"json/empty": {
giveFormat: formats.JSONFormat,
giveErr: "",
want: `{"error":""}`,
},
"json/html chars escaped": {
// Go's json.Marshal escapes <, >, & to \uXXXX sequences to prevent XSS when JSON is embedded in HTML
giveFormat: formats.JSONFormat,
giveErr: `bold & "quoted"`,
want: "{\"error\":\"\\u003cb\\u003ebold\\u003c/b\\u003e \\u0026 \\\"quoted\\\"\"}",
},
"xml/simple": {
giveFormat: formats.XMLFormat,
giveErr: "page not found",
want: "\npage not found",
},
"xml/empty": {
giveFormat: formats.XMLFormat,
giveErr: "",
want: "\n",
},
"xml/special chars escaped": {
giveFormat: formats.XMLFormat,
giveErr: `error & 'it'`,
want: "\n<b>error</b> & 'it'",
},
"html/simple": {
giveFormat: formats.HTMLFormat,
giveErr: "page not found",
want: "\n\npage not found\n",
},
"html/empty": {
giveFormat: formats.HTMLFormat,
giveErr: "",
want: "\n\n\n",
},
"html/special chars escaped": {
giveFormat: formats.HTMLFormat,
giveErr: ``,
want: "\n\n" +
"<script>alert("xss")</script>\n",
},
"unknown format": {
giveFormat: formats.Format(255),
giveErr: "something went wrong",
want: "something went wrong",
},
} {
t.Run(name, func(t *testing.T) {
t.Parallel()
assert.Equal(t, tt.want, string(tt.giveFormat.FormatError(tt.giveErr)))
})
}
}