1package favicon
2
3import (
4 "bytes"
5 "context"
6 "errors"
7 "testing"
8
9 appport "github.com/bnema/dumber/internal/application/port"
10 domainfavicon "github.com/bnema/dumber/internal/domain/favicon"
11)
12
13func TestBlobStoreWriteReadOverwriteAndInvalidate(t *testing.T) {
14 ctx := context.Background()
15 store := NewBlobStore(t.TempDir())
16 key := domainfavicon.Key("example.com")
17
18 if err := store.WriteOriginal(ctx, key, []byte("ico1"), "image/x-icon"); err != nil {
19 t.Fatal(err)
20 }
21 if err := store.WriteOriginal(ctx, key, []byte("ico2"), "image/png"); err != nil {
22 t.Fatal(err)
23 }
24 got, ct, err := store.ReadOriginal(ctx, key)
25 if err != nil {
26 t.Fatal(err)
27 }
28 if string(got) != "ico2" || ct != "image/png" {
29 t.Fatalf("original = %q %q", got, ct)
30 }
31
32 if err = store.WritePNG(ctx, key, []byte("png")); err != nil {
33 t.Fatal(err)
34 }
35 if err = store.WriteSizedPNG(ctx, key, 32, []byte("sized")); err != nil {
36 t.Fatal(err)
37 }
38 if got, _, err = store.ReadPNG(ctx, key); err != nil || !bytes.Equal(got, []byte("png")) {
39 t.Fatalf("png = %q %v", got, err)
40 }
41 if got, _, err = store.ReadSizedPNG(ctx, key, 32); err != nil || !bytes.Equal(got, []byte("sized")) {
42 t.Fatalf("sized = %q %v", got, err)
43 }
44 if err = store.RemoveDerived(ctx, key); err != nil {
45 t.Fatal(err)
46 }
47 _, _, err = store.ReadPNG(ctx, key)
48 if !errors.Is(err, appport.ErrFaviconMiss) {
49 t.Fatalf("ReadPNG err = %v", err)
50 }
51 _, _, err = store.ReadSizedPNG(ctx, key, 32)
52 if !errors.Is(err, appport.ErrFaviconMiss) {
53 t.Fatalf("ReadSizedPNG err = %v", err)
54 }
55 if got, _, err = store.ReadOriginal(ctx, key); err != nil || string(got) != "ico2" {
56 t.Fatalf("original after invalidate = %q %v", got, err)
57 }
58}
59
60func TestBlobStoreRemoveDerivedNoopsWhenDiskCacheDisabled(t *testing.T) {
61 store := NewBlobStore("")
62 if err := store.RemoveDerived(context.Background(), "example.com"); err != nil {
63 t.Fatal(err)
64 }
65}
66
67func TestBlobStoreRemoveDerivedDoesNotRemoveSimilarDomainKeys(t *testing.T) {
68 ctx := context.Background()
69 store := NewBlobStore(t.TempDir())
70 if err := store.WriteSizedPNG(ctx, "example.com", 32, []byte("example")); err != nil {
71 t.Fatal(err)
72 }
73 if err := store.WritePNG(ctx, "example.com.au", []byte("other-png")); err != nil {
74 t.Fatal(err)
75 }
76 if err := store.WriteSizedPNG(ctx, "example.com.au", 32, []byte("other-sized")); err != nil {
77 t.Fatal(err)
78 }
79 if err := store.RemoveDerived(ctx, "example.com"); err != nil {
80 t.Fatal(err)
81 }
82 if _, _, err := store.ReadSizedPNG(ctx, "example.com", 32); !errors.Is(err, appport.ErrFaviconMiss) {
83 t.Fatalf("example.com sized err = %v", err)
84 }
85 if got, _, err := store.ReadPNG(ctx, "example.com.au"); err != nil || string(got) != "other-png" {
86 t.Fatalf("example.com.au png = %q, %v", got, err)
87 }
88 if got, _, err := store.ReadSizedPNG(ctx, "example.com.au", 32); err != nil || string(got) != "other-sized" {
89 t.Fatalf("example.com.au sized = %q, %v", got, err)
90 }
91}