120 lines
3 KiB
Go
120 lines
3 KiB
Go
package eacl_test
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"testing"
|
|
|
|
refsgrpc "git.frostfs.info/TrueCloudLab/frostfs-api-go/v2/refs/grpc"
|
|
cidtest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/container/id/test"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/eacl"
|
|
eacltest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/eacl/test"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/version"
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTable(t *testing.T) {
|
|
v := version.NewVersion()
|
|
|
|
sha := sha256.Sum256([]byte("container id"))
|
|
id := cidtest.IDWithChecksum(sha)
|
|
|
|
v.SetMajor(3)
|
|
v.SetMinor(2)
|
|
|
|
table := eacl.NewTable()
|
|
table.SetVersion(v)
|
|
table.SetCID(id)
|
|
table.AddRecord(eacl.CreateRecord(eacl.ActionAllow, eacl.OperationPut))
|
|
|
|
v2 := table.ToV2()
|
|
require.NotNil(t, v2)
|
|
require.Equal(t, uint32(3), v2.GetVersion().GetMajor())
|
|
require.Equal(t, uint32(2), v2.GetVersion().GetMinor())
|
|
require.Equal(t, sha[:], v2.GetContainerId().GetValue())
|
|
require.Len(t, v2.GetRecords(), 1)
|
|
|
|
newTable := eacl.NewTableFromV2(v2)
|
|
require.True(t, proto.Equal(table.ToV2(), newTable.ToV2()))
|
|
|
|
t.Run("new from nil v2 table", func(t *testing.T) {
|
|
require.Equal(t, new(eacl.Table), eacl.NewTableFromV2(nil))
|
|
})
|
|
|
|
t.Run("create table", func(t *testing.T) {
|
|
id := cidtest.ID()
|
|
|
|
table := eacl.CreateTable(id)
|
|
cID, set := table.CID()
|
|
require.True(t, set)
|
|
require.Equal(t, id, cID)
|
|
require.Equal(t, version.Current(), table.Version())
|
|
})
|
|
}
|
|
|
|
func TestTable_AddRecord(t *testing.T) {
|
|
records := []eacl.Record{
|
|
*eacl.CreateRecord(eacl.ActionDeny, eacl.OperationDelete),
|
|
*eacl.CreateRecord(eacl.ActionAllow, eacl.OperationPut),
|
|
}
|
|
|
|
table := eacl.NewTable()
|
|
for _, record := range records {
|
|
table.AddRecord(&record)
|
|
}
|
|
|
|
require.Equal(t, records, table.Records())
|
|
}
|
|
|
|
func TestTableEncoding(t *testing.T) {
|
|
tab := eacltest.Table()
|
|
|
|
t.Run("binary", func(t *testing.T) {
|
|
data, err := tab.Marshal()
|
|
require.NoError(t, err)
|
|
|
|
tab2 := eacl.NewTable()
|
|
require.NoError(t, tab2.Unmarshal(data))
|
|
|
|
// FIXME: we compare v2 messages because
|
|
// Filter contains fmt.Stringer interface
|
|
require.Equal(t, tab.ToV2(), tab2.ToV2())
|
|
})
|
|
|
|
t.Run("json", func(t *testing.T) {
|
|
data, err := tab.MarshalJSON()
|
|
require.NoError(t, err)
|
|
|
|
tab2 := eacl.NewTable()
|
|
require.NoError(t, tab2.UnmarshalJSON(data))
|
|
|
|
require.Equal(t, tab.ToV2(), tab2.ToV2())
|
|
})
|
|
}
|
|
|
|
func TestTable_ToV2(t *testing.T) {
|
|
t.Run("nil", func(t *testing.T) {
|
|
var x *eacl.Table
|
|
|
|
require.Nil(t, x.ToV2())
|
|
})
|
|
|
|
t.Run("default values", func(t *testing.T) {
|
|
table := eacl.NewTable()
|
|
|
|
// check initial values
|
|
require.Equal(t, version.Current(), table.Version())
|
|
require.Nil(t, table.Records())
|
|
_, set := table.CID()
|
|
require.False(t, set)
|
|
|
|
// convert to v2 message
|
|
tableV2 := table.ToV2()
|
|
|
|
var verV2 refsgrpc.Version
|
|
version.Current().WriteToV2(&verV2)
|
|
require.True(t, proto.Equal(&verV2, tableV2.GetVersion()))
|
|
require.Nil(t, tableV2.GetRecords())
|
|
require.Nil(t, tableV2.GetContainerId())
|
|
})
|
|
}
|