68 lines
2.1 KiB
Go
68 lines
2.1 KiB
Go
// Package storagetest provides shared adapter checks. Use a dedicated test
|
|
// bucket/root and a Store configured with a 64-byte object limit.
|
|
package storagetest
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"git.yuntuo.hk/k2safe/go-omni/storage"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func Run(t *testing.T, store storage.Store) {
|
|
t.Helper()
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
key := "contract/文件 +%.txt"
|
|
defer store.Delete(ctx, key)
|
|
if _, err := store.Get(ctx, key); !errors.Is(err, storage.ErrNotFound) {
|
|
t.Fatal("missing object", err)
|
|
}
|
|
for _, body := range []string{"original", "replacement"} {
|
|
info, err := store.Put(ctx, key, strings.NewReader(body), int64(len(body)), "text/plain")
|
|
if err != nil || info.Size != int64(len(body)) {
|
|
t.Fatal(info, err)
|
|
}
|
|
object, err := store.Get(ctx, key)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data, err := io.ReadAll(object.Body)
|
|
object.Body.Close()
|
|
if err != nil || string(data) != body || object.ContentType != "text/plain" {
|
|
t.Fatal(string(data), object.Info, err)
|
|
}
|
|
}
|
|
if _, err := store.Put(ctx, "tooBig", strings.NewReader(strings.Repeat("a", 65)), 65, "text/plain"); !errors.Is(err, storage.ErrTooLarge) {
|
|
t.Fatal(err)
|
|
}
|
|
for _, size := range []int64{0, 2, 4} {
|
|
if _, err := store.Put(ctx, "sizeMismatch", strings.NewReader("abc"), size, "text/plain"); err == nil {
|
|
t.Fatalf("declared size %d accepted for 3 bytes", size)
|
|
}
|
|
}
|
|
defer store.Delete(context.Background(), "sizeMismatch")
|
|
for _, key := range []string{"../bad", "a/../b", "/absolute", "a//b", "a\\b"} {
|
|
if _, err := store.Put(ctx, key, strings.NewReader("a"), 1, "text/plain"); err == nil {
|
|
t.Fatal(key)
|
|
}
|
|
}
|
|
canceled, cancelRequest := context.WithCancel(ctx)
|
|
cancelRequest()
|
|
if _, err := store.Get(canceled, key); !errors.Is(err, context.Canceled) {
|
|
t.Fatal(err)
|
|
}
|
|
if err := store.Delete(ctx, key); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := store.Delete(ctx, key); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := store.Get(ctx, key); !errors.Is(err, storage.ErrNotFound) {
|
|
t.Fatal(err)
|
|
}
|
|
}
|