add unit test for block header

[bytom/bytom.git] / protocol / txpool_test.go

diff --git a/protocol/txpool_test.go b/protocol/txpool_test.go
index 4ad559a..fefbd76 100644 (file)
--- a/protocol/txpool_test.go
+++ b/protocol/txpool_test.go
@@ -4,7 +4,8 @@ import (
        "testing"
 
        "github.com/bytom/consensus"
-       "github.com/bytom/protocol/bc/legacy"
+       "github.com/bytom/protocol/bc/types"
+       "github.com/bytom/protocol/vm/vmutil"
 )
 
 func TestTxPool(t *testing.T) {
@@ -49,16 +50,19 @@ func TestTxPool(t *testing.T) {
        }
 }
 
-func mockCoinbaseTx(serializedSize uint64, amount uint64) *legacy.Tx {
-       oldTx := &legacy.TxData{
+func mockCoinbaseTx(serializedSize uint64, amount uint64) *types.Tx {
+       cp, _ := vmutil.DefaultCoinbaseProgram()
+       oldTx := &types.TxData{
                SerializedSize: serializedSize,
-               Outputs: []*legacy.TxOutput{
-                       legacy.NewTxOutput(*consensus.BTMAssetID, amount, []byte{1}, nil),
+               Inputs: []*types.TxInput{
+                       types.NewCoinbaseInput(nil),
+               },
+               Outputs: []*types.TxOutput{
+                       types.NewTxOutput(*consensus.BTMAssetID, amount, cp),
                },
        }
-
-       return &legacy.Tx{
+       return &types.Tx{
                TxData: *oldTx,
-               Tx:     legacy.MapTx(oldTx),
+               Tx:     types.MapTx(oldTx),
        }
 }