mirror of
https://github.com/gofiber/storage.git
synced 2025-11-01 04:02:44 +08:00
88 lines
1.7 KiB
Go
88 lines
1.7 KiB
Go
package s3
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"testing"
|
|
|
|
"github.com/aws/aws-sdk-go-v2/service/s3/types"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_S3_CreateDeleteBucket(t *testing.T) {
|
|
bkt := "test-new-bucket"
|
|
|
|
testStore := newTestStore(t)
|
|
defer testStore.Close()
|
|
|
|
err := testStore.CreateBucket(bkt)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.DeleteBucket(bkt)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_S3_DeleteMany(t *testing.T) {
|
|
val := []byte("doe")
|
|
|
|
testStore := newTestStore(t)
|
|
defer testStore.Close()
|
|
|
|
err := testStore.Set("john1", val, 0)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.Set("john2", val, 0)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.Set("john3", val, 0)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.DeleteMany("john1", "john2")
|
|
require.NoError(t, err)
|
|
|
|
result, err := testStore.Get("john1")
|
|
require.NoError(t, err)
|
|
require.Zero(t, len(result))
|
|
|
|
result, err = testStore.Get("john2")
|
|
require.NoError(t, err)
|
|
require.Zero(t, len(result))
|
|
|
|
result, err = testStore.Get("john3")
|
|
require.NoError(t, err)
|
|
require.Equal(t, val, result)
|
|
}
|
|
|
|
func Test_S3_SetWithChecksum(t *testing.T) {
|
|
var (
|
|
key = "set-with-checksum"
|
|
val = []byte("doe")
|
|
)
|
|
|
|
// Create SHA-256 hash and get checksum.
|
|
sha256Hash := sha256.New()
|
|
sha256Hash.Write(val)
|
|
sha256sum := sha256Hash.Sum(nil)
|
|
|
|
checksum := map[types.ChecksumAlgorithm][]byte{
|
|
types.ChecksumAlgorithmSha256: sha256sum,
|
|
}
|
|
|
|
testStore := newTestStore(t)
|
|
defer testStore.Close()
|
|
|
|
err := testStore.SetWithChecksum(key, val, checksum)
|
|
require.NoError(t, err)
|
|
|
|
result, err := testStore.Get(key)
|
|
require.NoError(t, err)
|
|
|
|
// Compare value.
|
|
require.Equal(t, result, val)
|
|
|
|
// Compare checksum.
|
|
hash2 := sha256.New()
|
|
hash2.Write(result)
|
|
sha256sum2 := hash2.Sum(nil)
|
|
require.Equal(t, sha256sum, sha256sum2)
|
|
}
|