forked from TrueCloudLab/frostfs-api-go
140 lines
2.7 KiB
Go
140 lines
2.7 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neofs-api-go/refs"
|
|
"github.com/nspcc-dev/neofs-crypto/test"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func (m TestRequest) SignedData() ([]byte, error) {
|
|
return SignedDataFromReader(m)
|
|
}
|
|
|
|
func (m TestRequest) SignedDataSize() (sz int) {
|
|
sz += 4
|
|
|
|
sz += len(m.StringField)
|
|
|
|
sz += len(m.BytesField)
|
|
|
|
sz += m.CustomField.Size()
|
|
|
|
return
|
|
}
|
|
|
|
func (m TestRequest) ReadSignedData(p []byte) (int, error) {
|
|
if len(p) < m.SignedDataSize() {
|
|
return 0, io.ErrUnexpectedEOF
|
|
}
|
|
|
|
var off int
|
|
|
|
binary.BigEndian.PutUint32(p[off:], uint32(m.IntField))
|
|
off += 4
|
|
|
|
off += copy(p[off:], []byte(m.StringField))
|
|
|
|
off += copy(p[off:], m.BytesField)
|
|
|
|
n, err := m.CustomField.MarshalTo(p[off:])
|
|
off += n
|
|
|
|
return off, err
|
|
}
|
|
|
|
func BenchmarkSignDataWithSessionToken(b *testing.B) {
|
|
key := test.DecodeKey(0)
|
|
|
|
customField := testCustomField{1, 2, 3, 4, 5, 6, 7, 8}
|
|
|
|
token := new(Token)
|
|
|
|
req := &TestRequest{
|
|
IntField: math.MaxInt32,
|
|
StringField: "TestRequestStringField",
|
|
BytesField: make([]byte, 1<<22),
|
|
CustomField: &customField,
|
|
}
|
|
|
|
req.SetTTL(math.MaxInt32 - 8)
|
|
req.SetEpoch(math.MaxInt64 - 12)
|
|
req.SetToken(token)
|
|
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
require.NoError(b, SignRequestData(key, req))
|
|
}
|
|
}
|
|
|
|
func BenchmarkVerifyAccumulatedSignaturesWithToken(b *testing.B) {
|
|
customField := testCustomField{1, 2, 3, 4, 5, 6, 7, 8}
|
|
|
|
token := new(Token)
|
|
|
|
req := &TestRequest{
|
|
IntField: math.MaxInt32,
|
|
StringField: "TestRequestStringField",
|
|
BytesField: make([]byte, 1<<22),
|
|
CustomField: &customField,
|
|
}
|
|
|
|
req.SetTTL(math.MaxInt32 - 8)
|
|
req.SetEpoch(math.MaxInt64 - 12)
|
|
req.SetToken(token)
|
|
|
|
for i := 0; i < 10; i++ {
|
|
key := test.DecodeKey(i)
|
|
require.NoError(b, SignRequestData(key, req))
|
|
}
|
|
|
|
b.ResetTimer()
|
|
b.ReportAllocs()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
require.NoError(b, VerifyRequestData(req))
|
|
}
|
|
}
|
|
|
|
func TestRequestVerificationHeader_SetToken(t *testing.T) {
|
|
id, err := refs.NewUUID()
|
|
require.NoError(t, err)
|
|
|
|
token := new(Token)
|
|
token.SetID(id)
|
|
|
|
h := new(RequestVerificationHeader)
|
|
|
|
h.SetToken(token)
|
|
|
|
require.Equal(t, token, h.GetToken())
|
|
}
|
|
|
|
func TestRequestVerificationHeader_SetBearer(t *testing.T) {
|
|
aclRules := []byte{1, 2, 3}
|
|
|
|
token := new(BearerTokenMsg)
|
|
token.SetACLRules(aclRules)
|
|
|
|
h := new(RequestVerificationHeader)
|
|
|
|
h.SetBearer(token)
|
|
|
|
require.Equal(t, token, h.GetBearer())
|
|
}
|
|
|
|
func TestRequestVerificationHeader_GetBearerToken(t *testing.T) {
|
|
s := new(RequestVerificationHeader)
|
|
|
|
require.Nil(t, s.GetBearerToken())
|
|
|
|
bearer := new(BearerTokenMsg)
|
|
s.SetBearer(bearer)
|
|
require.Equal(t, wrapBearerTokenMsg(bearer), s.GetBearerToken())
|
|
}
|