update

[bytom/vapor.git] / wallet / wallet_test.go

diff --git a/wallet/wallet_test.go b/wallet/wallet_test.go
index 823eab9..592bfb1 100644 (file)
--- a/wallet/wallet_test.go
+++ b/wallet/wallet_test.go
@@ -17,7 +17,7 @@ import (
        "github.com/vapor/consensus"
        "github.com/vapor/crypto/ed25519/chainkd"
        "github.com/vapor/database"
-       dbm "github.com/vapor/database/leveldb"
+       "github.com/vapor/database/dbutils"
        "github.com/vapor/event"
        "github.com/vapor/protocol"
        "github.com/vapor/protocol/bc"
@@ -52,7 +52,7 @@ func TestWalletVersion(t *testing.T) {
        }
        defer os.RemoveAll(dirPath)
 
-       testDB := dbm.NewDB("testdb", "leveldb", "temp")
+       testDB := database.NewDB("testdb", "leveldb", "temp")
        defer func() {
                testDB.Close()
                os.RemoveAll("temp")
@@ -117,7 +117,7 @@ func TestWalletUpdate(t *testing.T) {
        defer os.RemoveAll(dirPath)
 
        config.CommonConfig = config.DefaultConfig()
-       testDB := dbm.NewDB("testdb", "leveldb", "temp")
+       testDB := database.NewDB("testdb", "leveldb", "temp")
        defer func() {
                testDB.Close()
                os.RemoveAll("temp")
@@ -214,7 +214,7 @@ func TestRescanWallet(t *testing.T) {
        defer os.RemoveAll(dirPath)
 
        config.CommonConfig = config.DefaultConfig()
-       testDB := dbm.NewDB("testdb", "leveldb", "temp")
+       testDB := database.NewDB("testdb", "leveldb", "temp")
        defer func() {
                testDB.Close()
                os.RemoveAll("temp")
@@ -265,7 +265,7 @@ func TestMemPoolTxQueryLoop(t *testing.T) {
                t.Fatal(err)
        }
        config.CommonConfig = config.DefaultConfig()
-       testDB := dbm.NewDB("testdb", "leveldb", dirPath)
+       testDB := database.NewDB("testdb", "leveldb", dirPath)
        defer func() {
                testDB.Close()
                os.RemoveAll(dirPath)
@@ -378,7 +378,7 @@ func mockTxData(utxos []*account.UTXO, testAccount *account.Account) (*txbuilder
        return tplBuilder.Build()
 }
 
-func mockWallet(walletDB dbm.DB, account *account.Manager, asset *asset.Registry, chain *protocol.Chain, dispatcher *event.Dispatcher, txIndexFlag bool) *Wallet {
+func mockWallet(walletDB dbutils.DB, account *account.Manager, asset *asset.Registry, chain *protocol.Chain, dispatcher *event.Dispatcher, txIndexFlag bool) *Wallet {
        wallet := &Wallet{
                DB:              walletDB,
                AccountMgr:      account,