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) }