status-go/server/server_pairing_test.go

88 lines
1.8 KiB
Go
Raw Normal View History

package server
import (
"crypto/rand"
"testing"
"time"
"github.com/stretchr/testify/suite"
)
func TestPairingServerSuite(t *testing.T) {
suite.Run(t, new(PairingServerSuite))
}
type PairingServerSuite struct {
suite.Suite
TestPairingServerComponents
}
func (s *PairingServerSuite) SetupSuite() {
s.SetupPairingServerComponents(s.T())
}
func (s *PairingServerSuite) TestPairingServer_StartPairing() {
modes := []Mode{
Receiving,
Sending,
}
for _, m := range modes {
s.PS.mode = m
// Random payload
data := make([]byte, 32)
_, err := rand.Read(data)
s.Require().NoError(err)
if m == Sending {
err := s.PS.MountPayload(data)
s.Require().NoError(err)
}
err = s.PS.StartPairing()
s.Require().NoError(err)
// Give time for the sever to be ready, hacky I know, I'll iron this out
time.Sleep(10 * time.Millisecond)
cp, err := s.PS.MakeConnectionParams()
s.Require().NoError(err)
qr, err := cp.ToString()
s.Require().NoError(err)
// Client reads QR code and parses the connection string
ccp := new(ConnectionParams)
err = ccp.FromString(qr)
s.Require().NoError(err)
c, err := NewPairingClient(ccp)
s.Require().NoError(err)
if m == Receiving {
err := c.MountPayload(data)
s.Require().NoError(err)
}
err = c.PairAccount()
s.Require().NoError(err)
switch m {
case Receiving:
s.Require().Equal(data, s.PS.payload.Received())
s.Require().Equal(s.PS.payload.received.encrypted, c.payload.toSend.encrypted)
s.Require().Nil(s.PS.payload.ToSend())
s.Require().Nil(c.payload.Received())
case Sending:
s.Require().Equal(c.payload.Received(), data)
s.Require().Equal(c.payload.received.encrypted, s.PS.payload.toSend.encrypted)
s.Require().Nil(c.payload.ToSend())
s.Require().Nil(s.PS.payload.Received())
}
// Reset the server's PayloadManager
s.PS.payload.ResetPayload()
}
}