X-Git-Url: http://git.osdn.net/view?a=blobdiff_plain;f=test%2Fwallet_test_util.go;fp=test%2Fwallet_test_util.go;h=570aecccae948554dbc0fbd2a070c0d8df719366;hb=f5528cbb934d3fc20f7e0fd3264fb9d7e5672d52;hp=b03b430bd4aee18320bb27dfdc92fd7c7895182f;hpb=41d53561acfb5ed7ae357808708fa6cd9032feb0;p=bytom%2Fvapor.git

diff --git a/test/wallet_test_util.go b/test/wallet_test_util.go
index b03b430b..570aeccc 100644
--- a/test/wallet_test_util.go
+++ b/test/wallet_test_util.go
@@ -12,6 +12,7 @@ import (
 	"github.com/vapor/blockchain/pseudohsm"
 	"github.com/vapor/blockchain/signers"
 	"github.com/vapor/crypto/ed25519/chainkd"
+	"github.com/vapor/database"
 	dbm "github.com/vapor/database/leveldb"
 	"github.com/vapor/event"
 	"github.com/vapor/protocol"
@@ -241,10 +242,12 @@ func (cfg *walletTestConfig) Run() error {
 		return err
 	}
 	walletDB := dbm.NewDB("wallet", "leveldb", path.Join(dirPath, "wallet_db"))
-	accountManager := account.NewManager(walletDB, chain)
+	walletStore := database.NewWalletStore(walletDB)
+	accountStore := database.NewAccountStore(walletDB)
+	accountManager := account.NewManager(accountStore, chain)
 	assets := asset.NewRegistry(walletDB, chain)
 	dispatcher := event.NewDispatcher()
-	wallet, err := w.NewWallet(walletDB, accountManager, assets, hsm, chain, dispatcher, false)
+	wallet, err := w.NewWallet(walletStore, accountManager, assets, hsm, chain, dispatcher, false)
 	if err != nil {
 		return err
 	}