[#222] Add tests for shards
Signed-off-by: Alex Vanin <alexey@nspcc.ru>remotes/KirillovDenis/release/v0.21.1
Alex Vanin
parent
ce92df3543
commit
6bec50963b
|
|
@ -0,0 +1,82 @@
|
|||
package shard_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/shard"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestShard_Delete(t *testing.T) {
|
||||
sh := newShard(t, false)
|
||||
shWC := newShard(t, true)
|
||||
|
||||
defer func() {
|
||||
releaseShard(sh, t)
|
||||
releaseShard(shWC, t)
|
||||
}()
|
||||
|
||||
t.Run("without write cache", func(t *testing.T) {
|
||||
testShardDelete(t, sh)
|
||||
})
|
||||
|
||||
t.Run("with write cache", func(t *testing.T) {
|
||||
testShardDelete(t, shWC)
|
||||
})
|
||||
}
|
||||
|
||||
func testShardDelete(t *testing.T, sh *shard.Shard) {
|
||||
cid := generateCID()
|
||||
|
||||
obj := generateRawObjectWithCID(t, cid)
|
||||
addAttribute(obj, "foo", "bar")
|
||||
|
||||
putPrm := new(shard.PutPrm)
|
||||
getPrm := new(shard.GetPrm)
|
||||
|
||||
t.Run("big object", func(t *testing.T) {
|
||||
addPayload(obj, 1<<20)
|
||||
|
||||
putPrm.WithObject(obj.Object())
|
||||
getPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
delPrm := new(shard.DeletePrm)
|
||||
delPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
_, err := sh.Put(putPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Get(getPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Delete(delPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Get(getPrm)
|
||||
require.EqualError(t, err, object.ErrNotFound.Error())
|
||||
})
|
||||
|
||||
t.Run("small object", func(t *testing.T) {
|
||||
obj.SetID(generateOID())
|
||||
addPayload(obj, 1<<5)
|
||||
|
||||
putPrm.WithObject(obj.Object())
|
||||
getPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
delPrm := new(shard.DeletePrm)
|
||||
delPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
_, err := sh.Put(putPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Get(getPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Delete(delPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Get(getPrm)
|
||||
require.EqualError(t, err, object.ErrNotFound.Error())
|
||||
})
|
||||
}
|
||||
|
|
@ -0,0 +1,65 @@
|
|||
package shard_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/shard"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestShard_Get(t *testing.T) {
|
||||
sh := newShard(t, false)
|
||||
shWC := newShard(t, true)
|
||||
|
||||
defer func() {
|
||||
releaseShard(sh, t)
|
||||
releaseShard(shWC, t)
|
||||
}()
|
||||
|
||||
t.Run("without write cache", func(t *testing.T) {
|
||||
testShardGet(t, sh)
|
||||
})
|
||||
|
||||
t.Run("with write cache", func(t *testing.T) {
|
||||
testShardGet(t, shWC)
|
||||
})
|
||||
}
|
||||
|
||||
func testShardGet(t *testing.T, sh *shard.Shard) {
|
||||
obj := generateRawObject(t)
|
||||
addAttribute(obj, "foo", "bar")
|
||||
|
||||
putPrm := new(shard.PutPrm)
|
||||
getPrm := new(shard.GetPrm)
|
||||
|
||||
t.Run("small object", func(t *testing.T) {
|
||||
addPayload(obj, 1<<5)
|
||||
|
||||
putPrm.WithObject(obj.Object())
|
||||
|
||||
_, err := sh.Put(putPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
getPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
res, err := sh.Get(getPrm)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, obj.Object(), res.Object())
|
||||
})
|
||||
|
||||
t.Run("big object", func(t *testing.T) {
|
||||
obj.SetID(generateOID())
|
||||
addPayload(obj, 1<<20) // big obj
|
||||
|
||||
putPrm.WithObject(obj.Object())
|
||||
|
||||
_, err := sh.Put(putPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
getPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
res, err := sh.Get(getPrm)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, obj.Object(), res.Object())
|
||||
})
|
||||
}
|
||||
|
|
@ -0,0 +1,44 @@
|
|||
package shard_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/shard"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestShard_Head(t *testing.T) {
|
||||
sh := newShard(t, false)
|
||||
shWC := newShard(t, true)
|
||||
|
||||
defer func() {
|
||||
releaseShard(sh, t)
|
||||
releaseShard(shWC, t)
|
||||
}()
|
||||
|
||||
t.Run("without write cache", func(t *testing.T) {
|
||||
testShardHead(t, sh)
|
||||
})
|
||||
|
||||
t.Run("with write cache", func(t *testing.T) {
|
||||
testShardHead(t, shWC)
|
||||
})
|
||||
}
|
||||
|
||||
func testShardHead(t *testing.T, sh *shard.Shard) {
|
||||
obj := generateRawObject(t)
|
||||
addAttribute(obj, "foo", "bar")
|
||||
|
||||
putPrm := new(shard.PutPrm)
|
||||
putPrm.WithObject(obj.Object())
|
||||
|
||||
_, err := sh.Put(putPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
headPrm := new(shard.HeadPrm)
|
||||
headPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
res, err := sh.Head(headPrm)
|
||||
require.NoError(t, err)
|
||||
require.Equal(t, obj.Object(), res.Object())
|
||||
}
|
||||
|
|
@ -0,0 +1,57 @@
|
|||
package shard_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/shard"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestShard_Inhume(t *testing.T) {
|
||||
sh := newShard(t, false)
|
||||
shWC := newShard(t, true)
|
||||
|
||||
defer func() {
|
||||
releaseShard(sh, t)
|
||||
releaseShard(shWC, t)
|
||||
}()
|
||||
|
||||
t.Run("without write cache", func(t *testing.T) {
|
||||
testShardInhume(t, sh)
|
||||
})
|
||||
|
||||
t.Run("with write cache", func(t *testing.T) {
|
||||
testShardInhume(t, shWC)
|
||||
})
|
||||
}
|
||||
|
||||
func testShardInhume(t *testing.T, sh *shard.Shard) {
|
||||
cid := generateCID()
|
||||
|
||||
obj := generateRawObjectWithCID(t, cid)
|
||||
addAttribute(obj, "foo", "bar")
|
||||
|
||||
ts := generateRawObjectWithCID(t, cid)
|
||||
|
||||
putPrm := new(shard.PutPrm)
|
||||
putPrm.WithObject(obj.Object())
|
||||
|
||||
inhPrm := new(shard.InhumePrm)
|
||||
inhPrm.WithTarget(obj.Object().Address(), ts.Object().Address())
|
||||
|
||||
getPrm := new(shard.GetPrm)
|
||||
getPrm.WithAddress(obj.Object().Address())
|
||||
|
||||
_, err := sh.Put(putPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Get(getPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Inhume(inhPrm)
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sh.Get(getPrm)
|
||||
require.EqualError(t, err, object.ErrAlreadyRemoved.Error())
|
||||
}
|
||||
|
|
@ -0,0 +1,129 @@
|
|||
package shard_test
|
||||
|
||||
import (
|
||||
"crypto/rand"
|
||||
"crypto/sha256"
|
||||
"os"
|
||||
"path"
|
||||
"testing"
|
||||
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg"
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg/container"
|
||||
objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
|
||||
"github.com/nspcc-dev/neofs-api-go/pkg/owner"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/blobstor"
|
||||
meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase/v2"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/local_object_storage/shard"
|
||||
"github.com/nspcc-dev/neofs-node/pkg/util/test"
|
||||
"github.com/nspcc-dev/tzhash/tz"
|
||||
"github.com/stretchr/testify/require"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
func newShard(t testing.TB, enableWriteCache bool) *shard.Shard {
|
||||
rootPath := t.Name()
|
||||
if enableWriteCache {
|
||||
rootPath = path.Join(rootPath, "wc")
|
||||
} else {
|
||||
rootPath = path.Join(rootPath, "nowc")
|
||||
}
|
||||
|
||||
opts := []shard.Option{
|
||||
shard.WithLogger(zap.L()),
|
||||
shard.WithBlobStorOptions(
|
||||
blobstor.WithRootPath(path.Join(rootPath, "blob")),
|
||||
blobstor.WithBlobovniczaShallowWidth(2),
|
||||
blobstor.WithBlobovniczaShallowDepth(2),
|
||||
),
|
||||
shard.WithMetaBaseOptions(
|
||||
meta.WithPath(path.Join(rootPath, "meta")),
|
||||
),
|
||||
shard.WithWriteCache(enableWriteCache),
|
||||
shard.WithWriteCacheOptions(
|
||||
blobstor.WithRootPath(path.Join(rootPath, "wcache")),
|
||||
),
|
||||
}
|
||||
|
||||
sh := shard.New(opts...)
|
||||
|
||||
require.NoError(t, sh.Open())
|
||||
require.NoError(t, sh.Init())
|
||||
|
||||
return sh
|
||||
}
|
||||
|
||||
func releaseShard(s *shard.Shard, t testing.TB) {
|
||||
s.Close()
|
||||
os.RemoveAll(t.Name())
|
||||
}
|
||||
|
||||
func generateRawObject(t *testing.T) *object.RawObject {
|
||||
return generateRawObjectWithCID(t, generateCID())
|
||||
}
|
||||
|
||||
func generateRawObjectWithCID(t *testing.T, cid *container.ID) *object.RawObject {
|
||||
version := pkg.NewVersion()
|
||||
version.SetMajor(2)
|
||||
version.SetMinor(1)
|
||||
|
||||
w, err := owner.NEO3WalletFromPublicKey(&test.DecodeKey(-1).PublicKey)
|
||||
require.NoError(t, err)
|
||||
|
||||
ownerID := owner.NewID()
|
||||
ownerID.SetNeo3Wallet(w)
|
||||
|
||||
csum := new(pkg.Checksum)
|
||||
csum.SetSHA256(sha256.Sum256(w.Bytes()))
|
||||
|
||||
csumTZ := new(pkg.Checksum)
|
||||
csumTZ.SetTillichZemor(tz.Sum(csum.Sum()))
|
||||
|
||||
obj := object.NewRaw()
|
||||
obj.SetID(generateOID())
|
||||
obj.SetOwnerID(ownerID)
|
||||
obj.SetContainerID(cid)
|
||||
obj.SetVersion(version)
|
||||
obj.SetPayloadChecksum(csum)
|
||||
obj.SetPayloadHomomorphicHash(csumTZ)
|
||||
|
||||
return obj
|
||||
}
|
||||
|
||||
func addAttribute(obj *object.RawObject, key, val string) {
|
||||
attr := objectSDK.NewAttribute()
|
||||
attr.SetKey(key)
|
||||
attr.SetValue(val)
|
||||
|
||||
attrs := obj.Attributes()
|
||||
attrs = append(attrs, attr)
|
||||
obj.SetAttributes(attrs...)
|
||||
}
|
||||
|
||||
func addPayload(obj *object.RawObject, size int) {
|
||||
buf := make([]byte, size)
|
||||
_, _ = rand.Read(buf)
|
||||
|
||||
obj.SetPayload(buf)
|
||||
obj.SetPayloadSize(uint64(size))
|
||||
}
|
||||
|
||||
func generateCID() *container.ID {
|
||||
cs := [sha256.Size]byte{}
|
||||
_, _ = rand.Read(cs[:])
|
||||
|
||||
id := container.NewID()
|
||||
id.SetSHA256(cs)
|
||||
|
||||
return id
|
||||
}
|
||||
|
||||
func generateOID() *objectSDK.ID {
|
||||
cs := [sha256.Size]byte{}
|
||||
_, _ = rand.Read(cs[:])
|
||||
|
||||
id := objectSDK.NewID()
|
||||
id.SetSHA256(cs)
|
||||
|
||||
return id
|
||||
}
|
||||
Loading…
Reference in New Issue