package/storage
1
0
Fork 0
mirror of https://github.com/gofiber/storage.git synced 2025-10-04 08:16:36 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

azureblob: add support for context management

Browse Source
This commit is contained in:
Muhammed Efe Cetin
2025-02-11 15:27:15 +03:00
parent 1b443d036e
commit 2192ac6cfe
2 changed files with 122 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

77
azureblob/azureblob_test.go
View File

@@ -63,6 +63,23 @@ func Test_AzureBlob_Get(t *testing.T) {
require.Equal(t, val, result)
}
func Test_AzureBlob_GetWithContext(t *testing.T) {
var (
key = "john"
val = []byte("doe")
)
err := testStore.Set(key, val, 0)
require.NoError(t, err)
ctx, cancel := context.WithCancel(context.Background())
cancel()
result, err := testStore.GetWithContext(ctx, key)
require.ErrorIs(t, err, context.Canceled)
require.Zero(t, len(result))
}
func Test_AzureBlob_Set(t *testing.T) {
var (
key = "john"
@@ -76,6 +93,19 @@ func Test_AzureBlob_Set(t *testing.T) {
require.NoError(t, err)
}
func Test_AzureBlob_SetWithContext(t *testing.T) {
var (
key = "john"
val = []byte("doe")
)
ctx, cancel := context.WithCancel(context.Background())
cancel()
err := testStore.SetWithContext(ctx, key, val, 0)
require.ErrorIs(t, err, context.Canceled)
}
func Test_AzureBlob_Delete(t *testing.T) {
var (
key = "john"
@@ -101,6 +131,26 @@ func Test_AzureBlob_Delete(t *testing.T) {
require.Zero(t, len(result))
}
func Test_AzureBlob_DeleteWithContext(t *testing.T) {
var (
key = "john"
val = []byte("doe")
)
err := testStore.Set(key, val, 0)
require.NoError(t, err)
ctx, cancel := context.WithCancel(context.Background())
cancel()
err = testStore.DeleteWithContext(ctx, key)
require.ErrorIs(t, err, context.Canceled)
result, err := testStore.Get(key)
require.NoError(t, err)
require.Equal(t, val, result)
}
func Test_AzureBlob_Override(t *testing.T) {
var (
key = "john"
@@ -172,6 +222,33 @@ func Test_AzureBlob_Conn(t *testing.T) {
require.True(t, testStore.Conn() != nil)
}
func Test_AzureBlob_ResetWithContext(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)
ctx, cancel := context.WithCancel(context.Background())
cancel()
err = testStore.ResetWithContext(ctx)
require.ErrorIs(t, err, context.Canceled)
result, err := testStore.Get("john1")
require.NoError(t, err)
require.Equal(t, val, result)
result, err = testStore.Get("john2")
require.NoError(t, err)
require.Equal(t, val, result)
}
func Test_AzureBlob_Close(t *testing.T) {
testStore := newTestStore(t)
require.NoError(t, testStore.Close())