status-go/multiaccounts/accounts/database_test.go

package accounts

import (
	"database/sql"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/status-im/status-go/appdatabase"
	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/multiaccounts/common"
)

func setupTestDB(t *testing.T) (*Database, func()) {
	db, stop, err := appdatabase.SetupTestSQLDB("settings-tests-")
	if err != nil {
		require.NoError(t, stop())
	}
	require.NoError(t, err)

	d, err := NewDB(db)
	if err != nil {
		require.NoError(t, stop())
	}
	require.NoError(t, err)

	return d, func() {
		require.NoError(t, stop())
	}
}

func TestGetAddresses(t *testing.T) {
	db, stop := setupTestDB(t)
	defer stop()
	accounts := []*Account{
		{Address: types.Address{0x01}, Chat: true, Wallet: true},
		{Address: types.Address{0x02}},
	}
	require.NoError(t, db.SaveOrUpdateAccounts(accounts))
	addresses, err := db.GetAddresses()
	require.NoError(t, err)
	require.Equal(t, []types.Address{{0x01}, {0x02}}, addresses)
}

func TestGetWalletAddress(t *testing.T) {
	db, stop := setupTestDB(t)
	defer stop()
	address := types.Address{0x01}
	_, err := db.GetWalletAddress()
	require.Equal(t, err, sql.ErrNoRows)
	require.NoError(t, db.SaveOrUpdateAccounts([]*Account{{Address: address, Wallet: true}}))
	wallet, err := db.GetWalletAddress()
	require.NoError(t, err)
	require.Equal(t, address, wallet)
}

func TestGetChatAddress(t *testing.T) {
	db, stop := setupTestDB(t)
	defer stop()
	address := types.Address{0x01}
	_, err := db.GetChatAddress()
	require.Equal(t, err, sql.ErrNoRows)
	require.NoError(t, db.SaveOrUpdateAccounts([]*Account{{Address: address, Chat: true}}))
	chat, err := db.GetChatAddress()
	require.NoError(t, err)
	require.Equal(t, address, chat)
}

func TestAddressExists(t *testing.T) {
	db, stop := setupTestDB(t)
	defer stop()

	accounts := []*Account{
		{Address: types.Address{0x01}, Chat: true, Wallet: true},
	}
	require.NoError(t, db.SaveOrUpdateAccounts(accounts))

	exists, err := db.AddressExists(accounts[0].Address)
	require.NoError(t, err)
	require.True(t, exists)
}

func TestAddressDoesntExist(t *testing.T) {
	db, stop := setupTestDB(t)
	defer stop()
	exists, err := db.AddressExists(types.Address{1, 1, 1})
	require.NoError(t, err)
	require.False(t, exists)
}

func TestWatchOnlyAccounts(t *testing.T) {
	db, stop := setupTestDB(t)
	defer stop()

	// check the db
	dbAccounts, err := db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, 0, len(dbAccounts))

	woAccounts := GetWatchOnlyAccountsForTest()

	// try to save keypair with watch only accounts
	kp := &Keypair{}
	kp.Accounts = append(kp.Accounts, woAccounts...)
	err = db.SaveOrUpdateKeypair(kp)
	require.Error(t, err)

	// check the db after that trying to save keypair with watch only accounts
	dbAccounts, err = db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, 0, len(dbAccounts))

	// save watch only accounts
	err = db.SaveOrUpdateAccounts(woAccounts)
	require.NoError(t, err)
	_, err = db.GetKeypairByKeyUID(woAccounts[0].KeyUID)
	require.Error(t, err)
	require.True(t, err == ErrDbKeypairNotFound)
	dbAccounts, err = db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, len(woAccounts), len(dbAccounts))
	require.Equal(t, woAccounts, dbAccounts)

	// try to save the same watch only account again
	err = db.SaveOrUpdateAccounts(woAccounts[:1])
	require.NoError(t, err)
	dbAccounts, err = db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, len(woAccounts), len(dbAccounts))
	dbAcc, err := db.GetAccountByAddress(woAccounts[:1][0].Address)
	require.NoError(t, err)
	require.Equal(t, woAccounts[:1][0], dbAcc)

	// try to save new watch only account
	wo4 := &Account{
		Address: types.Address{0x14},
		Type:    AccountTypeWatch,
		Name:    "WatchOnlyAcc4",
		ColorID: common.CustomizationColorPrimary,
		Emoji:   "emoji-1",
	}
	err = db.SaveOrUpdateAccounts([]*Account{wo4})
	require.NoError(t, err)
	dbAccounts, err = db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, len(woAccounts)+1, len(dbAccounts))
	dbAcc, err = db.GetAccountByAddress(wo4.Address)
	require.NoError(t, err)
	require.Equal(t, wo4, dbAcc)

	// updated watch onl to save the same account after it's saved
	wo4.Name = wo4.Name + "updated"
	wo4.ColorID = common.CustomizationColorCamel
	wo4.Emoji = wo4.Emoji + "updated"
	err = db.SaveOrUpdateAccounts([]*Account{wo4})
	require.NoError(t, err)
	dbAccounts, err = db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, len(woAccounts)+1, len(dbAccounts))
	dbAcc, err = db.GetAccountByAddress(wo4.Address)
	require.NoError(t, err)
	require.Equal(t, wo4, dbAcc)

	// try to delete keypair for watch only account
	err = db.DeleteKeypair(wo4.KeyUID)
	require.Error(t, err)
	require.True(t, err == ErrDbKeypairNotFound)

	// try to delete watch only account
	err = db.DeleteAccount(wo4.Address)
	require.NoError(t, err)
	dbAccounts, err = db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, len(woAccounts), len(dbAccounts))
	_, err = db.GetAccountByAddress(wo4.Address)
	require.Error(t, err)
	require.True(t, err == ErrDbAccountNotFound)
}

