neo-go/pkg/services/notary/node_test.go

package notary

import (
	"testing"

	"github.com/nspcc-dev/neo-go/internal/fakechain"
	"github.com/nspcc-dev/neo-go/pkg/config"
	"github.com/nspcc-dev/neo-go/pkg/config/netmode"
	"github.com/nspcc-dev/neo-go/pkg/core/mempool"
	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
	"github.com/nspcc-dev/neo-go/pkg/wallet"
	"github.com/stretchr/testify/require"
	"go.uber.org/zap/zaptest"
)

func getTestNotary(t *testing.T, bc Ledger, walletPath, pass string) (*wallet.Account, *Notary, *mempool.Pool) {
	mainCfg := config.P2PNotary{
		Enabled: true,
		UnlockWallet: config.Wallet{
			Path:     walletPath,
			Password: pass,
		},
	}
	mp := mempool.New(10, 1, true, nil)
	cfg := Config{
		MainCfg: mainCfg,
		Chain:   bc,
		Log:     zaptest.NewLogger(t),
	}
	ntr, err := NewNotary(cfg, netmode.UnitTestNet, mp, nil)
	require.NoError(t, err)

	w, err := wallet.NewWalletFromFile(walletPath)
	require.NoError(t, err)
	require.NoError(t, w.Accounts[0].Decrypt(pass, w.Scrypt))
	return w.Accounts[0], ntr, mp
}

func TestUpdateNotaryNodes(t *testing.T) {
	bc := fakechain.NewFakeChain()
	acc, ntr, _ := getTestNotary(t, bc, "./testdata/notary1.json", "one")
	randomKey, err := keys.NewPrivateKey()
	require.NoError(t, err)
	// currAcc is nil before UpdateNotaryNodes call
	require.Nil(t, ntr.currAccount)
	// set account for the first time
	ntr.UpdateNotaryNodes(keys.PublicKeys{acc.PublicKey()})
	require.Equal(t, acc, ntr.currAccount)

	t.Run("account is already set", func(t *testing.T) {
		ntr.UpdateNotaryNodes(keys.PublicKeys{acc.PublicKey(), randomKey.PublicKey()})
		require.Equal(t, acc, ntr.currAccount)
	})

	t.Run("another account from the same wallet", func(t *testing.T) {
		t.Run("good config password", func(t *testing.T) {
			w, err := wallet.NewWalletFromFile("./testdata/notary1.json")
			require.NoError(t, err)
			require.NoError(t, w.Accounts[1].Decrypt("one", w.Scrypt))
			ntr.UpdateNotaryNodes(keys.PublicKeys{w.Accounts[1].PublicKey()})
			require.Equal(t, w.Accounts[1], ntr.currAccount)
		})
		t.Run("bad config password", func(t *testing.T) {
			w, err := wallet.NewWalletFromFile("./testdata/notary1.json")
			require.NoError(t, err)
			require.NoError(t, w.Accounts[2].Decrypt("four", w.Scrypt))
			ntr.UpdateNotaryNodes(keys.PublicKeys{w.Accounts[2].PublicKey()})
			require.Nil(t, ntr.currAccount)
		})
	})

	t.Run("unknown account", func(t *testing.T) {
		ntr.UpdateNotaryNodes(keys.PublicKeys{randomKey.PublicKey()})
		require.Nil(t, ntr.currAccount)
	})
}