mirror of
https://github.com/imgproxy/imgproxy.git
synced 2024-11-24 08:12:38 +02:00
145 lines
3.6 KiB
Go
145 lines
3.6 KiB
Go
package etag
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/imgproxy/imgproxy/v3/config"
|
|
"github.com/imgproxy/imgproxy/v3/imagedata"
|
|
"github.com/imgproxy/imgproxy/v3/options"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
var (
|
|
po = options.NewProcessingOptions()
|
|
|
|
imgWithETag = imagedata.ImageData{
|
|
Data: []byte("Hello Test"),
|
|
Headers: map[string]string{"ETag": `"loremipsumdolor"`},
|
|
}
|
|
imgWithoutETag = imagedata.ImageData{
|
|
Data: []byte("Hello Test"),
|
|
}
|
|
|
|
etagReq = `"yj0WO6sFU4GCciYUBWjzvvfqrBh869doeOC2Pp5EI1Y/RImxvcmVtaXBzdW1kb2xvciI"`
|
|
etagData = `"yj0WO6sFU4GCciYUBWjzvvfqrBh869doeOC2Pp5EI1Y/DvyChhMNu_sFX7jrjoyrgQbnFwfoOVv7kzp_Fbs6hQBg"`
|
|
)
|
|
|
|
type EtagTestSuite struct {
|
|
suite.Suite
|
|
|
|
h Handler
|
|
}
|
|
|
|
func (s *EtagTestSuite) SetupSuite() {
|
|
logrus.SetOutput(ioutil.Discard)
|
|
}
|
|
|
|
func (s *EtagTestSuite) TeardownSuite() {
|
|
logrus.SetOutput(os.Stdout)
|
|
}
|
|
|
|
func (s *EtagTestSuite) SetupTest() {
|
|
s.h = Handler{}
|
|
config.Reset()
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestGenerateActualReq() {
|
|
s.h.SetActualProcessingOptions(po)
|
|
s.h.SetActualImageData(&imgWithETag)
|
|
|
|
assert.Equal(s.T(), etagReq, s.h.GenerateActualETag())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestGenerateActualData() {
|
|
s.h.SetActualProcessingOptions(po)
|
|
s.h.SetActualImageData(&imgWithoutETag)
|
|
|
|
assert.Equal(s.T(), etagData, s.h.GenerateActualETag())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestGenerateExpectedReq() {
|
|
s.h.ParseExpectedETag(etagReq)
|
|
assert.Equal(s.T(), etagReq, s.h.GenerateExpectedETag())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestGenerateExpectedData() {
|
|
s.h.ParseExpectedETag(etagData)
|
|
assert.Equal(s.T(), etagData, s.h.GenerateExpectedETag())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestProcessingOptionsCheckSuccess() {
|
|
s.h.ParseExpectedETag(etagReq)
|
|
|
|
assert.True(s.T(), s.h.SetActualProcessingOptions(po))
|
|
assert.True(s.T(), s.h.ProcessingOptionsMatch())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestProcessingOptionsCheckFailure() {
|
|
i := strings.Index(etagReq, "/")
|
|
wrongEtag := `"wrongpohash` + etagReq[i:]
|
|
|
|
s.h.ParseExpectedETag(wrongEtag)
|
|
|
|
assert.False(s.T(), s.h.SetActualProcessingOptions(po))
|
|
assert.False(s.T(), s.h.ProcessingOptionsMatch())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestImageETagExpectedPresent() {
|
|
s.h.ParseExpectedETag(etagReq)
|
|
|
|
assert.Equal(s.T(), imgWithETag.Headers["ETag"], s.h.ImageEtagExpected())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestImageETagExpectedBlank() {
|
|
s.h.ParseExpectedETag(etagData)
|
|
|
|
assert.Empty(s.T(), s.h.ImageEtagExpected())
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestImageDataCheckDataToDataSuccess() {
|
|
s.h.ParseExpectedETag(etagData)
|
|
assert.True(s.T(), s.h.SetActualImageData(&imgWithoutETag))
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestImageDataCheckDataToDataFailure() {
|
|
i := strings.Index(etagData, "/")
|
|
wrongEtag := etagData[:i] + `/Dwrongimghash"`
|
|
|
|
s.h.ParseExpectedETag(wrongEtag)
|
|
assert.False(s.T(), s.h.SetActualImageData(&imgWithoutETag))
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestImageDataCheckDataToReqSuccess() {
|
|
s.h.ParseExpectedETag(etagData)
|
|
assert.True(s.T(), s.h.SetActualImageData(&imgWithETag))
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestImageDataCheckDataToReqFailure() {
|
|
i := strings.Index(etagData, "/")
|
|
wrongEtag := etagData[:i] + `/Dwrongimghash"`
|
|
|
|
s.h.ParseExpectedETag(wrongEtag)
|
|
assert.False(s.T(), s.h.SetActualImageData(&imgWithETag))
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestImageDataCheckReqToDataFailure() {
|
|
s.h.ParseExpectedETag(etagReq)
|
|
assert.False(s.T(), s.h.SetActualImageData(&imgWithoutETag))
|
|
}
|
|
|
|
func (s *EtagTestSuite) TestETagBusterFailure() {
|
|
config.ETagBuster = "busted"
|
|
|
|
s.h.ParseExpectedETag(etagReq)
|
|
assert.False(s.T(), s.h.SetActualImageData(&imgWithoutETag))
|
|
}
|
|
|
|
func TestEtag(t *testing.T) {
|
|
suite.Run(t, new(EtagTestSuite))
|
|
}
|