62 lines
1.7 KiB
Go
62 lines
1.7 KiB
Go
// Copyright 2023 PingCAP, Inc. Licensed under Apache-2.0.
|
|
|
|
package storage_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
backup "github.com/pingcap/kvproto/pkg/brpb"
|
|
"github.com/pingcap/tidb/br/pkg/storage"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func createMockStorage(t *testing.T) (storage.ExternalStorage, string) {
|
|
tempdir := t.TempDir()
|
|
storage, err := storage.New(context.Background(), &backup.StorageBackend{
|
|
Backend: &backup.StorageBackend_Local{
|
|
Local: &backup.Local{
|
|
Path: tempdir,
|
|
},
|
|
},
|
|
}, nil)
|
|
require.NoError(t, err)
|
|
return storage, tempdir
|
|
}
|
|
|
|
func requireFileExists(t *testing.T, path string) {
|
|
_, err := os.Stat(path)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func requireFileNotExists(t *testing.T, path string) {
|
|
_, err := os.Stat(path)
|
|
require.True(t, os.IsNotExist(err))
|
|
}
|
|
|
|
func TestTryLockRemote(t *testing.T) {
|
|
ctx := context.Background()
|
|
strg, pth := createMockStorage(t)
|
|
err := storage.TryLockRemote(ctx, strg, "test.lock", "This file is mine!")
|
|
require.NoError(t, err)
|
|
requireFileExists(t, filepath.Join(pth, "test.lock"))
|
|
err = storage.UnlockRemote(ctx, strg, "test.lock")
|
|
require.NoError(t, err)
|
|
requireFileNotExists(t, filepath.Join(pth, "test.lock"))
|
|
}
|
|
|
|
func TestConflictLock(t *testing.T) {
|
|
ctx := context.Background()
|
|
strg, pth := createMockStorage(t)
|
|
err := storage.TryLockRemote(ctx, strg, "test.lock", "This file is mine!")
|
|
require.NoError(t, err)
|
|
err = storage.TryLockRemote(ctx, strg, "test.lock", "This file is mine!")
|
|
require.ErrorContains(t, err, "locked, meta = Locked")
|
|
requireFileExists(t, filepath.Join(pth, "test.lock"))
|
|
err = storage.UnlockRemote(ctx, strg, "test.lock")
|
|
require.NoError(t, err)
|
|
requireFileNotExists(t, filepath.Join(pth, "test.lock"))
|
|
}
|