frostfs-node/pkg/services/object/util/key_test.go

package util_test

import (
	"context"
	"crypto/elliptic"
	"testing"

	"github.com/google/uuid"
	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
	"github.com/nspcc-dev/neofs-api-go/pkg/session"
	sessionV2 "github.com/nspcc-dev/neofs-api-go/v2/session"
	"github.com/nspcc-dev/neofs-node/pkg/services/object/util"
	tokenStorage "github.com/nspcc-dev/neofs-node/pkg/services/session/storage"
	"github.com/stretchr/testify/require"
)

func TestNewKeyStorage(t *testing.T) {
	nodeKey, err := keys.NewPrivateKey()
	require.NoError(t, err)

	tokenStor := tokenStorage.New()
	stor := util.NewKeyStorage(&nodeKey.PrivateKey, tokenStor, mockedNetworkState{42})

	t.Run("node key", func(t *testing.T) {
		key, err := stor.GetKey(nil)
		require.NoError(t, err)
		require.Equal(t, nodeKey.PrivateKey, *key)
	})

	t.Run("unknown token", func(t *testing.T) {
		tok := generateToken(t)
		_, err = stor.GetKey(tok)
		require.Error(t, err)
	})

	t.Run("known token", func(t *testing.T) {
		tok := createToken(t, tokenStor, 100)
		pubKey, err := keys.NewPublicKeyFromBytes(tok.SessionKey(), elliptic.P256())
		require.NoError(t, err)

		key, err := stor.GetKey(tok)
		require.NoError(t, err)
		require.Equal(t, pubKey.X, key.PublicKey.X)
		require.Equal(t, pubKey.Y, key.PublicKey.Y)
	})

	t.Run("expired token", func(t *testing.T) {
		tok := createToken(t, tokenStor, 30)
		_, err := stor.GetKey(tok)
		require.Error(t, err)
	})
}

func generateToken(t *testing.T) *session.Token {
	key, err := keys.NewPrivateKey()
	require.NoError(t, err)

	pubKey := key.PublicKey().Bytes()
	id, err := uuid.New().MarshalBinary()
	require.NoError(t, err)

	tok := session.NewToken()
	tok.SetSessionKey(pubKey)
	tok.SetID(id)

	return tok
}

func createToken(t *testing.T, store *tokenStorage.TokenStore, exp uint64) *session.Token {
	req := new(sessionV2.CreateRequestBody)
	req.SetOwnerID(nil)
	req.SetExpiration(exp)

	resp, err := store.Create(context.Background(), req)
	require.NoError(t, err)

	tok := session.NewToken()
	tok.SetSessionKey(resp.GetSessionKey())
	tok.SetID(resp.GetID())

	return tok
}

type mockedNetworkState struct {
	value uint64
}

func (m mockedNetworkState) CurrentEpoch() uint64 {
	return m.value
}