56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
|
package account
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/status-im/status-go/extkeys"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMnemonicPhraseLengthToEntropyStrenght(t *testing.T) {
|
||
|
|
scenarios := []struct {
|
||
|
|
phraseLength int
|
||
|
|
expectedStrength extkeys.EntropyStrength
|
||
|
|
expectedError error
|
||
|
|
}{
|
||
|
|
{12, 128, nil},
|
||
|
|
{15, 160, nil},
|
||
|
|
{18, 192, nil},
|
||
|
|
{21, 224, nil},
|
||
|
|
{24, 256, nil},
|
||
|
|
// invalid
|
||
|
|
{11, 0, ErrInvalidMnemonicPhraseLength},
|
||
|
|
{14, 0, ErrInvalidMnemonicPhraseLength},
|
||
|
|
{25, 0, ErrInvalidMnemonicPhraseLength},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, s := range scenarios {
|
||
|
|
strength, err := mnemonicPhraseLengthToEntropyStrenght(s.phraseLength)
|
||
|
|
assert.Equal(t, s.expectedError, err)
|
||
|
|
assert.Equal(t, s.expectedStrength, strength)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOnboarding(t *testing.T) {
|
||
|
|
count := 2
|
||
|
|
wordsCount := 24
|
||
|
|
o, _ := NewOnboarding(count, wordsCount)
|
||
|
|
assert.Equal(t, count, len(o.accounts))
|
||
|
|
|
||
|
|
for id, a := range o.accounts {
|
||
|
|
words := strings.Split(a.mnemonic, " ")
|
||
|
|
|
||
|
|
assert.Equal(t, wordsCount, len(words))
|
||
|
|
assert.NotEmpty(t, a.Info.WalletAddress)
|
||
|
|
assert.NotEmpty(t, a.Info.WalletPubKey)
|
||
|
|
assert.NotEmpty(t, a.Info.ChatAddress)
|
||
|
|
assert.NotEmpty(t, a.Info.ChatPubKey)
|
||
|
|
|
||
|
|
retrieved, err := o.Account(id)
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.Equal(t, a, retrieved)
|
||
|
|
}
|
||
|
|
}
|