status-go/server/server_pairing_test.go

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, nil)
		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.pem.Received())
			s.Require().Equal(s.PS.payload.pem.received.encrypted, c.payload.pem.toSend.encrypted)
			s.Require().Nil(s.PS.payload.pem.ToSend())
			s.Require().Nil(c.payload.pem.Received())
		case Sending:
			s.Require().Equal(c.payload.pem.Received(), data)
			s.Require().Equal(c.payload.pem.received.encrypted, s.PS.payload.pem.toSend.encrypted)
			s.Require().Nil(c.payload.pem.ToSend())
			s.Require().Nil(s.PS.payload.pem.Received())
		}

		// Reset the server's PayloadEncryptionManager
		s.PS.payload.pem.ResetPayload()
	}
}