blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-04-06 03:57:46 +08:00
Code Issues Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
Denis Galeev 2025-03-31 17:44:15 +03:00
parent 0fe68969cf
commit c4e549a738
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
render/data.go
View File

@ -4,7 +4,10 @@
package render package render
import "net/http" import (
"net/http"
"strconv"
)
// Data contains ContentType and bytes data. // Data contains ContentType and bytes data.
type Data struct { type Data struct {
@ -15,6 +18,9 @@ type Data struct {
// Render (Data) writes data with custom ContentType. // Render (Data) writes data with custom ContentType.
func (r Data) Render(w http.ResponseWriter) (err error) { func (r Data) Render(w http.ResponseWriter) (err error) {
r.WriteContentType(w) r.WriteContentType(w)
if len(r.Data) > 0 {
w.Header().Set("Content-Length", strconv.Itoa(len(r.Data)))
}
_, err = w.Write(r.Data) _, err = w.Write(r.Data)
return return
} }

2
render/render_test.go
View File

@ -421,7 +421,7 @@ func TestRenderData(t *testing.T) {
require.NoError(t, err) require.NoError(t, err)
assert.Equal(t, "#!PNG some raw data", w.Body.String()) assert.Equal(t, "#!PNG some raw data", w.Body.String())
assert.Equal(t, "image/png", w.Header().Get("Content-Type")) assert.Equal(t, "image/png", w.Header().Get("Content-Type"))
assert.Equal(t, "", w.Header().Get("Content-Length")) assert.Equal(t, "19", w.Header().Get("Content-Length"))
} }
func TestRenderDataContentLength(t *testing.T) { func TestRenderDataContentLength(t *testing.T) {