72 lines
2.4 KiB
Go
72 lines
2.4 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/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) {
|
||
|
|
bc.(*fakechain.FakeChain).ProtocolConfiguration.P2PNotary = config.P2PNotary{
|
||
|
|
Enabled: true,
|
||
|
|
UnlockWallet: config.Wallet{
|
||
|
|
Path: walletPath,
|
||
|
|
Password: pass,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
mp := mempool.New(10, 1, true)
|
||
|
|
ntr, err := NewNotary(bc, mp, zaptest.NewLogger(t), nil)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
w, err := wallet.NewWalletFromFile(walletPath)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.NoError(t, w.Accounts[0].Decrypt(pass))
|
||
|
|
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"))
|
||
|
|
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"))
|
||
|
|
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)
|
||
|
|
})
|
||
|
|
}
|