Initial commit

Initial public review release v0.10.0

lib/ir/storage_test.go

@@ -0,0 +1,101 @@
+package ir
+
+import (
+	"testing"
+
+	crypto "github.com/nspcc-dev/neofs-crypto"
+	"github.com/pkg/errors"
+	"github.com/stretchr/testify/require"
+)
+
+type testInfoReceiver struct {
+	keys [][]byte
+
+	err error
+}
+
+func (s testInfoReceiver) GetIRInfo(GetInfoParams) (*GetInfoResult, error) {
+	if s.err != nil {
+		return nil, s.err
+	}
+
+	nodes := make([]Node, 0, len(s.keys))
+
+	for i := range s.keys {
+		node := Node{}
+		node.SetKey(s.keys[i])
+
+		nodes = append(nodes, node)
+	}
+
+	info := Info{}
+	info.SetNodes(nodes)
+
+	res := new(GetInfoResult)
+	res.SetInfo(info)
+
+	return res, nil
+}
+
+func (s *testInfoReceiver) addKey(key []byte) {
+	s.keys = append(s.keys, key)
+}
+
+func TestGetInfoResult(t *testing.T) {
+	s := GetInfoResult{}
+
+	info := Info{}
+
+	n := Node{}
+	n.SetKey([]byte{1, 2, 3})
+
+	info.SetNodes([]Node{
+		n,
+	})
+
+	s.SetInfo(info)
+
+	require.Equal(t, info, s.Info())
+}
+
+func TestIsInnerRingKey(t *testing.T) {
+	var (
+		res bool
+		err error
+		s   = new(testInfoReceiver)
+	)
+
+	// empty public key
+	res, err = IsInnerRingKey(nil, nil)
+	require.EqualError(t, err, crypto.ErrEmptyPublicKey.Error())
+
+	key := []byte{1, 2, 3}
+
+	// nil Storage
+	res, err = IsInnerRingKey(nil, key)
+	require.EqualError(t, err, ErrNilStorage.Error())
+
+	// force Storage to return an error
+	s.err = errors.New("some error")
+
+	// Storage error
+	res, err = IsInnerRingKey(s, key)
+	require.EqualError(t, errors.Cause(err), s.err.Error())
+
+	// reset Storage error
+	s.err = nil
+
+	// IR keys don't contain key
+	s.addKey(append(key, 1))
+
+	res, err = IsInnerRingKey(s, key)
+	require.NoError(t, err)
+	require.False(t, res)
+
+	// IR keys contain key
+	s.addKey(key)
+
+	res, err = IsInnerRingKey(s, key)
+	require.NoError(t, err)
+	require.True(t, res)
+}