status-im/status-go

package accounts

import (
    "testing"

    "github.com/stretchr/testify/require"

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

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

    kp1 := GetProfileKeypairForTest(true, true, true)
    keycard1 := GetProfileKeycardForTest()

    kp2 := GetSeedImportedKeypair1ForTest()
    keycard2 := GetKeycardForSeedImportedKeypair1ForTest()

    keycard2Copy := GetKeycardForSeedImportedKeypair1ForTest()
    keycard2Copy.KeycardUID = keycard2Copy.KeycardUID + "C"
    keycard2Copy.KeycardName = keycard2Copy.KeycardName + "Copy"
    keycard2Copy.Position = keycard2Copy.Position + 1

    // Pre-condition
    err := db.SaveOrUpdateKeypair(kp1)
    require.NoError(t, err)
    err = db.SaveOrUpdateKeypair(kp2)
    require.NoError(t, err)
    dbKeypairs, err := db.GetActiveKeypairs()
    require.NoError(t, err)
    require.Equal(t, 2, len(dbKeypairs))

    // Test adding/reading keycards
    err = db.SaveOrUpdateKeycard(*keycard1, 0, false)
    require.NoError(t, err)
    dbKeycard1, err := db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard1, dbKeycard1))

    err = db.SaveOrUpdateKeycard(*keycard2, 0, false)
    require.NoError(t, err)
    dbKeycard2, err := db.GetKeycardByKeycardUID(keycard2.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard2, dbKeycard2))

    err = db.SaveOrUpdateKeycard(*keycard2Copy, 0, false)
    require.NoError(t, err)
    dbKeycard2Copy, err := db.GetKeycardByKeycardUID(keycard2Copy.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard2Copy, dbKeycard2Copy))

    dbKeycards, err := db.GetKeycardsWithSameKeyUID(keycard2.KeyUID)
    require.NoError(t, err)
    require.Equal(t, 2, len(dbKeycards))
    require.True(t, Contains(dbKeycards, keycard2, SameKeycards))
    require.True(t, Contains(dbKeycards, keycard2Copy, SameKeycards))

    dbKeycards, err = db.GetAllKnownKeycards()
    require.NoError(t, err)
    require.Equal(t, 3, len(dbKeycards))
    require.True(t, Contains(dbKeycards, keycard1, SameKeycards))
    require.True(t, Contains(dbKeycards, keycard2, SameKeycards))
    require.True(t, Contains(dbKeycards, keycard2Copy, SameKeycards))

    nextPosition, err := db.GetPositionForNextNewKeycard()
    require.NoError(t, err)
    require.Equal(t, uint64(len(dbKeycards)), nextPosition)

    // test adding additional accounts to keycard
    keycard1.AccountsAddresses = append(keycard1.AccountsAddresses, types.Address{0x05}, types.Address{0x06})
    err = db.SaveOrUpdateKeycard(*keycard1, 0, false)
    require.NoError(t, err)
    dbKeycard1, err = db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.NoError(t, err)
    require.Equal(t, len(keycard1.AccountsAddresses), len(dbKeycard1.AccountsAddresses))
    require.True(t, SameKeycards(keycard1, dbKeycard1))

    // Test seting a new keycard name
    keycard1.KeycardName = "Card101"
    err = db.SetKeycardName(keycard1.KeycardUID, keycard1.KeycardName, 1000)
    require.NoError(t, err)
    dbKeycard1, err = db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard1, dbKeycard1))

    // Test locking a keycard
    keycard1.KeycardLocked = true
    err = db.KeycardLocked(keycard1.KeycardUID, 1001)
    require.NoError(t, err)
    dbKeycard1, err = db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard1, dbKeycard1))

    // Test unlocking a keycard
    keycard1.KeycardLocked = false
    err = db.KeycardUnlocked(keycard1.KeycardUID, 1002)
    require.NoError(t, err)
    dbKeycard1, err = db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard1, dbKeycard1))

    // Test update keycard uid
    oldKeycardUID := keycard1.KeycardUID
    keycard1.KeycardUID = "00000000000000000000000000000000"
    err = db.UpdateKeycardUID(oldKeycardUID, keycard1.KeycardUID, 1003)
    require.NoError(t, err)
    dbKeycard1, err = db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard1, dbKeycard1))

    // Test detleting accounts (addresses) for a certain keycard
    const numOfAccountsToRemove = 2
    require.Greater(t, len(keycard1.AccountsAddresses), numOfAccountsToRemove)
    accountsToRemove := keycard1.AccountsAddresses[:numOfAccountsToRemove]
    keycard1.AccountsAddresses = keycard1.AccountsAddresses[numOfAccountsToRemove:]
    err = db.DeleteKeycardAccounts(keycard1.KeycardUID, accountsToRemove, 1004)
    require.NoError(t, err)
    dbKeycard1, err = db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard1, dbKeycard1))

    // Test detleting a keycard
    err = db.DeleteKeycard(keycard1.KeycardUID, 1006)
    require.NoError(t, err)
    dbKeycards, err = db.GetAllKnownKeycards()
    require.NoError(t, err)
    require.Equal(t, 2, len(dbKeycards))
    dbKeycards, err = db.GetKeycardsWithSameKeyUID(keycard1.KeyUID)
    require.NoError(t, err)
    require.Equal(t, 0, len(dbKeycards))
    dbKeycard, err := db.GetKeycardByKeycardUID(keycard1.KeycardUID)
    require.Error(t, err)
    require.True(t, err == ErrNoKeycardForPassedKeycardUID)
    require.Nil(t, dbKeycard)

    // Test detleting all keycards for KeyUID
    dbKeycards, err = db.GetKeycardsWithSameKeyUID(keycard2.KeyUID)
    require.NoError(t, err)
    require.Equal(t, 2, len(dbKeycards))
    err = db.DeleteAllKeycardsWithKeyUID(keycard2.KeyUID, 1007)
    require.NoError(t, err)
    dbKeycards, err = db.GetAllKnownKeycards()
    require.NoError(t, err)
    require.Equal(t, 0, len(dbKeycards))
    dbKeycards, err = db.GetKeycardsWithSameKeyUID(keycard2.KeyUID)
    require.NoError(t, err)
    require.Equal(t, 0, len(dbKeycards))
    dbKeycard, err = db.GetKeycardByKeycardUID(keycard2.KeycardUID)
    require.Error(t, err)
    require.True(t, err == ErrNoKeycardForPassedKeycardUID)
    require.Nil(t, dbKeycard)
}

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

    kp2 := GetSeedImportedKeypair1ForTest()
    keycard2 := GetKeycardForSeedImportedKeypair1ForTest()

    keycard2Copy := GetKeycardForSeedImportedKeypair1ForTest()
    keycard2Copy.KeycardUID = keycard2Copy.KeycardUID + "C"
    keycard2Copy.KeycardName = keycard2Copy.KeycardName + "Copy"
    keycard2Copy.Position = keycard2Copy.Position + 1

    // Pre-condition
    err := db.SaveOrUpdateKeypair(kp2)
    require.NoError(t, err)
    dbKeypairs, err := db.GetActiveKeypairs()
    require.NoError(t, err)
    require.Equal(t, 1, len(dbKeypairs))

    // Pre-condition - save keycards referring to previously added keypair
    err = db.SaveOrUpdateKeycard(*keycard2, 0, false)
    require.NoError(t, err)
    dbKeycard2, err := db.GetKeycardByKeycardUID(keycard2.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard2, dbKeycard2))

    err = db.SaveOrUpdateKeycard(*keycard2Copy, 0, false)
    require.NoError(t, err)
    dbKeycard2Copy, err := db.GetKeycardByKeycardUID(keycard2Copy.KeycardUID)
    require.NoError(t, err)
    require.True(t, SameKeycards(keycard2Copy, dbKeycard2Copy))

    // Check db state
    dbKeycards, err := db.GetKeycardsWithSameKeyUID(keycard2.KeyUID)
    require.NoError(t, err)
    require.Equal(t, 2, len(dbKeycards))
    require.True(t, Contains(dbKeycards, keycard2, SameKeycards))
    require.True(t, Contains(dbKeycards, keycard2Copy, SameKeycards))

    // Remove keypair
    err = db.RemoveKeypair(kp2.KeyUID, 0)
    require.NoError(t, err)

    // Check db state after deletion
    dbKeypairs, err = db.GetActiveKeypairs()
    require.NoError(t, err)
    require.Equal(t, 0, len(dbKeypairs))

    dbKeycards, err = db.GetAllKnownKeycards()
    require.NoError(t, err)
    require.Equal(t, 0, len(dbKeycards))
    dbKeycards, err = db.GetKeycardsWithSameKeyUID(kp2.KeyUID)
    require.NoError(t, err)
    require.Equal(t, 0, len(dbKeycards))
    dbKeycard, err := db.GetKeycardByKeycardUID(keycard2.KeycardUID)
    require.Error(t, err)
    require.True(t, err == ErrNoKeycardForPassedKeycardUID)
    require.Nil(t, dbKeycard)
}