mirror of
https://github.com/1f349/violet.git
synced 2024-09-16 17:06:48 +01:00
56 lines
1.4 KiB
Go
56 lines
1.4 KiB
Go
package favicons
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"database/sql"
|
|
_ "embed"
|
|
"github.com/1f349/violet"
|
|
"github.com/1f349/violet/database"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
"github.com/stretchr/testify/assert"
|
|
"image/png"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
//go:embed example.svg
|
|
exampleSvg []byte
|
|
//go:embed example.png
|
|
examplePng []byte
|
|
//go:embed example.ico
|
|
exampleIco []byte
|
|
)
|
|
|
|
func TestFaviconsNew(t *testing.T) {
|
|
getFaviconViaRequest = func(_ string) ([]byte, error) { return exampleSvg, nil }
|
|
|
|
db, err := violet.InitDB("file:TestFaviconsNew?mode=memory&cache=shared")
|
|
assert.NoError(t, err)
|
|
|
|
favicons := New(db, "inkscape")
|
|
err = db.UpdateFaviconCache(context.Background(), database.UpdateFaviconCacheParams{
|
|
Host: "example.com",
|
|
Svg: sql.NullString{
|
|
String: "https://example.com/assets/logo.svg",
|
|
Valid: true,
|
|
},
|
|
})
|
|
assert.NoError(t, err)
|
|
favicons.cLock.Lock()
|
|
assert.NoError(t, favicons.internalCompile(favicons.faviconMap))
|
|
favicons.cLock.Unlock()
|
|
|
|
icons := favicons.GetIcons("example.com")
|
|
assert.Equal(t, "https://example.com/assets/logo.svg", icons.Svg.Url)
|
|
|
|
assert.Equal(t, "74cdc17d0502a690941799c327d9ca1ed042e76c784def43a42937f2eed270b4", icons.Svg.Hash)
|
|
assert.NotEqual(t, "", icons.Png.Hash)
|
|
assert.NotEqual(t, "", icons.Ico.Hash)
|
|
|
|
// verify png bytes are a valid png image
|
|
pngRaw := bytes.NewBuffer(icons.Png.Raw)
|
|
_, err = png.Decode(pngRaw)
|
|
assert.NoError(t, err)
|
|
}
|