From eee22d16fcb1967563a1c37d48c39df9a86bd83c Mon Sep 17 00:00:00 2001
From: Chengcheng Zhang <943420582@qq.com>
Date: Thu, 4 Jul 2019 12:33:53 +0800
Subject: [PATCH] update

---
 wallet/wallet_test.go | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/wallet/wallet_test.go b/wallet/wallet_test.go
index 9a42be3f..392cea66 100644
--- a/wallet/wallet_test.go
+++ b/wallet/wallet_test.go
@@ -63,7 +63,7 @@ func TestWalletVersion(t *testing.T) {
 	defer os.RemoveAll(dirPath)
 
 	testDB := dbm.NewDB("testdb", "leveldb", "temp")
-	walletStore := mock.NewMockWalletStore(testDB)
+	walletStore := NewMockWalletStore(testDB)
 	defer func() {
 		testDB.Close()
 		os.RemoveAll("temp")
@@ -135,7 +135,7 @@ func TestWalletUpdate(t *testing.T) {
 	}()
 
 	store := database.NewStore(testDB)
-	walletStore := mock.NewMockWalletStore(testDB)
+	walletStore := NewMockWalletStore(testDB)
 	dispatcher := event.NewDispatcher()
 	txPool := protocol.NewTxPool(store, dispatcher)
 
@@ -229,7 +229,7 @@ func TestRescanWallet(t *testing.T) {
 
 	config.CommonConfig = config.DefaultConfig()
 	testDB := dbm.NewDB("testdb", "leveldb", "temp")
-	walletStore := mock.NewMockWalletStore(testDB)
+	walletStore := NewMockWalletStore(testDB)
 	defer func() {
 		testDB.Close()
 		os.RemoveAll("temp")
@@ -338,7 +338,7 @@ func TestMemPoolTxQueryLoop(t *testing.T) {
 	//block := mockSingleBlock(tx)
 	txStatus := bc.NewTransactionStatus()
 	txStatus.SetStatus(0, false)
-	walletStore := mock.NewMockWalletStore(testDB)
+	walletStore := NewMockWalletStore(testDB)
 	w, err := NewWallet(walletStore, accountManager, reg, hsm, chain, dispatcher, false)
 	go w.memPoolTxQueryLoop()
 	w.eventDispatcher.Post(protocol.TxMsgEvent{TxMsg: &protocol.TxPoolMsg{TxDesc: &protocol.TxDesc{Tx: tx}, MsgType: protocol.MsgNewTx}})
-- 
2.11.0