42 lines
990 B
Go
42 lines
990 B
Go
|
package storage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
type fakeDeleter struct {
|
||
|
|
deleted []string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *fakeDeleter) DeleteObject(_ context.Context, key string) error {
|
||
|
|
f.deleted = append(f.deleted, key)
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPurgePublishAttachment(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
deleter := &fakeDeleter{}
|
||
|
|
|
||
|
|
if err := PurgePublishAttachment(ctx, deleter, ""); err != nil {
|
||
|
|
t.Fatalf("empty key: %v", err)
|
||
|
|
}
|
||
|
|
if len(deleter.deleted) != 0 {
|
||
|
|
t.Fatalf("expected no delete for empty key")
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := PurgePublishAttachment(ctx, deleter, "avatars/foo.png"); err != nil {
|
||
|
|
t.Fatalf("invalid key: %v", err)
|
||
|
|
}
|
||
|
|
if len(deleter.deleted) != 0 {
|
||
|
|
t.Fatalf("expected no delete for non-publish key")
|
||
|
|
}
|
||
|
|
|
||
|
|
key := "publish-attachments/t1/a1/test.png"
|
||
|
|
if err := PurgePublishAttachment(ctx, deleter, key); err != nil {
|
||
|
|
t.Fatalf("purge: %v", err)
|
||
|
|
}
|
||
|
|
if len(deleter.deleted) != 1 || deleter.deleted[0] != key {
|
||
|
|
t.Fatalf("deleted = %v, want [%s]", deleter.deleted, key)
|
||
|
|
}
|
||
|
|
}
|