realloc/frostfs-node
1
0
Fork 0
forked from TrueCloudLab/frostfs-node
Code Releases Packages Activity

[#943] service/object: Check session token expiration

Browse source 
Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
Alex Vanin 2021-10-25 15:19:50 +03:00 • committed by Alex Vanin
parent 508a28fdc0
commit 2fbdcbdee1
4 changed files with 36 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

16
pkg/services/object/util/key_test.go
View file

@ -19,7 +19,7 @@ func TestNewKeyStorage(t *testing.T) {
require.NoError(t, err)
tokenStor := tokenStorage.New()
stor := util.NewKeyStorage(&nodeKey.PrivateKey, tokenStor)
stor := util.NewKeyStorage(&nodeKey.PrivateKey, tokenStor, mockedNetworkState{42})
t.Run("node key", func(t *testing.T) {
key, err := stor.GetKey(nil)
@ -43,6 +43,12 @@ func TestNewKeyStorage(t *testing.T) {
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 {
@ -74,3 +80,11 @@ func createToken(t *testing.T, store *tokenStorage.TokenStore, exp uint64) *sess
return tok
}
type mockedNetworkState struct {
value uint64
}
func (m mockedNetworkState) CurrentEpoch() uint64 {
return m.value
}