From 75104a4a34d053b2468a6e648e8527aa89805d03 Mon Sep 17 00:00:00 2001
From: Vsevolod Brekelov <vbrekelov@nspcc.ru>
Date: Wed, 6 Nov 2019 13:15:47 +0300
Subject: [PATCH] wallet: add unit tests

---
 pkg/wallet/wallet_test.go | 113 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 113 insertions(+)

diff --git a/pkg/wallet/wallet_test.go b/pkg/wallet/wallet_test.go
index 93b08c118..2a80ca765 100644
--- a/pkg/wallet/wallet_test.go
+++ b/pkg/wallet/wallet_test.go
@@ -1,8 +1,121 @@
 package wallet
 
 import (
+	"encoding/json"
+	"io/ioutil"
+	"os"
 	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+const (
+	walletTemplate = "testWallet"
 )
 
 func TestNewWallet(t *testing.T) {
+	wallet := checkWalletConstructor(t)
+	require.NotNil(t, wallet)
+}
+
+func TestNewWalletFromFile_Negative_EmptyFile(t *testing.T) {
+	_ = checkWalletConstructor(t)
+	walletFromFile, err2 := NewWalletFromFile(walletTemplate)
+	require.Errorf(t, err2, "EOF")
+	require.Nil(t, walletFromFile)
+}
+
+func TestNewWalletFromFile_Negative_NoFile(t *testing.T) {
+	_, err := NewWalletFromFile(walletTemplate)
+	require.Errorf(t, err, "open testWallet: no such file or directory")
+}
+
+func TestCreateAccount(t *testing.T) {
+	wallet := checkWalletConstructor(t)
+
+	errAcc := wallet.CreateAccount("testName", "testPass")
+	require.NoError(t, errAcc)
+	accounts := wallet.Accounts
+	require.Len(t, accounts, 1)
+}
+
+func TestAddAccount(t *testing.T) {
+	wallet := checkWalletConstructor(t)
+
+	wallet.AddAccount(&Account{
+		privateKey:   nil,
+		publicKey:    nil,
+		wif:          "",
+		Address:      "",
+		EncryptedWIF: "",
+		Label:        "",
+		Contract:     nil,
+		Locked:       false,
+		Default:      false,
+	})
+	accounts := wallet.Accounts
+	require.Len(t, accounts, 1)
+}
+
+func TestPath(t *testing.T) {
+	wallet := checkWalletConstructor(t)
+
+	path := wallet.Path()
+	require.NotEmpty(t, path)
+}
+
+func TestSave(t *testing.T) {
+	file, err := ioutil.TempFile("", walletTemplate)
+	require.NoError(t, err)
+	wallet, err := NewWallet(file.Name())
+	require.NoError(t, err)
+	wallet.AddAccount(&Account{
+		privateKey:   nil,
+		publicKey:    nil,
+		wif:          "",
+		Address:      "",
+		EncryptedWIF: "",
+		Label:        "",
+		Contract:     nil,
+		Locked:       false,
+		Default:      false,
+	})
+
+	defer removeWallet(t, file.Name())
+	errForSave := wallet.Save()
+	require.NoError(t, errForSave)
+
+	openedWallet, err := NewWalletFromFile(wallet.path)
+	require.NoError(t, err)
+	require.Equal(t, wallet.Accounts, openedWallet.Accounts)
+}
+
+func TestJSONMarshallUnmarshal(t *testing.T) {
+	wallet := checkWalletConstructor(t)
+
+	bytes, err := wallet.JSON()
+	require.NoError(t, err)
+	require.NotNil(t, bytes)
+
+	unmarshalledWallet := &Wallet{}
+	errUnmarshal := json.Unmarshal(bytes, unmarshalledWallet)
+
+	require.NoError(t, errUnmarshal)
+	require.Equal(t, wallet.Version, unmarshalledWallet.Version)
+	require.Equal(t, wallet.Accounts, unmarshalledWallet.Accounts)
+	require.Equal(t, wallet.Scrypt, unmarshalledWallet.Scrypt)
+}
+
+func checkWalletConstructor(t *testing.T) *Wallet {
+	file, err := ioutil.TempFile("", walletTemplate)
+	require.NoError(t, err)
+	wallet, err := NewWallet(file.Name())
+	defer removeWallet(t, file.Name())
+	require.NoError(t, err)
+	return wallet
+}
+
+func removeWallet(t *testing.T, walletPath string) {
+	err := os.RemoveAll(walletPath)
+	require.NoError(t, err)
 }