mirror of https://github.com/jlelse/GoBlog
39 lines
1.0 KiB
Go
39 lines
1.0 KiB
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)
|
|
})
|
|
}
|