"context"
"time"
- dbm "github.com/tendermint/tmlibs/db"
-
- "github.com/bytom/blockchain/account"
- "github.com/bytom/blockchain/pseudohsm"
- "github.com/bytom/blockchain/txbuilder"
- "github.com/bytom/blockchain/txdb"
- cfg "github.com/bytom/config"
- "github.com/bytom/consensus"
- "github.com/bytom/crypto/ed25519/chainkd"
- "github.com/bytom/protocol"
- "github.com/bytom/protocol/bc"
- "github.com/bytom/protocol/bc/legacy"
- "github.com/bytom/protocol/vm"
+ "github.com/bytom/bytom/account"
+ "github.com/bytom/bytom/blockchain/pseudohsm"
+ "github.com/bytom/bytom/blockchain/txbuilder"
+ "github.com/bytom/bytom/consensus"
+ "github.com/bytom/bytom/crypto/ed25519/chainkd"
+ "github.com/bytom/bytom/database"
+ "github.com/bytom/bytom/event"
+ "github.com/bytom/bytom/protocol"
+ "github.com/bytom/bytom/protocol/bc"
+ "github.com/bytom/bytom/protocol/bc/types"
+ "github.com/bytom/bytom/protocol/vm"
+ dbm "github.com/bytom/bytom/database/leveldb"
)
-// Mock transaction pool
-func MockTxPool() *protocol.TxPool {
- return protocol.NewTxPool()
-}
+const (
+ vmVersion = 1
+ assetVersion = 1
+)
-func MockChain(testDB dbm.DB) (*protocol.Chain, error) {
- store := txdb.NewStore(testDB)
- txPool := MockTxPool()
- genesisBlock := cfg.GenerateGenesisBlock()
- chain, err := protocol.NewChain(genesisBlock.Hash(), store, txPool)
- if err != nil {
- return nil, err
- }
- return chain, nil
+// MockChain mock chain with genesis block
+func MockChain(testDB dbm.DB) (*protocol.Chain, *database.Store, *protocol.TxPool, error) {
+ store := database.NewStore(testDB)
+ dispatcher := event.NewDispatcher()
+ txPool := protocol.NewTxPool(store, dispatcher)
+ chain, err := protocol.NewChain(store, txPool)
+ return chain, store, txPool, err
}
+// MockUTXO mock a utxo
func MockUTXO(controlProg *account.CtrlProgram) *account.UTXO {
utxo := &account.UTXO{}
utxo.OutputID = bc.Hash{V0: 1}
utxo.AccountID = controlProg.AccountID
utxo.Address = controlProg.Address
utxo.ControlProgramIndex = controlProg.KeyIndex
+ utxo.Change = controlProg.Change
return utxo
}
-func MockTx(utxo *account.UTXO, testAccount *account.Account) (*txbuilder.Template, *legacy.TxData, error) {
- txInput, sigInst, err := account.UtxoToInputs(testAccount.Signer, utxo, nil)
+// MockTx mock a tx
+func MockTx(utxo *account.UTXO, testAccount *account.Account) (*txbuilder.Template, *types.TxData, error) {
+ txInput, sigInst, err := account.UtxoToInputs(testAccount.Signer, utxo)
if err != nil {
return nil, nil, err
}
b := txbuilder.NewBuilder(time.Now())
- b.AddInput(txInput, sigInst)
- out := legacy.NewTxOutput(*consensus.BTMAssetID, 100, []byte{byte(vm.OP_FAIL)}, nil)
- b.AddOutput(out)
+ if err := b.AddInput(txInput, sigInst); err != nil {
+ return nil, nil, err
+ }
+ out := types.NewTxOutput(*consensus.BTMAssetID, 100, []byte{byte(vm.OP_FAIL)})
+ if err := b.AddOutput(out); err != nil {
+ return nil, nil, err
+ }
return b.Build()
}
-func MockSign(tpl *txbuilder.Template, hsm *pseudohsm.HSM) error {
- return txbuilder.Sign(nil, tpl, nil, []string{"password", "password"}, func(_ context.Context, xpub chainkd.XPub, path [][]byte, data [32]byte, password string) ([]byte, error) {
- sigBytes, err := hsm.XSign(xpub, path, data[:], password)
- return sigBytes, err
+// MockSign sign a tx
+func MockSign(tpl *txbuilder.Template, hsm *pseudohsm.HSM, password string) (bool, 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 {
+ return false, err
+ }
+ return txbuilder.SignProgress(tpl), nil
}
-// Mock block
+// MockBlock mock a block
func MockBlock() *bc.Block {
return &bc.Block{
BlockHeader: &bc.BlockHeader{Height: 1},