txpool handle orphan (#1228)

[bytom/bytom.git] / test / util.go

diff --git a/test/util.go b/test/util.go
index 94daf03..421aa3d 100644 (file)
--- a/test/util.go
+++ b/test/util.go
@@ -23,15 +23,10 @@ const (
        assetVersion = 1
 )
 
-// MockTxPool mock transaction pool
-func MockTxPool() *protocol.TxPool {
-       return protocol.NewTxPool()
-}
-
 // MockChain mock chain with genesis block
 func MockChain(testDB dbm.DB) (*protocol.Chain, *leveldb.Store, *protocol.TxPool, error) {
        store := leveldb.NewStore(testDB)
-       txPool := MockTxPool()
+       txPool := protocol.NewTxPool(store)
        chain, err := protocol.NewChain(store, txPool)
        return chain, store, txPool, err
 }
@@ -67,7 +62,7 @@ func MockTx(utxo *account.UTXO, testAccount *account.Account) (*txbuilder.Templa
 
 // MockSign sign a tx
 func MockSign(tpl *txbuilder.Template, hsm *pseudohsm.HSM, password string) (bool, error) {
-       err := txbuilder.Sign(nil, tpl, nil, password, func(_ context.Context, xpub chainkd.XPub, path [][]byte, data [32]byte, password string) ([]byte, error) {
+       err := txbuilder.Sign(nil, tpl, password, func(_ context.Context, xpub chainkd.XPub, path [][]byte, data [32]byte, password string) ([]byte, error) {
                return hsm.XSign(xpub, path, data[:], password)
        })
        if err != nil {