status-go/server/qrops_test.go

121 lines
3.1 KiB
Go

package server
import (
"encoding/base64"
"io/ioutil"
"net/url"
"os"
"testing"
"time"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"
"github.com/status-im/status-go/images"
"github.com/status-im/status-go/multiaccounts"
"github.com/status-im/status-go/server/servertest"
)
type QROpsTestSuite struct {
suite.Suite
servertest.TestKeyComponents
servertest.TestLoggerComponents
server *MediaServer
serverNoPort *MediaServer
testStart time.Time
multiaccountsDB *multiaccounts.Database
}
var (
keyUID = "0xdeadbeef"
qrURL = "https://github.com/status-im/status-go/pull/3154"
)
func TestQROpsTestSuite(t *testing.T) {
suite.Run(t, new(QROpsTestSuite))
}
func (s *QROpsTestSuite) SetupTest() {
s.SetupKeyComponents(s.T())
s.SetupLoggerComponents()
mediaServer, err := NewMediaServer(nil, nil, nil)
s.Require().NoError(err)
s.server = mediaServer
err = s.server.SetPort(customPortForTests)
s.Require().NoError(err)
err = s.server.Start()
s.Require().NoError(err)
s.serverNoPort = &MediaServer{Server: Server{
hostname: DefaultIP.String(),
portManger: newPortManager(s.Logger, nil),
}}
go func() {
time.Sleep(waitTime)
s.serverNoPort.port = 80
}()
s.testStart = time.Now()
}
func seedTestDBWithIdentityImagesForQRCodeTests(s *QROpsTestSuite, db *multiaccounts.Database, keyUID string) {
iis := images.SampleIdentityImageForQRCode()
err := db.StoreIdentityImages(keyUID, iis, false)
s.Require().NoError(err)
}
// TestQROpsCodeWithoutSuperImposingLogo tests the QR code generation logic, it also allows us to debug in case
// things go wrong, here we compare generate a new QR code and compare its bytes with an already generated one.
func (s *QROpsTestSuite) TestQROpsCodeWithoutSuperImposingLogo() {
var params = url.Values{}
params.Set("url", base64.StdEncoding.EncodeToString([]byte(qrURL)))
params.Set("allowProfileImage", "false")
params.Set("level", "2")
params.Set("size", "200")
params.Set("imageName", "")
payload := generateQRBytes(params, s.Logger, s.multiaccountsDB)
expectedPayload, err := images.Asset("_assets/tests/qr/defaultQR.png")
s.Require().NoError(err)
s.Require().NotEmpty(payload)
require.Equal(s.T(), payload, expectedPayload)
}
func (s *QROpsTestSuite) TestQROpsCodeWithSuperImposingLogo() {
tmpfile, err := ioutil.TempFile("", "accounts-tests-")
s.Require().NoError(err)
db, err := multiaccounts.InitializeDB(tmpfile.Name())
s.Require().NoError(err)
seedTestDBWithIdentityImagesForQRCodeTests(s, db, keyUID)
var params = url.Values{}
params.Set("url", base64.StdEncoding.EncodeToString([]byte(qrURL)))
params.Set("allowProfileImage", "true")
params.Set("level", "2")
params.Set("size", "200")
params.Set("keyUid", keyUID)
params.Set("imageName", "large")
payload := generateQRBytes(params, s.Logger, db)
s.Require().NotEmpty(payload)
expectedPayload, err := images.Asset("_assets/tests/qr/QRWithLogo.png")
require.Equal(s.T(), payload, expectedPayload)
s.Require().NoError(err)
err = db.Close()
s.Require().NoError(err)
err = os.Remove(tmpfile.Name())
s.Require().NoError(err)
}