77 lines
2.5 KiB
Go
77 lines
2.5 KiB
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/blockchainer"
|
|
"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 blockchainer.Blockchainer, 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)
|
|
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.PrivateKey().PublicKey()})
|
|
require.Equal(t, acc, ntr.currAccount)
|
|
|
|
t.Run("account is already set", func(t *testing.T) {
|
|
ntr.UpdateNotaryNodes(keys.PublicKeys{acc.PrivateKey().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].PrivateKey().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].PrivateKey().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)
|
|
})
|
|
}
|