func TestUpdateKeypairName(t *testing.T) {
	db, stop := setupTestDB(t)
	defer stop()

	kp := GetProfileKeypairForTest(true, false, false)

	// check the db
	dbAccounts, err := db.GetAccounts()
	require.NoError(t, err)
	require.Equal(t, 0, len(dbAccounts))

	// save keypair
	err = db.SaveOrUpdateKeypair(kp)
	require.NoError(t, err)
	dbKeypairs, err := db.GetKeypairs()
	require.NoError(t, err)
	require.Equal(t, 1, len(dbKeypairs))
	require.True(t, SameKeypairs(kp, dbKeypairs[0]))

	// update keypair name
	kp.Name = kp.Name + "updated"
	err = db.UpdateKeypairName(kp.KeyUID, kp.Name, kp.Clock)
	require.NoError(t, err)

	// check keypair
	dbKp, err := db.GetKeypairByKeyUID(kp.KeyUID)
	require.NoError(t, err)
	require.Equal(t, len(kp.Accounts), len(dbKp.Accounts))
	require.True(t, SameKeypairs(kp, dbKp))
}

func TestKeypairs(t *testing.T) {
	keypairs := []*Keypair{
		GetProfileKeypairForTest(true, true, true),
		GetSeedImportedKeypair1ForTest(),
		GetPrivKeyImportedKeypairForTest(), // in this context (when testing db functions) there is not limitations for private key imported keypair
	}

	for _, kp := range keypairs {
		t.Run("test keypair "+kp.Name, func(t *testing.T) {
			db, stop := setupTestDB(t)
			defer stop()

			// check the db
			dbKeypairs, err := db.GetKeypairs()
			require.NoError(t, err)
			require.Equal(t, 0, len(dbKeypairs))
			dbAccounts, err := db.GetAccounts()
			require.NoError(t, err)
			require.Equal(t, 0, len(dbAccounts))

			expectedLastUsedDerivationIndex := uint64(len(kp.Accounts) - 1)
			if kp.Type == KeypairTypeProfile {
				expectedLastUsedDerivationIndex-- // subtract one more in case of profile keypair because of chat account
			}

			// save keypair
			err = db.SaveOrUpdateKeypair(kp)
			require.NoError(t, err)
			dbKeypairs, err = db.GetKeypairs()
			require.NoError(t, err)
			require.Equal(t, 1, len(dbKeypairs))
			dbKp, err := db.GetKeypairByKeyUID(kp.KeyUID)
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts), len(dbKp.Accounts))
			kp.LastUsedDerivationIndex = expectedLastUsedDerivationIndex
			require.Equal(t, kp, dbKp)
			dbAccounts, err = db.GetAccounts()
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts), len(dbAccounts))

			// delete keypair
			err = db.DeleteKeypair(kp.KeyUID)
			require.NoError(t, err)
			_, err = db.GetKeypairByKeyUID(kp.KeyUID)
			require.Error(t, err)
			require.True(t, err == ErrDbKeypairNotFound)

			// save keypair again to test the flow below
			err = db.SaveOrUpdateKeypair(kp)
			require.NoError(t, err)
			dbKeypairs, err = db.GetKeypairs()
			require.NoError(t, err)
			require.Equal(t, 1, len(dbKeypairs))

			ind := len(kp.Accounts) - 1
			accToUpdate := kp.Accounts[ind]

			// try to save the same account again
			err = db.SaveOrUpdateAccounts([]*Account{accToUpdate})
			require.NoError(t, err)
			dbKp, err = db.GetKeypairByKeyUID(kp.KeyUID)
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts), len(dbKp.Accounts))
			require.Equal(t, kp, dbKp)
			dbAccounts, err = db.GetAccounts()
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts), len(dbAccounts))

			// update an existing account
			accToUpdate.Name = accToUpdate.Name + "updated"
			accToUpdate.ColorID = common.CustomizationColorBrown
			accToUpdate.Emoji = accToUpdate.Emoji + "updated"

			err = db.SaveOrUpdateAccounts([]*Account{accToUpdate})
			require.NoError(t, err)
			dbKp, err = db.GetKeypairByKeyUID(kp.KeyUID)
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts), len(dbKp.Accounts))
			dbAccounts, err = db.GetAccounts()
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts), len(dbAccounts))
			dbAcc, err := db.GetAccountByAddress(accToUpdate.Address)
			require.NoError(t, err)
			require.Equal(t, accToUpdate, dbAcc)

			// update keypair name
			kpToUpdate := kp
			kpToUpdate.Name = kpToUpdate.Name + "updated"
			err = db.SaveOrUpdateKeypair(kp)
			require.NoError(t, err)
			dbKeypairs, err = db.GetKeypairs()
			require.NoError(t, err)
			require.Equal(t, 1, len(dbKeypairs))
			dbKp, err = db.GetKeypairByKeyUID(kp.KeyUID)
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts), len(dbKp.Accounts))
			require.Equal(t, kpToUpdate, dbKp)

			// save new account to an existing keypair which is out of the default Status' derivation root path
			accToAdd := kp.Accounts[ind]
			accToAdd.Address = types.Address{0x08}
			accToAdd.Path = "m/44'/60'/0'/0/10"
			accToAdd.PublicKey = types.Hex2Bytes("0x000000008")
			accToAdd.Name = "Generated Acc 8"

			err = db.SaveOrUpdateAccounts([]*Account{accToAdd})
			require.NoError(t, err)
			dbKp, err = db.GetKeypairByKeyUID(kp.KeyUID)
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts)+1, len(dbKp.Accounts))
			require.Equal(t, kp.LastUsedDerivationIndex, dbKp.LastUsedDerivationIndex)
			dbAccounts, err = db.GetAccounts()
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts)+1, len(dbAccounts))
			dbAcc, err = db.GetAccountByAddress(accToUpdate.Address)
			require.NoError(t, err)
			require.Equal(t, accToAdd, dbAcc)

			// save new account to an existing keypair which follows Status' default derivation root path
			accToAdd = kp.Accounts[ind]
			accToAdd.Address = types.Address{0x09}
			accToAdd.Path = "m/44'/60'/0'/0/3"
			accToAdd.PublicKey = types.Hex2Bytes("0x000000009")
			accToAdd.Name = "Generated Acc 9"

			expectedLastUsedDerivationIndex = 3
			if kp.Type == KeypairTypeSeed {
				accToAdd.Path = "m/44'/60'/0'/0/2"
				expectedLastUsedDerivationIndex = 2
			} else if kp.Type == KeypairTypeKey {
				accToAdd.Path = "m/44'/60'/0'/0/1"
				expectedLastUsedDerivationIndex = 1
			}

			err = db.SaveOrUpdateAccounts([]*Account{accToAdd})
			require.NoError(t, err)
			dbKp, err = db.GetKeypairByKeyUID(kp.KeyUID)
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts)+2, len(dbKp.Accounts))
			require.Equal(t, expectedLastUsedDerivationIndex, dbKp.LastUsedDerivationIndex)
			dbAccounts, err = db.GetAccounts()
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts)+2, len(dbAccounts))
			dbAcc, err = db.GetAccountByAddress(accToUpdate.Address)
			require.NoError(t, err)
			require.Equal(t, accToAdd, dbAcc)

			// delete account
			err = db.DeleteAccount(accToAdd.Address)
			require.NoError(t, err)
			dbAccounts, err = db.GetAccounts()
			require.NoError(t, err)
			require.Equal(t, len(kp.Accounts)+1, len(dbAccounts))
			_, err = db.GetAccountByAddress(accToAdd.Address)
			require.Error(t, err)
			require.True(t, err == ErrDbAccountNotFound)

			for _, acc := range dbAccounts {
				err = db.DeleteAccount(acc.Address)
				require.NoError(t, err)
			}

			_, err = db.GetKeypairByKeyUID(kp.KeyUID)
			require.Error(t, err)
			require.True(t, err == ErrDbKeypairNotFound)
		})
	}
}