feat: support html function to sets content type and data

vicanso · Sep 3, 2021 · d407ab1 · d407ab1
1 parent 0cbbe5d
commit d407ab1
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/context.go b/context.go
@@ -634,6 +634,12 @@ func (c *Context) ReadFile(key string) ([]byte, *multipart.FileHeader, error) {
 	return buf, header, nil
 }
 
+// HTML sets content type and response body as html
+func (c *Context) HTML(html string) {
+	c.SetContentTypeByExt(".html")
+	c.BodyBuffer = bytes.NewBufferString(html)
+}
+
 // DisableReuse sets the context disable reuse
 func (c *Context) DisableReuse() {
 	atomic.StoreInt32(&c.reuseStatus, ReuseContextDisabled)

diff --git a/context_test.go b/context_test.go
@@ -545,6 +545,16 @@ func TestReadFile(t *testing.T) {
 	assert.Equal(testData, data)
 }
 
+func TestHTML(t *testing.T) {
+	assert := assert.New(t)
+
+	resp := httptest.NewRecorder()
+	c := NewContext(resp, nil)
+	html := "<html><body></body></html>"
+	c.HTML(html)
+	assert.Equal("text/html; charset=utf-8", resp.Header().Get(HeaderContentType))
+	assert.Equal(html, c.BodyBuffer.String())
+}
 
 func TestRedirect(t *testing.T) {
 	assert := assert.New(t)