From c4e549a738a5e28e351cb43d9523620f3dc4b4dd Mon Sep 17 00:00:00 2001
From: Denis Galeev <dengaleev@gmail.com>
Date: Mon, 31 Mar 2025 17:44:15 +0300
Subject: [PATCH] fix test
---
render/data.go | 8 +++++++-
render/render_test.go | 2 +-
2 files changed, 8 insertions(+), 2 deletions(-)
diff --git a/render/data.go b/render/data.go
index a653ea30..2c0ad5e3 100644
--- a/render/data.go
+++ b/render/data.go
@@ -4,7 +4,10 @@
package render
-import "net/http"
+import (
+ "net/http"
+ "strconv"
+)
// Data contains ContentType and bytes data.
type Data struct {
@@ -15,6 +18,9 @@ type Data struct {
// Render (Data) writes data with custom ContentType.
func (r Data) Render(w http.ResponseWriter) (err error) {
r.WriteContentType(w)
+ if len(r.Data) > 0 {
+ w.Header().Set("Content-Length", strconv.Itoa(len(r.Data)))
+ }
_, err = w.Write(r.Data)
return
}
diff --git a/render/render_test.go b/render/render_test.go
index e10726d3..dc26c512 100644
--- a/render/render_test.go
+++ b/render/render_test.go
@@ -421,7 +421,7 @@ func TestRenderData(t *testing.T) {
require.NoError(t, err)
assert.Equal(t, "#!PNG some raw data", w.Body.String())
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) {