119 lines
3.0 KiB
Go
119 lines
3.0 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"
|
||
|
|
)
|
||
|
|
|
||
|
|
type QROpsTestSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
TestKeyComponents
|
||
|
|
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)
|
||
|
|
}
|