package multiaccounts import ( "encoding/json" "io/ioutil" "os" "testing" "github.com/status-im/status-go/images" "github.com/status-im/status-go/protocol/sqlite" "github.com/stretchr/testify/require" ) func setupTestDB(t *testing.T) (*Database, func()) { tmpfile, err := ioutil.TempFile("", "accounts-tests-") require.NoError(t, err) db, err := InitializeDB(tmpfile.Name()) require.NoError(t, err) return db, func() { require.NoError(t, db.Close()) require.NoError(t, os.Remove(tmpfile.Name())) } } func TestAccounts(t *testing.T) { db, stop := setupTestDB(t) defer stop() expected := Account{Name: "string", KeyUID: "string", CustomizationColor: CustomizationColorBlue, ColorHash: ColorHash{{4, 3}, {4, 0}, {4, 3}, {4, 0}}, ColorID: 10, KDFIterations: sqlite.ReducedKDFIterationsNumber} require.NoError(t, db.SaveAccount(expected)) accounts, err := db.GetAccounts() require.NoError(t, err) require.Len(t, accounts, 1) require.Equal(t, expected, accounts[0]) } func TestAccountsUpdate(t *testing.T) { db, stop := setupTestDB(t) defer stop() expected := Account{KeyUID: "string", CustomizationColor: CustomizationColorBlue, ColorHash: ColorHash{{4, 3}, {4, 0}, {4, 3}, {4, 0}}, ColorID: 10, KDFIterations: sqlite.ReducedKDFIterationsNumber} require.NoError(t, db.SaveAccount(expected)) expected.Name = "chars" expected.CustomizationColor = CustomizationColorMagenta require.NoError(t, db.UpdateAccount(expected)) rst, err := db.GetAccounts() require.NoError(t, err) require.Len(t, rst, 1) require.Equal(t, expected, rst[0]) } func TestLoginUpdate(t *testing.T) { db, stop := setupTestDB(t) defer stop() accounts := []Account{{Name: "first", KeyUID: "0x1", KDFIterations: sqlite.ReducedKDFIterationsNumber}, {Name: "second", KeyUID: "0x2", KDFIterations: sqlite.ReducedKDFIterationsNumber}} for _, acc := range accounts { require.NoError(t, db.SaveAccount(acc)) } require.NoError(t, db.UpdateAccountTimestamp(accounts[0].KeyUID, 100)) require.NoError(t, db.UpdateAccountTimestamp(accounts[1].KeyUID, 10)) accounts[0].Timestamp = 100 accounts[1].Timestamp = 10 rst, err := db.GetAccounts() require.NoError(t, err) require.Equal(t, accounts, rst) } // Profile Image tests var ( keyUID = "0xdeadbeef" keyUID2 = "0x1337beef" ) func seedTestDBWithIdentityImages(t *testing.T, db *Database, keyUID string) { iis := images.SampleIdentityImages() require.NoError(t, db.StoreIdentityImages(keyUID, iis, false)) } func TestDatabase_GetIdentityImages(t *testing.T) { db, stop := setupTestDB(t) defer stop() seedTestDBWithIdentityImages(t, db, keyUID) expected := `[{"keyUid":"0xdeadbeef","type":"large","uri":"data:image/png;base64,iVBORw0KGgoAAAANSUg=","width":240,"height":300,"fileSize":1024,"resizeTarget":240,"clock":0},{"keyUid":"0xdeadbeef","type":"thumbnail","uri":"data:image/jpeg;base64,/9j/2wCEAFA3PEY8MlA=","width":80,"height":80,"fileSize":256,"resizeTarget":80,"clock":0}]` oiis, err := db.GetIdentityImages(keyUID) require.NoError(t, err) joiis, err := json.Marshal(oiis) require.NoError(t, err) require.Exactly(t, expected, string(joiis)) oiis, err = db.GetIdentityImages(keyUID2) require.NoError(t, err) require.Exactly(t, 0, len(oiis)) } func TestDatabase_GetIdentityImage(t *testing.T) { db, stop := setupTestDB(t) defer stop() seedTestDBWithIdentityImages(t, db, keyUID) cs := []struct { KeyUID string Name string Expected string }{ { keyUID, images.SmallDimName, `{"keyUid":"0xdeadbeef","type":"thumbnail","uri":"data:image/jpeg;base64,/9j/2wCEAFA3PEY8MlA=","width":80,"height":80,"fileSize":256,"resizeTarget":80,"clock":0}`, }, { keyUID, images.LargeDimName, `{"keyUid":"0xdeadbeef","type":"large","uri":"data:image/png;base64,iVBORw0KGgoAAAANSUg=","width":240,"height":300,"fileSize":1024,"resizeTarget":240,"clock":0}`, }, { keyUID2, images.LargeDimName, "null", }, } for _, c := range cs { oii, err := db.GetIdentityImage(c.KeyUID, c.Name) require.NoError(t, err) joii, err := json.Marshal(oii) require.NoError(t, err) require.Exactly(t, c.Expected, string(joii)) } } func TestDatabase_DeleteIdentityImage(t *testing.T) { db, stop := setupTestDB(t) defer stop() seedTestDBWithIdentityImages(t, db, keyUID) require.NoError(t, db.DeleteIdentityImage(keyUID)) oii, err := db.GetIdentityImage(keyUID, images.SmallDimName) require.NoError(t, err) require.Empty(t, oii) } func TestDatabase_GetAccountsWithIdentityImages(t *testing.T) { db, stop := setupTestDB(t) defer stop() testAccs := []Account{ {Name: "string", KeyUID: keyUID, Identicon: "data"}, {Name: "string", KeyUID: keyUID2}, {Name: "string", KeyUID: keyUID2 + "2"}, {Name: "string", KeyUID: keyUID2 + "3"}, } expected := `[{"name":"string","timestamp":100,"identicon":"data","colorHash":null,"colorId":0,"keycard-pairing":"","key-uid":"0xdeadbeef","images":[{"keyUid":"0xdeadbeef","type":"large","uri":"data:image/png;base64,iVBORw0KGgoAAAANSUg=","width":240,"height":300,"fileSize":1024,"resizeTarget":240,"clock":0},{"keyUid":"0xdeadbeef","type":"thumbnail","uri":"data:image/jpeg;base64,/9j/2wCEAFA3PEY8MlA=","width":80,"height":80,"fileSize":256,"resizeTarget":80,"clock":0}],"kdfIterations":3200},{"name":"string","timestamp":10,"identicon":"","colorHash":null,"colorId":0,"keycard-pairing":"","key-uid":"0x1337beef","images":null,"kdfIterations":3200},{"name":"string","timestamp":0,"identicon":"","colorHash":null,"colorId":0,"keycard-pairing":"","key-uid":"0x1337beef2","images":null,"kdfIterations":3200},{"name":"string","timestamp":0,"identicon":"","colorHash":null,"colorId":0,"keycard-pairing":"","key-uid":"0x1337beef3","images":[{"keyUid":"0x1337beef3","type":"large","uri":"data:image/png;base64,iVBORw0KGgoAAAANSUg=","width":240,"height":300,"fileSize":1024,"resizeTarget":240,"clock":0},{"keyUid":"0x1337beef3","type":"thumbnail","uri":"data:image/jpeg;base64,/9j/2wCEAFA3PEY8MlA=","width":80,"height":80,"fileSize":256,"resizeTarget":80,"clock":0}],"kdfIterations":3200}]` for _, a := range testAccs { require.NoError(t, db.SaveAccount(a)) } seedTestDBWithIdentityImages(t, db, keyUID) seedTestDBWithIdentityImages(t, db, keyUID2+"3") err := db.UpdateAccountTimestamp(keyUID, 100) require.NoError(t, err) err = db.UpdateAccountTimestamp(keyUID2, 10) require.NoError(t, err) accs, err := db.GetAccounts() require.NoError(t, err) accJSON, err := json.Marshal(accs) require.NoError(t, err) require.Exactly(t, expected, string(accJSON)) } func TestDatabase_GetAccount(t *testing.T) { db, stop := setupTestDB(t) defer stop() expected := Account{Name: "string", KeyUID: keyUID, ColorHash: ColorHash{{4, 3}, {4, 0}, {4, 3}, {4, 0}}, ColorID: 10, KDFIterations: sqlite.ReducedKDFIterationsNumber} require.NoError(t, db.SaveAccount(expected)) account, err := db.GetAccount(expected.KeyUID) require.NoError(t, err) require.Equal(t, &expected, account) } func TestDatabase_SaveAccountWithIdentityImages(t *testing.T) { db, stop := setupTestDB(t) defer stop() expected := Account{ Name: "string", KeyUID: keyUID, ColorHash: ColorHash{{4, 3}, {4, 0}, {4, 3}, {4, 0}}, ColorID: 10, Images: images.SampleIdentityImages(), } require.NoError(t, db.SaveAccount(expected)) account, err := db.GetAccount(expected.KeyUID) require.NoError(t, err) require.Exactly(t, expected.ColorHash, account.ColorHash) require.Exactly(t, expected.ColorID, account.ColorID) require.Exactly(t, expected.Identicon, account.Identicon) require.Exactly(t, expected.KeycardPairing, account.KeycardPairing) require.Exactly(t, expected.KeyUID, account.KeyUID) require.Exactly(t, expected.Name, account.Name) require.Exactly(t, expected.Timestamp, account.Timestamp) require.Len(t, expected.Images, 2) matches := 0 for _, expImg := range expected.Images { for _, accImg := range account.Images { if expImg.Name != accImg.Name { continue } matches++ require.Exactly(t, expImg.Clock, accImg.Clock) require.Exactly(t, keyUID, accImg.KeyUID) require.Exactly(t, expImg.Name, accImg.Name) require.Exactly(t, expImg.ResizeTarget, accImg.ResizeTarget) require.Exactly(t, expImg.Payload, accImg.Payload) require.Exactly(t, expImg.Height, accImg.Height) require.Exactly(t, expImg.Width, accImg.Width) require.Exactly(t, expImg.FileSize, accImg.FileSize) } } require.Equal(t, 2, matches) }