mediaCompression_test.go

package main

import (
	"crypto/sha256"
	"fmt"
	"io"
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
)

func Test_compress(t *testing.T) {
	fakeFileContent := "Test"
	hash := sha256.New()
	_, _ = io.WriteString(hash, fakeFileContent)
	fakeSha256 := fmt.Sprintf("%x", hash.Sum(nil))

	var uf mediaStorageSaveFunc = func(filename string, f io.Reader) (location string, err error) {
		return "https://example.com/" + filename, nil
	}

	t.Run("Cloudflare", func(t *testing.T) {
		fakeClient := newFakeHttpClient()
		fakeClient.setHandler(http.HandlerFunc(func(rw http.ResponseWriter, r *http.Request) {
			assert.Equal(t, "https://www.cloudflare.com/cdn-cgi/image/f=jpeg,q=75,metadata=none,fit=scale-down,w=2000,h=3000/https://example.com/original.jpg", r.URL.String())

			rw.WriteHeader(http.StatusOK)
			_, _ = io.WriteString(rw, fakeFileContent)
		}))

		cf := &cloudflare{}
		res, err := cf.compress("https://example.com/original.jpg", uf, fakeClient.Client)

		assert.Nil(t, err)
		assert.Equal(t, "https://example.com/"+fakeSha256+".jpeg", res)
	})
}