131 lines
4.5 KiB
Go
131 lines
4.5 KiB
Go
package coordinatorproto
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"github.com/anytypeio/any-sync/util/crypto"
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/stretchr/testify/require"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type fixture struct {
|
|
coordinatorKey crypto.PrivKey
|
|
accountKey crypto.PubKey
|
|
accountIdentity []byte
|
|
ctx context.Context
|
|
originalReceipt *SpaceReceipt
|
|
signedReceipt *SpaceReceiptWithSignature
|
|
spaceId string
|
|
peerId string
|
|
}
|
|
|
|
func newFixture(t *testing.T) *fixture {
|
|
coordinatorKey, _, err := crypto.GenerateEd25519Key(rand.Reader)
|
|
require.NoError(t, err)
|
|
signKey, _, err := crypto.GenerateEd25519Key(rand.Reader)
|
|
require.NoError(t, err)
|
|
signKeyRaw, err := signKey.GetPublic().Raw()
|
|
require.NoError(t, err)
|
|
return &fixture{
|
|
spaceId: "spaceId",
|
|
peerId: "peerId",
|
|
accountIdentity: signKeyRaw,
|
|
coordinatorKey: coordinatorKey,
|
|
accountKey: signKey.GetPublic(),
|
|
}
|
|
}
|
|
|
|
func (fx *fixture) prepareReceipt(t *testing.T, validPeriod time.Duration) {
|
|
var err error
|
|
fx.signedReceipt, err = PrepareSpaceReceipt(fx.spaceId, fx.peerId, validPeriod, fx.accountKey, fx.coordinatorKey)
|
|
require.NoError(t, err)
|
|
fx.originalReceipt = &SpaceReceipt{}
|
|
err = proto.Unmarshal(fx.signedReceipt.SpaceReceiptPayload, fx.originalReceipt)
|
|
require.NoError(t, err)
|
|
return
|
|
}
|
|
|
|
func (fx *fixture) updateReceipt(t *testing.T, update func(t *testing.T, receipt *SpaceReceipt)) {
|
|
update(t, fx.originalReceipt)
|
|
marshalled, err := proto.Marshal(fx.originalReceipt)
|
|
require.NoError(t, err)
|
|
signature, err := fx.coordinatorKey.Sign(marshalled)
|
|
require.NoError(t, err)
|
|
fx.signedReceipt = &SpaceReceiptWithSignature{
|
|
SpaceReceiptPayload: marshalled,
|
|
Signature: signature,
|
|
}
|
|
}
|
|
|
|
func TestReceiptValid(t *testing.T) {
|
|
fx := newFixture(t)
|
|
fx.prepareReceipt(t, time.Second)
|
|
err := CheckReceipt(fx.peerId, fx.spaceId, fx.accountIdentity, []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestReceiptIncorrectSpaceId(t *testing.T) {
|
|
fx := newFixture(t)
|
|
fx.prepareReceipt(t, time.Second)
|
|
err := CheckReceipt(fx.peerId, "otherId", fx.accountIdentity, []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.Error(t, errReceiptSpaceIdIncorrect, err)
|
|
}
|
|
|
|
func TestReceiptIncorrectPeerId(t *testing.T) {
|
|
fx := newFixture(t)
|
|
fx.prepareReceipt(t, time.Second)
|
|
err := CheckReceipt("otherId", fx.spaceId, fx.accountIdentity, []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.Error(t, errReceiptPeerIdIncorrect, err)
|
|
}
|
|
|
|
func TestReceiptIncorrectAccountIdentity(t *testing.T) {
|
|
fx := newFixture(t)
|
|
fx.prepareReceipt(t, time.Second)
|
|
err := CheckReceipt(fx.peerId, fx.spaceId, []byte("some identity"), []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.Error(t, errReceiptAccountIncorrect, err)
|
|
}
|
|
|
|
func TestReceiptIncorrectCoordinatorKey(t *testing.T) {
|
|
fx := newFixture(t)
|
|
fx.prepareReceipt(t, time.Second)
|
|
|
|
t.Run("random key payload", func(t *testing.T) {
|
|
fx.updateReceipt(t, func(t *testing.T, receipt *SpaceReceipt) {
|
|
receipt.AccountIdentity = []byte("some random stuff")
|
|
})
|
|
err := CheckReceipt(fx.peerId, fx.spaceId, fx.accountIdentity, []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.Error(t, err)
|
|
})
|
|
t.Run("random incorrect key", func(t *testing.T) {
|
|
fx.updateReceipt(t, func(t *testing.T, receipt *SpaceReceipt) {
|
|
randomKey, _, err := crypto.GenerateEd25519Key(rand.Reader)
|
|
require.NoError(t, err)
|
|
keyBytes, err := randomKey.GetPublic().Marshall()
|
|
require.NoError(t, err)
|
|
receipt.AccountIdentity = keyBytes
|
|
})
|
|
err := CheckReceipt(fx.peerId, fx.spaceId, fx.accountIdentity, []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.Error(t, errNoSuchCoordinatorNode, err)
|
|
})
|
|
}
|
|
|
|
func TestReceiptIncorrectSignature(t *testing.T) {
|
|
fx := newFixture(t)
|
|
fx.prepareReceipt(t, time.Second)
|
|
fx.signedReceipt.Signature = []byte("random sig")
|
|
err := CheckReceipt(fx.peerId, fx.spaceId, fx.accountIdentity, []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.Error(t, errReceiptSignatureIncorrect, err)
|
|
}
|
|
|
|
func TestReceiptExpired(t *testing.T) {
|
|
fx := newFixture(t)
|
|
fx.prepareReceipt(t, time.Second)
|
|
fx.updateReceipt(t, func(t *testing.T, receipt *SpaceReceipt) {
|
|
receipt.ValidUntil = uint64(time.Now().Add(-time.Second).Unix())
|
|
})
|
|
err := CheckReceipt(fx.peerId, fx.spaceId, fx.accountIdentity, []string{fx.coordinatorKey.GetPublic().PeerId()}, fx.signedReceipt)
|
|
require.Error(t, errReceiptExpired, err)
|
|
}
|