From 0adef6eff9027f36d04b7307b77f326ac7cde6d4 Mon Sep 17 00:00:00 2001
From: Chengcheng Zhang <943420582@qq.com>
Date: Wed, 3 Jul 2019 17:13:32 +0800
Subject: [PATCH] update

---
 wallet/recovery_test.go | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/wallet/recovery_test.go b/wallet/recovery_test.go
index 2e7fe205..1c2c699c 100644
--- a/wallet/recovery_test.go
+++ b/wallet/recovery_test.go
@@ -138,7 +138,7 @@ func TestXPubsRecoveryLock(t *testing.T) {
 	defer os.RemoveAll(dirPath)
 
 	testDB := dbm.NewDB("testdb", "leveldb", dirPath)
-	walletStore := newMockWalletStore(testDB)
+	walletStore := mock.NewMockWalletStore(testDB)
 	hsm, err := pseudohsm.New(dirPath)
 	if err != nil {
 		t.Fatal(err)
@@ -149,7 +149,7 @@ func TestXPubsRecoveryLock(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	acctStore := newMockAccountStore(testDB)
+	acctStore := mock.NewMockAccountStore(testDB)
 	acctMgr := account.NewManager(acctStore, nil)
 	recoveryMgr := newRecoveryManager(walletStore, acctMgr)
 	recoveryMgr.state = newRecoveryState()
@@ -189,7 +189,7 @@ func TestExtendScanAddresses(t *testing.T) {
 	defer os.RemoveAll(dirPath)
 
 	testDB := dbm.NewDB("testdb", "leveldb", dirPath)
-	walletStore := newMockWalletStore(testDB)
+	walletStore := mock.NewMockWalletStore(testDB)
 	hsm, err := pseudohsm.New(dirPath)
 	if err != nil {
 		t.Fatal(err)
@@ -200,7 +200,7 @@ func TestExtendScanAddresses(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	acctStore := newMockAccountStore(testDB)
+	acctStore := mock.NewMockAccountStore(testDB)
 	acctMgr := account.NewManager(acctStore, nil)
 	recoveryMgr := newRecoveryManager(walletStore, acctMgr)
 	acc1 := &account.Account{ID: "testA", Alias: "test1", Signer: &signers.Signer{XPubs: []chainkd.XPub{xpub.XPub}, KeyIndex: 1, DeriveRule: signers.BIP0044}}
@@ -247,7 +247,7 @@ func TestRecoveryFromXPubs(t *testing.T) {
 
 	testDB := dbm.NewDB("testdb", "leveldb", dirPath)
 	recoveryDB := dbm.NewDB("recdb", "leveldb", dirPath)
-	recoveryStore := newMockWalletStore(recoveryDB)
+	recoveryStore := mock.NewMockWalletStore(recoveryDB)
 	hsm, err := pseudohsm.New(dirPath)
 	if err != nil {
 		t.Fatal(err)
@@ -258,10 +258,10 @@ func TestRecoveryFromXPubs(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	acctStore := newMockAccountStore(testDB)
+	acctStore := mock.NewMockAccountStore(testDB)
 	acctMgr := account.NewManager(acctStore, nil)
 	txs, err := MockTxsP2PKH(acctMgr, xpub.XPub, false)
-	recActStore := newMockAccountStore(recoveryDB)
+	recActStore := mock.NewMockAccountStore(recoveryDB)
 	recAcctMgr := account.NewManager(recActStore, nil)
 	recoveryMgr := newRecoveryManager(recoveryStore, recAcctMgr)
 
-- 
2.11.0