X-Git-Url: http://git.osdn.net/view?a=blobdiff_plain;f=wallet%2Frecovery_test.go;h=0b6fb56d177683aafe1bafde263ba3492e3ed1b4;hb=229d9e2c39150278f9e1de83aaab64409e338e8f;hp=5c63c2166d296ac9cc9efa0fb4d573b8ff1547c6;hpb=db158dcf09436b003defd333f1a665e7e051d820;p=bytom%2Fvapor.git

diff --git a/wallet/recovery_test.go b/wallet/recovery_test.go
index 5c63c216..0b6fb56d 100644
--- a/wallet/recovery_test.go
+++ b/wallet/recovery_test.go
@@ -57,9 +57,12 @@ func MockSimpleUtxo(index uint64, assetID *bc.AssetID, amount uint64, ctrlProg *
 	return utxo
 }
 
+func AddInput(sourceID bc.Hash, assetID bc.AssetID, amount uint64, pos uint64, controlProgram []byte) *types.TxInput {
+	return types.NewSpendInput(nil, sourceID, assetID, amount, pos, controlProgram)
+}
+
 func AddTxOutput(assetID bc.AssetID, amount uint64, controlProgram []byte) *types.TxOutput {
-	out := types.NewTxOutput(assetID, amount, controlProgram)
-	return out
+	return types.NewIntraChainOutput(assetID, amount, controlProgram)
 }
 
 func BuildTx(baseUtxo *account.UTXO, signer *signers.Signer) (*txbuilder.Template, error) {
@@ -78,7 +81,9 @@ func BuildTx(baseUtxo *account.UTXO, signer *signers.Signer) (*txbuilder.Templat
 
 func CreateTxBuilder(baseUtxo *account.UTXO, signer *signers.Signer) (*txbuilder.TemplateBuilder, error) {
 	tplBuilder := txbuilder.NewBuilder(time.Now())
+	txInput := AddInput(bc.Hash{V0: 1}, baseUtxo.AssetID, 10000, uint64(1), baseUtxo.ControlProgram)
 	txOutput := AddTxOutput(baseUtxo.AssetID, 100, baseUtxo.ControlProgram)
+	tplBuilder.AddInput(txInput, &txbuilder.SigningInstruction{Position: 0})
 	tplBuilder.AddOutput(txOutput)
 	return tplBuilder, nil
 }