+++ /dev/null
-package keys
-
-import (
- "testing"
-
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
-
- cmn "github.com/tendermint/tmlibs/common"
-)
-
-func TestLengthCalc(t *testing.T) {
- assert := assert.New(t)
-
- cases := []struct {
- bytes, words int
- flexible bool
- }{
- {1, 1, false},
- {2, 2, false},
- // bytes pairs with same word count
- {3, 3, true},
- {4, 3, true},
- {5, 4, false},
- // bytes pairs with same word count
- {10, 8, true},
- {11, 8, true},
- {12, 9, false},
- {13, 10, false},
- {20, 15, false},
- // bytes pairs with same word count
- {21, 16, true},
- {32, 24, true},
- }
-
- for _, tc := range cases {
- wl := wordlenFromBytes(tc.bytes)
- assert.Equal(tc.words, wl, "%d", tc.bytes)
-
- bl, flex := bytelenFromWords(tc.words)
- assert.Equal(tc.flexible, flex, "%d", tc.words)
- if !flex {
- assert.Equal(tc.bytes, bl, "%d", tc.words)
- } else {
- // check if it is either tc.bytes or tc.bytes +1
- choices := []int{tc.bytes, tc.bytes + 1}
- assert.Contains(choices, bl, "%d", tc.words)
- }
- }
-}
-
-func TestEncodeDecode(t *testing.T) {
- assert, require := assert.New(t), require.New(t)
-
- codec, err := LoadCodec("english")
- require.Nil(err, "%+v", err)
-
- cases := [][]byte{
- {7, 8, 9}, // TODO: 3 words -> 3 or 4 bytes
- {12, 54, 99, 11}, // TODO: 3 words -> 3 or 4 bytes
- {0, 54, 99, 11}, // TODO: 3 words -> 3 or 4 bytes, detect leading 0
- {1, 2, 3, 4, 5}, // normal
- {0, 0, 0, 0, 122, 23, 82, 195}, // leading 0s (8 chars, unclear)
- {0, 0, 0, 0, 5, 22, 123, 55, 22}, // leading 0s (9 chars, clear)
- {22, 44, 55, 1, 13, 0, 0, 0, 0}, // trailing 0s (9 chars, clear)
- {0, 5, 253, 2, 0}, // leading and trailing zeros
- {255, 196, 172, 234, 192, 255}, // big numbers
- {255, 196, 172, 1, 234, 192, 255}, // big numbers, two length choices
- // others?
- }
-
- for i, tc := range cases {
- w, err := codec.BytesToWords(tc)
- if assert.Nil(err, "%d: %v", i, err) {
- b, err := codec.WordsToBytes(w)
- if assert.Nil(err, "%d: %v", i, err) {
- assert.Equal(len(tc), len(b))
- assert.Equal(tc, b)
- }
- }
- }
-}
-
-func TestCheckInvalidLists(t *testing.T) {
- assert := assert.New(t)
-
- trivial := []string{"abc", "def"}
- short := make([]string, 1234)
- long := make([]string, BankSize+1)
- right := make([]string, BankSize)
- dups := make([]string, BankSize)
-
- for _, list := range [][]string{short, long, right, dups} {
- for i := range list {
- list[i] = cmn.RandStr(8)
- }
- }
- // create one single duplicate
- dups[192] = dups[782]
-
- cases := []struct {
- words []string
- loadable bool
- valid bool
- }{
- {trivial, false, false},
- {short, false, false},
- {long, false, false},
- {dups, true, false}, // we only check dups on first use...
- {right, true, true},
- }
-
- for i, tc := range cases {
- codec, err := NewCodec(tc.words)
- if !tc.loadable {
- assert.NotNil(err, "%d", i)
- } else if assert.Nil(err, "%d: %+v", i, err) {
- data := cmn.RandBytes(32)
- w, err := codec.BytesToWords(data)
- if tc.valid {
- assert.Nil(err, "%d: %+v", i, err)
- b, err1 := codec.WordsToBytes(w)
- assert.Nil(err1, "%d: %+v", i, err1)
- assert.Equal(data, b)
- } else {
- assert.NotNil(err, "%d", i)
- }
- }
- }
-
-}
-
-func getRandWord(c *WordCodec) string {
- idx := cmn.RandInt() % BankSize
- return c.words[idx]
-}
-
-func getDiffWord(c *WordCodec, not string) string {
- w := getRandWord(c)
- if w == not {
- w = getRandWord(c)
- }
- return w
-}
-
-func TestCheckTypoDetection(t *testing.T) {
- assert, require := assert.New(t), require.New(t)
-
- banks := []string{"english", "spanish", "japanese", "chinese_simplified"}
-
- for _, bank := range banks {
- codec, err := LoadCodec(bank)
- require.Nil(err, "%s: %+v", bank, err)
- for i := 0; i < 1000; i++ {
- numBytes := cmn.RandInt()%60 + 4
- data := cmn.RandBytes(numBytes)
-
- words, err := codec.BytesToWords(data)
- assert.Nil(err, "%s: %+v", bank, err)
- good, err := codec.WordsToBytes(words)
- assert.Nil(err, "%s: %+v", bank, err)
- assert.Equal(data, good, bank)
-
- // now try some tweaks...
- cut := words[1:]
- _, err = codec.WordsToBytes(cut)
- assert.NotNil(err, "%s: %s", bank, words)
-
- // swap a word within the bank, should fails
- words[3] = getDiffWord(codec, words[3])
- _, err = codec.WordsToBytes(words)
- assert.NotNil(err, "%s: %s", bank, words)
-
- // put a random word here, must fail
- words[3] = cmn.RandStr(10)
- _, err = codec.WordsToBytes(words)
- assert.NotNil(err, "%s: %s", bank, words)
- }
- }
-}
OSDN Git Service
