Add svg package tests

Author: DarthSim

processing_handler_test.go

@@ -324,29 +324,6 @@ func (s *ProcessingHandlerTestSuite) TestSkipProcessingSVG() {
 	require.True(s.T(), bytes.Equal(expected.Data, actual))
 }
 
-func (s *ProcessingHandlerTestSuite) TestPreserveOriginSVGHeaders() {
-	rw := s.send("/unsafe/rs:fill:4:4/plain/local:///test1.svg")
-	res := rw.Result()
-
-	require.Equal(s.T(), 200, res.StatusCode)
-
-	actual := s.readBody(res)
-	originHeaders := map[string]string{
-		"Content-Type":  "image/svg+xml",
-		"Cache-Control": "public, max-age=12345",
-	}
-
-	expected, err := svg.Sanitize(&imagedata.ImageData{
-		Data:    s.readTestFile("test1.svg"),
-		Headers: originHeaders,
-	})
-
-	require.Nil(s.T(), err)
-
-	require.True(s.T(), bytes.Equal(expected.Data, actual))
-	require.Equal(s.T(), originHeaders, expected.Headers)
-}
-
 func (s *ProcessingHandlerTestSuite) TestNotSkipProcessingSVGToJPG() {
 	rw := s.send("/unsafe/rs:fill:4:4/plain/local:///test1.svg@jpg")
 	res := rw.Result()

svg/svg.go

@@ -15,17 +15,17 @@ import (
 
 var feDropShadowName = []byte("feDropShadow")
 
-const feDropShadowTemplate = `
-	<feMerge result="dsin-%[1]s"><feMergeNode %[3]s /></feMerge>
-	<feGaussianBlur %[4]s />
-	<feOffset %[5]s result="dsof-%[2]s" />
-	<feFlood %[6]s />
-	<feComposite in2="dsof-%[2]s" operator="in" />
-	<feMerge %[7]s>
-		<feMergeNode />
-		<feMergeNode in="dsin-%[1]s" />
-	</feMerge>
-`
+var feDropShadowTemplate = strings.TrimSpace(`
+  <feMerge result="dsin-%[1]s"><feMergeNode %[3]s /></feMerge>
+  <feGaussianBlur %[4]s />
+  <feOffset %[5]s result="dsof-%[2]s" />
+  <feFlood %[6]s />
+  <feComposite in2="dsof-%[2]s" operator="in" />
+  <feMerge %[7]s>
+    <feMergeNode />
+    <feMergeNode in="dsin-%[1]s" />
+  </feMerge>
+`)
 
 func Sanitize(data *imagedata.ImageData) (*imagedata.ImageData, error) {
 	r := bytes.NewReader(data.Data)

svg/svg_test.go

@@ -0,0 +1,83 @@
+package svg
+
+import (
+	"os"
+	"path/filepath"
+	"regexp"
+	"testing"
+
+	"github.com/imgproxy/imgproxy/v3/config"
+	"github.com/imgproxy/imgproxy/v3/imagedata"
+	"github.com/imgproxy/imgproxy/v3/imagetype"
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/suite"
+)
+
+type SvgTestSuite struct {
+	suite.Suite
+}
+
+func (s *SvgTestSuite) SetupSuite() {
+	config.Reset()
+
+	err := imagedata.Init()
+	require.Nil(s.T(), err)
+}
+
+func (s *SvgTestSuite) readTestFile(name string) *imagedata.ImageData {
+	wd, err := os.Getwd()
+	require.Nil(s.T(), err)
+
+	data, err := os.ReadFile(filepath.Join(wd, "..", "testdata", name))
+	require.Nil(s.T(), err)
+
+	return &imagedata.ImageData{
+		Type: imagetype.SVG,
+		Data: data,
+		Headers: map[string]string{
+			"Content-Type":  "image/svg+xml",
+			"Cache-Control": "public, max-age=12345",
+		},
+	}
+}
+
+func (s *SvgTestSuite) TestSanitize() {
+	origin := s.readTestFile("test1.svg")
+	expected := s.readTestFile("test1.sanitized.svg")
+
+	actual, err := Sanitize(origin)
+
+	require.Nil(s.T(), err)
+	require.Equal(s.T(), string(expected.Data), string(actual.Data))
+	require.Equal(s.T(), origin.Headers, actual.Headers)
+}
+
+func (s *SvgTestSuite) TestFixUnsupportedDropShadow() {
+	origin := s.readTestFile("test1.drop-shadow.svg")
+	expected := s.readTestFile("test1.drop-shadow.fixed.svg")
+
+	actual, changed, err := FixUnsupported(origin)
+
+	// `FixUnsupported` generates random IDs, we need to replace them for the test
+	re := regexp.MustCompile(`"ds(in|of)-.+?"`)
+	actualData := re.ReplaceAllString(string(actual.Data), `"ds$1-test"`)
+
+	require.Nil(s.T(), err)
+	require.True(s.T(), changed)
+	require.Equal(s.T(), string(expected.Data), actualData)
+	require.Equal(s.T(), origin.Headers, actual.Headers)
+}
+
+func (s *SvgTestSuite) TestFixUnsupportedNothingChanged() {
+	origin := s.readTestFile("test1.svg")
+
+	actual, changed, err := FixUnsupported(origin)
+
+	require.Nil(s.T(), err)
+	require.False(s.T(), changed)
+	require.Equal(s.T(), origin, actual)
+}
+
+func TestSvg(t *testing.T) {
+	suite.Run(t, new(SvgTestSuite))
+}