mnemonic_test.go

package gorfc1751

import (
	"errors"
	"fmt"
	"math/rand"
	"testing"
)

type testReader struct{}

func (p *testReader) Read(b []byte) (int, error) {
	r := rand.New(rand.NewSource(1)) // nolint: gosec
	return r.Read(b)
}

type testReaderWithError struct{}

func (p *testReaderWithError) Read(_ []byte) (int, error) {
	return 0, errors.New("some error") // nolint: err113
}

func ExampleNewMnemonic() {
	r := rand.New(rand.NewSource(1)) // nolint: gosec
	fmt.Println(NewMnemonic(r, 128))
	// Output: BARK TROD AMY UP LUG KNOB GAS WHEN NEWT POT KEY MEAN
}

func TestMnemonicPanics(t *testing.T) {
	t.Parallel()

	testPanicMnemonicBitSize(t, 32)
	testPanicMnemonicBitSize(t, 65)
	testPanicMnemonicBitSize(t, 224)

	testPanicMnemonicReadError(t)
}

func testPanicMnemonicBitSize(t *testing.T, bitSize uint64) {
	defer func() {
		if r := recover(); r == nil {
			t.Error("nothing panics")
			return
		}
	}()
	_ = NewMnemonic(&testReader{}, bitSize)
}

func testPanicMnemonicReadError(t *testing.T) {
	defer func() {
		if r := recover(); r == nil {
			t.Error("nothing panics")
			return
		}
	}()
	_ = NewMnemonic(&testReaderWithError{}, 64)
}

func TestNewMnemonic(t *testing.T) {
	t.Parallel()

	tests := []struct {
		name string
		have uint64
		want string
	}{
		{
			name: "test-1",
			have: 64,
			want: "BARK TROD AMY UP LUG KNOB",
		},
		{
			name: "test-2",
			have: 128,
			want: "BARK TROD AMY UP LUG KNOB GAS WHEN NEWT POT KEY MEAN",
		},
		{
			name: "test-3",
			have: 192,
			want: "BARK TROD AMY UP LUG KNOB GAS WHEN NEWT POT KEY MEAN HEAL RAM ROWS JET RIM MAE",
		},
		{
			name: "test-4",
			have: 256,
			want: "BARK TROD AMY UP LUG KNOB GAS WHEN NEWT POT KEY MEAN HEAL RAM ROWS JET RIM MAE EMMA SEAR AN JUKE NOB LIN",
		},
	}
	for _, tt := range tests {
		got := NewMnemonic(&testReader{}, tt.have)
		if got != tt.want {
			t.Errorf("\ntest\t= %s\nwant\t= %s\ngot\t= %s", tt.name, tt.want, got)
			continue
		}
	}
}