Paging (#234)

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

diff --git a/wallet/wallet_test.go b/wallet/wallet_test.go
index 3a65465..ac966f4 100644 (file)
--- a/wallet/wallet_test.go
+++ b/wallet/wallet_test.go
@@ -53,7 +53,10 @@ func TestWalletVersion(t *testing.T) {
        defer os.RemoveAll(dirPath)
 
        testDB := dbm.NewDB("testdb", "leveldb", "temp")
-       defer os.RemoveAll("temp")
+       defer func() {
+               testDB.Close()
+               os.RemoveAll("temp")
+       }()
 
        dispatcher := event.NewDispatcher()
        w := mockWallet(testDB, nil, nil, nil, dispatcher, false)
@@ -113,8 +116,12 @@ func TestWalletUpdate(t *testing.T) {
        }
        defer os.RemoveAll(dirPath)
 
+       config.CommonConfig = config.DefaultConfig()
        testDB := dbm.NewDB("testdb", "leveldb", "temp")
-       defer os.RemoveAll("temp")
+       defer func() {
+               testDB.Close()
+               os.RemoveAll("temp")
+       }()
 
        store := database.NewStore(testDB)
        dispatcher := event.NewDispatcher()
@@ -179,7 +186,7 @@ func TestWalletUpdate(t *testing.T) {
                t.Fatal(err)
        }
 
-       wants, err := w.GetTransactions("")
+       wants, err := w.GetTransactions(testAccount.ID, "", 1, false)
        if len(wants) != 1 {
                t.Fatal(err)
        }
@@ -206,8 +213,12 @@ func TestRescanWallet(t *testing.T) {
        }
        defer os.RemoveAll(dirPath)
 
+       config.CommonConfig = config.DefaultConfig()
        testDB := dbm.NewDB("testdb", "leveldb", "temp")
-       defer os.RemoveAll("temp")
+       defer func() {
+               testDB.Close()
+               os.RemoveAll("temp")
+       }()
 
        store := database.NewStore(testDB)
        dispatcher := event.NewDispatcher()
@@ -253,9 +264,12 @@ func TestMemPoolTxQueryLoop(t *testing.T) {
        if err != nil {
                t.Fatal(err)
        }
-       defer os.RemoveAll(dirPath)
-
+       config.CommonConfig = config.DefaultConfig()
        testDB := dbm.NewDB("testdb", "leveldb", dirPath)
+       defer func() {
+               testDB.Close()
+               os.RemoveAll(dirPath)
+       }()
 
        store := database.NewStore(testDB)
        dispatcher := event.NewDispatcher()