mirror of
https://github.com/gofiber/storage.git
synced 2025-09-28 05:12:19 +08:00
138 lines
2.6 KiB
Go
138 lines
2.6 KiB
Go
package azureblob
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Azure/azure-sdk-for-go/sdk/storage/azblob/bloberror"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func newStore() *Storage {
|
|
return New(Config{
|
|
Account: "devstoreaccount1",
|
|
Container: "test",
|
|
Endpoint: "http://127.0.0.1:10000/devstoreaccount1",
|
|
Credentials: Credentials{
|
|
Account: "devstoreaccount1",
|
|
Key: "Eby8vdM02xNOcqFlqUwJPLlmEtlCDXJ1OUzFT50uSRZ6IFsuFq2UVErCz4I6tq/K1SZFPTOtr/KBHBeksoGMGw==",
|
|
},
|
|
})
|
|
}
|
|
|
|
func Test_AzureBlob_Get(t *testing.T) {
|
|
var (
|
|
key = "john"
|
|
val = []byte("doe")
|
|
)
|
|
testStore := newStore()
|
|
|
|
err := testStore.Set(key, val, 0)
|
|
require.NoError(t, err)
|
|
|
|
result, err := testStore.Get(key)
|
|
require.NoError(t, err)
|
|
require.Equal(t, val, result)
|
|
}
|
|
|
|
func Test_AzureBlob_Set(t *testing.T) {
|
|
var (
|
|
key = "john"
|
|
val = []byte("doe")
|
|
)
|
|
|
|
testStore := newStore()
|
|
err := testStore.Set(key, val, 0)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_AzureBlob_Delete(t *testing.T) {
|
|
var (
|
|
key = "john"
|
|
val = []byte("doe")
|
|
)
|
|
testStore := newStore()
|
|
|
|
err := testStore.Set(key, val, 0)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.Delete(key)
|
|
require.NoError(t, err)
|
|
|
|
result, err := testStore.Get(key)
|
|
if err != nil {
|
|
if bloberror.HasCode(err, bloberror.BlobNotFound) {
|
|
err = nil
|
|
}
|
|
}
|
|
require.NoError(t, err)
|
|
require.Zero(t, len(result))
|
|
}
|
|
|
|
func Test_AzureBlob_Override(t *testing.T) {
|
|
var (
|
|
key = "john"
|
|
val = []byte("doe")
|
|
)
|
|
testStore := newStore()
|
|
|
|
err := testStore.Set(key, val, 0)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.Set(key, val, 0)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func Test_AzureBlob_Get_NotExist(t *testing.T) {
|
|
testStore := newStore()
|
|
result, err := testStore.Get("notexist")
|
|
if err != nil {
|
|
if bloberror.HasCode(err, bloberror.BlobNotFound) {
|
|
err = nil
|
|
}
|
|
}
|
|
require.NoError(t, err)
|
|
require.Zero(t, len(result))
|
|
}
|
|
|
|
func Test_AzureBlob_Reset(t *testing.T) {
|
|
val := []byte("doe")
|
|
testStore := newStore()
|
|
|
|
err := testStore.Set("john1", val, 0)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.Set("john2", val, 0)
|
|
require.NoError(t, err)
|
|
|
|
err = testStore.Reset()
|
|
require.NoError(t, err)
|
|
|
|
result, err := testStore.Get("john1")
|
|
if err != nil {
|
|
if bloberror.HasCode(err, bloberror.BlobNotFound) {
|
|
err = nil
|
|
}
|
|
}
|
|
require.NoError(t, err)
|
|
require.Zero(t, len(result))
|
|
|
|
result, err = testStore.Get("john2")
|
|
if err != nil {
|
|
if bloberror.HasCode(err, bloberror.BlobNotFound) {
|
|
err = nil
|
|
}
|
|
}
|
|
require.NoError(t, err)
|
|
require.Zero(t, len(result))
|
|
}
|
|
|
|
func Test_S3_Conn(t *testing.T) {
|
|
testStore := newStore()
|
|
require.True(t, testStore.Conn() != nil)
|
|
}
|
|
|
|
func Test_AzureBlob_Close(t *testing.T) {
|
|
testStore := newStore()
|
|
require.Nil(t, testStore.Close())
|
|
}
|