X-Git-Url: http://git.osdn.net/view?p=bytom%2Fvapor.git;a=blobdiff_plain;f=netsync%2Fchainmgr%2Ftx_keeper_test.go;fp=netsync%2Fchainmgr%2Ftx_keeper_test.go;h=dd269fdef28390624be82244cc1fa47df9528069;hp=7401af215dfda27750fbc0786217d9531c22b12a;hb=068fc645e200e34e38a75dc283e3e4f05ab15d7f;hpb=51100c2a5afb320a9b16674f8c66b067fe760eb3;ds=sidebyside

diff --git a/netsync/chainmgr/tx_keeper_test.go b/netsync/chainmgr/tx_keeper_test.go
index 7401af21..dd269fde 100644
--- a/netsync/chainmgr/tx_keeper_test.go
+++ b/netsync/chainmgr/tx_keeper_test.go
@@ -1,6 +1,8 @@
 package chainmgr
 
 import (
+	"io/ioutil"
+	"os"
 	"reflect"
 	"testing"
 	"time"
@@ -8,6 +10,7 @@ import (
 	"github.com/davecgh/go-spew/spew"
 
 	"github.com/vapor/consensus"
+	dbm "github.com/vapor/database/leveldb"
 	"github.com/vapor/protocol"
 	"github.com/vapor/protocol/bc"
 	"github.com/vapor/protocol/bc/types"
@@ -44,9 +47,17 @@ func getTransactions() []*types.Tx {
 }
 
 func TestSyncMempool(t *testing.T) {
+	tmpDir, err := ioutil.TempDir(".", "")
+	if err != nil {
+		t.Fatalf("failed to create temporary data folder: %v", err)
+	}
+	defer os.RemoveAll(tmpDir)
+	testDBA := dbm.NewDB("testdba", "leveldb", tmpDir)
+	testDBB := dbm.NewDB("testdbb", "leveldb", tmpDir)
+
 	blocks := mockBlocks(nil, 5)
-	a := mockSync(blocks, &mock.Mempool{})
-	b := mockSync(blocks, &mock.Mempool{})
+	a := mockSync(blocks, &mock.Mempool{}, testDBA)
+	b := mockSync(blocks, &mock.Mempool{}, testDBB)
 
 	netWork := NewNetWork()
 	netWork.Register(a, "192.168.0.1", "test node A", consensus.SFFullNode)