forked from TrueCloudLab/frostfs-node
2271944064
Cleanup is more idiomatic way to clean test data Signed-off-by: Alex Vanin <alexey@nspcc.ru>
108 lines
2.5 KiB
Go
108 lines
2.5 KiB
Go
package meta_test
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"math/rand"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/pkg"
|
|
cid "github.com/nspcc-dev/neofs-api-go/pkg/container/id"
|
|
cidtest "github.com/nspcc-dev/neofs-api-go/pkg/container/id/test"
|
|
objectSDK "github.com/nspcc-dev/neofs-api-go/pkg/object"
|
|
"github.com/nspcc-dev/neofs-api-go/pkg/owner"
|
|
ownertest "github.com/nspcc-dev/neofs-api-go/pkg/owner/test"
|
|
"github.com/nspcc-dev/neofs-node/pkg/core/object"
|
|
meta "github.com/nspcc-dev/neofs-node/pkg/local_object_storage/metabase"
|
|
"github.com/nspcc-dev/neofs-node/pkg/util/test"
|
|
"github.com/nspcc-dev/tzhash/tz"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// saves "big" object in DB.
|
|
func putBig(db *meta.DB, obj *object.Object) error {
|
|
return meta.Put(db, obj, nil)
|
|
}
|
|
|
|
func testSelect(t *testing.T, db *meta.DB, cid *cid.ID, fs objectSDK.SearchFilters, exp ...*objectSDK.Address) {
|
|
res, err := meta.Select(db, cid, fs)
|
|
require.NoError(t, err)
|
|
require.Len(t, res, len(exp))
|
|
|
|
for i := range exp {
|
|
require.Contains(t, res, exp[i])
|
|
}
|
|
}
|
|
|
|
func testOID() *objectSDK.ID {
|
|
cs := [sha256.Size]byte{}
|
|
_, _ = rand.Read(cs[:])
|
|
|
|
id := objectSDK.NewID()
|
|
id.SetSHA256(cs)
|
|
|
|
return id
|
|
}
|
|
|
|
func newDB(t testing.TB) *meta.DB {
|
|
path := t.Name()
|
|
|
|
bdb := meta.New(meta.WithPath(path), meta.WithPermissions(0600))
|
|
|
|
require.NoError(t, bdb.Open())
|
|
|
|
t.Cleanup(func() {
|
|
bdb.Close()
|
|
os.Remove(bdb.DumpInfo().Path)
|
|
})
|
|
|
|
return bdb
|
|
}
|
|
|
|
func generateRawObject(t *testing.T) *object.RawObject {
|
|
return generateRawObjectWithCID(t, cidtest.Generate())
|
|
}
|
|
|
|
func generateRawObjectWithCID(t *testing.T, cid *cid.ID) *object.RawObject {
|
|
version := pkg.NewVersion()
|
|
version.SetMajor(2)
|
|
version.SetMinor(1)
|
|
|
|
w, err := owner.NEO3WalletFromPublicKey(&test.DecodeKey(-1).PublicKey)
|
|
require.NoError(t, err)
|
|
|
|
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(testOID())
|
|
obj.SetOwnerID(ownertest.Generate())
|
|
obj.SetContainerID(cid)
|
|
obj.SetVersion(version)
|
|
obj.SetPayloadChecksum(csum)
|
|
obj.SetPayloadHomomorphicHash(csumTZ)
|
|
obj.SetPayload([]byte{1, 2, 3, 4, 5})
|
|
|
|
return obj
|
|
}
|
|
|
|
func generateAddress() *objectSDK.Address {
|
|
addr := objectSDK.NewAddress()
|
|
addr.SetContainerID(cidtest.Generate())
|
|
addr.SetObjectID(testOID())
|
|
|
|
return addr
|
|
}
|
|
|
|
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...)
|
|
}
|