mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-24 12:27:18 +00:00
a15e0797e4
* add option to log rejected gossip message * add bip39 supported mnemonic languages * Revert "add option to log rejected gossip message" This reverts commit 9a3d4486f6c8a623d9eb6b3af1a5a89aa62ecf2f. * Add mnemonic language flag * Update go.mod * Simplify language mapping * Add test for setBip39Lang * Update go.mod * Improve language matching Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> * Run gazelle + fix maligned struct Co-authored-by: Preston Van Loon <preston@prysmaticlabs.com> Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
47 lines
1.5 KiB
Go
47 lines
1.5 KiB
Go
package derived
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/v3/testing/assert"
|
|
"github.com/prysmaticlabs/prysm/v3/testing/require"
|
|
"github.com/tyler-smith/go-bip39"
|
|
"github.com/tyler-smith/go-bip39/wordlists"
|
|
)
|
|
|
|
func TestMnemonic_Generate_CanRecover(t *testing.T) {
|
|
generator := &MnemonicGenerator{}
|
|
data := make([]byte, 32)
|
|
copy(data, "hello-world")
|
|
phrase, err := generator.Generate(data)
|
|
require.NoError(t, err)
|
|
entropy, err := bip39.EntropyFromMnemonic(phrase)
|
|
require.NoError(t, err)
|
|
assert.DeepEqual(t, data, entropy, "Expected to recover original data")
|
|
}
|
|
|
|
func Test_setBip39Lang(t *testing.T) {
|
|
tests := []struct {
|
|
lang string
|
|
expectedWordlist []string
|
|
}{
|
|
{lang: "english", expectedWordlist: wordlists.English},
|
|
{lang: "chinese_traditional", expectedWordlist: wordlists.ChineseTraditional},
|
|
{lang: "chinese_simplified", expectedWordlist: wordlists.ChineseSimplified},
|
|
{lang: "czech", expectedWordlist: wordlists.Czech},
|
|
{lang: "french", expectedWordlist: wordlists.French},
|
|
{lang: "japanese", expectedWordlist: wordlists.Japanese},
|
|
{lang: "korean", expectedWordlist: wordlists.Korean},
|
|
{lang: "italian", expectedWordlist: wordlists.Italian},
|
|
{lang: "spanish", expectedWordlist: wordlists.Spanish},
|
|
{lang: "undefined", expectedWordlist: wordlists.English},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.lang, func(t *testing.T) {
|
|
setBip39Lang(tt.lang)
|
|
wordlist := bip39.GetWordList()
|
|
assert.DeepEqual(t, tt.expectedWordlist, wordlist, "Expected wordlist to match")
|
|
})
|
|
}
|
|
}
|