forked from TrueCloudLab/frostfs-node
Dmitrii Stepanov
d62c6e4ce6
Add tracing spans for PUT requests. Add tracing spans for DELETE requests. Add tracing spans for SELECT requests. Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
package blobovniczatree
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/core/object"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/common"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/internal/blobstortest"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/util/logger"
|
|
"github.com/nspcc-dev/neo-go/pkg/util/slice"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap/zaptest"
|
|
)
|
|
|
|
func TestExistsInvalidStorageID(t *testing.T) {
|
|
dir := t.TempDir()
|
|
b := NewBlobovniczaTree(
|
|
WithLogger(&logger.Logger{Logger: zaptest.NewLogger(t)}),
|
|
WithObjectSizeLimit(1024),
|
|
WithBlobovniczaShallowWidth(2),
|
|
WithBlobovniczaShallowDepth(2),
|
|
WithRootPath(dir),
|
|
WithBlobovniczaSize(1<<20))
|
|
require.NoError(t, b.Open(false))
|
|
require.NoError(t, b.Init())
|
|
t.Cleanup(func() { _ = b.Close() })
|
|
|
|
obj := blobstortest.NewObject(1024)
|
|
addr := object.AddressOf(obj)
|
|
d, err := obj.Marshal()
|
|
require.NoError(t, err)
|
|
|
|
putRes, err := b.Put(context.Background(), common.PutPrm{Address: addr, RawData: d, DontCompress: true})
|
|
require.NoError(t, err)
|
|
|
|
t.Run("valid but wrong storage id", func(t *testing.T) {
|
|
// "0/X/Y" <-> "1/X/Y"
|
|
storageID := slice.Copy(putRes.StorageID)
|
|
if storageID[0] == '0' {
|
|
storageID[0]++
|
|
} else {
|
|
storageID[0]--
|
|
}
|
|
|
|
res, err := b.Exists(context.Background(), common.ExistsPrm{Address: addr, StorageID: storageID})
|
|
require.NoError(t, err)
|
|
require.False(t, res.Exists)
|
|
})
|
|
|
|
t.Run("invalid storage id", func(t *testing.T) {
|
|
storageID := slice.Copy(putRes.StorageID)
|
|
storageID[0] = '9'
|
|
|
|
// An invalid boltdb file is created so that it returns an error when opened
|
|
badFileDir := filepath.Join(dir, "9", "0")
|
|
require.NoError(t, os.MkdirAll(badFileDir, os.ModePerm))
|
|
require.NoError(t, os.WriteFile(filepath.Join(badFileDir, "0"), []byte("not a boltdb file content"), 0777))
|
|
|
|
res, err := b.Exists(context.Background(), common.ExistsPrm{Address: addr, StorageID: storageID})
|
|
require.Error(t, err)
|
|
require.False(t, res.Exists)
|
|
})
|
|
}
|