From 56a616f21c52e773294aef327fdc45c4f7f6ec23 Mon Sep 17 00:00:00 2001
From: Anna Shaleva <shaleva.ann@nspcc.ru>
Date: Tue, 11 May 2021 11:39:13 +0300
Subject: [PATCH] wallet: add test to regenerate Notary testwallets

---
 pkg/wallet/regenerate_test.go | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/pkg/wallet/regenerate_test.go b/pkg/wallet/regenerate_test.go
index f12ced41e..56499e5a9 100644
--- a/pkg/wallet/regenerate_test.go
+++ b/pkg/wallet/regenerate_test.go
@@ -123,3 +123,35 @@ func TestRegenerateWalletTestdata(t *testing.T) {
 	require.NoError(t, w.savePretty())
 	w.Close()
 }
+
+func TestRegenerateNotaryWallets(t *testing.T) {
+	if !regenerate {
+		return
+	}
+	const (
+		walletDir = "../services/notary/testdata/"
+		acc1WIF   = "L1MstxuD8SvS9HuFcV5oYzcdA1xX8D9bD9qPwg8fU5SSywYBecg3"
+		acc2WIF   = "L2iGxPvxbyWpYEbCZk2L3PgT7sCQaSDAbBC4MRLAjhs1s2JZ1xs5"
+		acc3WIF   = "L1xD2yiUyARX8DAkWa8qGpWpwjqW2u717VzUJyByk6s7HinhRVZv"
+		acc4WIF   = "L1ioz93TNt6Nu1aoMpZQ4zgdtgC8ZvJMC6pyHFkrovdR3SFwbn6n"
+	)
+
+	acc1 := getAccount(t, acc1WIF, "one")
+	acc2 := getAccount(t, acc2WIF, "one")
+	acc3 := getAccount(t, acc3WIF, "four")
+
+	w, err := NewWallet(path.Join(walletDir, "notary1.json"))
+	require.NoError(t, err)
+	w.AddAccount(acc1)
+	w.AddAccount(acc2)
+	w.AddAccount(acc3)
+	require.NoError(t, w.savePretty())
+	w.Close()
+
+	acc4 := getAccount(t, acc4WIF, "two")
+	w, err = NewWallet(path.Join(walletDir, "notary2.json"))
+	require.NoError(t, err)
+	w.AddAccount(acc4)
+	require.NoError(t, w.savePretty())
+	w.Close()
+}