frostfs-sdk-go/pool/cache_test.go
Marina Biryukova 560cbbd1f1 [#234] pool: Update token expiration check in cache
Signed-off-by: Marina Biryukova <m.biryukova@yadro.com>
2024-07-05 12:36:17 +03:00

37 lines
841 B
Go

package pool
import (
"testing"
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/session"
sessiontest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/session/test"
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
"github.com/stretchr/testify/require"
)
func TestSessionCache_GetUnmodifiedToken(t *testing.T) {
const key = "Foo"
target := *sessiontest.Object()
pk, err := keys.NewPrivateKey()
require.NoError(t, err)
check := func(t *testing.T, tok session.Object, extra string) {
require.False(t, tok.VerifySignature(), extra)
}
cache, err := newCache(0)
require.NoError(t, err)
cache.Put(key, target)
value, ok := cache.Get(key)
require.True(t, ok)
check(t, value, "before sign")
err = value.Sign(pk.PrivateKey)
require.NoError(t, err)
value, ok = cache.Get(key)
require.True(t, ok)
check(t, value, "after sign")
}