mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-03-13 11:09:23 +01:00
Code Issues Projects Releases Wiki Activity

lnwallet: adapt tests to field changes

Browse source
This commit is contained in:
Olaoluwa Osuntokun 2015-12-18 21:47:47 -06:00
parent 420e34cc91
commit 29d381ad21
1 changed files with 26 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
lnwallet/wallet_test.go
View file

@ -294,14 +294,19 @@ func TestBasicWalletReservationWorkFlow(t *testing.T) {
// The channel reservation should now be populated with a multi-sig key // The channel reservation should now be populated with a multi-sig key
// from our HD chain, a change output with 3 BTC, and 2 outputs selected // from our HD chain, a change output with 3 BTC, and 2 outputs selected
// of 4 BTC each. // of 4 BTC each.
ourInputs, ourChange, ourMultsigKey := chanReservation.OurFunds() ourInputs, ourChange := chanReservation.OurFunds()
if len(ourInputs) != 2 { if len(ourInputs) != 2 {
t.Fatalf("outputs for funding tx not properly selected, have %v "+ t.Fatalf("outputs for funding tx not properly selected, have %v "+
"outputs should have 2", len(ourInputs)) "outputs should have 2", len(ourInputs))
} }
if ourMultsigKey == nil { ourMultiSigKey, ourCommitKey := chanReservation.OurKeys()
if ourMultiSigKey == nil {
t.Fatalf("alice's key for multi-sig not found") t.Fatalf("alice's key for multi-sig not found")
} }
if ourCommitKey == nil {
// TODO(roasbeef): gen commit
// t.Fatalf("alice's key for commit not found")
}
if ourChange[0].Value != 3e8 { if ourChange[0].Value != 3e8 {
t.Fatalf("coin selection failed, change output should be 3e8 "+ t.Fatalf("coin selection failed, change output should be 3e8 "+
"satoshis, is instead %v", ourChange[0].Value) "satoshis, is instead %v", ourChange[0].Value)
@ -315,22 +320,27 @@ func TestBasicWalletReservationWorkFlow(t *testing.T) {
// At this point, the reservation should have our signatures, and a // At this point, the reservation should have our signatures, and a
// partial funding transaction (missing bob's sigs). // partial funding transaction (missing bob's sigs).
if len(chanReservation.OurSigs()) != 2 { if len(chanReservation.OurFundingSigs()) != 2 {
t.Fatalf("only %v of our sigs present, should have 2", t.Fatalf("only %v of our sigs present, should have 2",
len(chanReservation.OurSigs())) len(chanReservation.OurFundingSigs()))
} }
// Additionally, the funding tx should have been populated. // Additionally, the funding tx should have been populated.
if chanReservation.fundingTx == nil { if chanReservation.partialState.fundingTx == nil {
t.Fatalf("funding transaction never created!") t.Fatalf("funding transaction never created!")
} }
// Their funds should also be filled in. // Their funds should also be filled in.
theirInputs, theirChange, theirMultsigKey := chanReservation.TheirFunds() theirInputs, theirChange := chanReservation.TheirFunds()
if len(theirInputs) != 1 { if len(theirInputs) != 1 {
t.Fatalf("bob's outputs for funding tx not properly selected, have %v "+ t.Fatalf("bob's outputs for funding tx not properly selected, have %v "+
"outputs should have 2", len(theirInputs)) "outputs should have 2", len(theirInputs))
} }
if theirMultsigKey == nil { theirMultisigKey, theirCommitKey := chanReservation.TheirKeys()
t.Fatalf("alice's key for multi-sig not found") if theirMultisigKey == nil {
t.Fatalf("bob's key for multi-sig not found")
}
if theirCommitKey == nil {
// TODO(roasbeef): gen commit
// t.Fatalf("alice's key for commit not found")
} }
if theirChange[0].Value != 2e8 { if theirChange[0].Value != 2e8 {
t.Fatalf("bob should have one change output with value 2e8"+ t.Fatalf("bob should have one change output with value 2e8"+
@ -339,7 +349,7 @@ func TestBasicWalletReservationWorkFlow(t *testing.T) {
// Alice responds with her output, change addr, multi-sig key and signatures. // Alice responds with her output, change addr, multi-sig key and signatures.
// Bob then responds with his signatures. // Bob then responds with his signatures.
bobsSigs, err := bobNode.signFundingTx(chanReservation.fundingTx) bobsSigs, err := bobNode.signFundingTx(chanReservation.partialState.fundingTx)
if err != nil { if err != nil {
t.Fatalf("unable to sign inputs for bob: %v", err) t.Fatalf("unable to sign inputs for bob: %v", err)
} }
@ -362,7 +372,7 @@ func TestBasicWalletReservationWorkFlow(t *testing.T) {
// The sub-bucket for this channel, keyed by the txid of the // The sub-bucket for this channel, keyed by the txid of the
// funding transaction // funding transaction
txid := chanReservation.fundingTx.TxSha() txid := chanReservation.partialState.fundingTx.TxSha()
bobChanBucket := openChanBucket.Bucket(txid[:]) bobChanBucket := openChanBucket.Bucket(txid[:])
if bobChanBucket == nil { if bobChanBucket == nil {
t.Fatalf("bucket for the alice-bob channe should exist, " + t.Fatalf("bucket for the alice-bob channe should exist, " +
@ -375,7 +385,7 @@ func TestBasicWalletReservationWorkFlow(t *testing.T) {
// The hash of the stored tx should be indentical to our funding tx. // The hash of the stored tx should be indentical to our funding tx.
storedTxId := storedTx.TxSha() storedTxId := storedTx.TxSha()
actualTxId := chanReservation.fundingTx.TxSha() actualTxId := chanReservation.partialState.fundingTx.TxSha()
if !bytes.Equal(storedTxId.Bytes(), actualTxId.Bytes()) { if !bytes.Equal(storedTxId.Bytes(), actualTxId.Bytes()) {
t.Fatalf("stored funding tx doesn't match actual") t.Fatalf("stored funding tx doesn't match actual")
} }
@ -386,8 +396,8 @@ func TestBasicWalletReservationWorkFlow(t *testing.T) {
// The funding tx should now be valid and complete. // The funding tx should now be valid and complete.
// Check each input and ensure all scripts are fully valid. // Check each input and ensure all scripts are fully valid.
var zeroHash wire.ShaHash var zeroHash wire.ShaHash
fundingTx := chanReservation.fundingTx fundingTx := chanReservation.partialState.fundingTx
for i, input := range chanReservation.fundingTx.TxIn { for i, input := range fundingTx.TxIn {
var pkscript []byte var pkscript []byte
// Bob's txin // Bob's txin
if bytes.Equal(input.PreviousOutPoint.Hash.Bytes(), if bytes.Equal(input.PreviousOutPoint.Hash.Bytes(),
@ -448,7 +458,7 @@ func TestFundingTransactionLockedOutputs(t *testing.T) {
// Neither should have any change, as all our output sizes are // Neither should have any change, as all our output sizes are
// identical (4BTC). // identical (4BTC).
ourInputs1, ourChange1, _ := chanReservation1.OurFunds() ourInputs1, ourChange1 := chanReservation1.OurFunds()
if len(ourInputs1) != 2 { if len(ourInputs1) != 2 {
t.Fatalf("outputs for funding tx not properly selected, has %v "+ t.Fatalf("outputs for funding tx not properly selected, has %v "+
"outputs should have 2", len(ourInputs1)) "outputs should have 2", len(ourInputs1))
@ -457,7 +467,7 @@ func TestFundingTransactionLockedOutputs(t *testing.T) {
t.Fatalf("funding transaction should have no change, instead has %v", t.Fatalf("funding transaction should have no change, instead has %v",
len(ourChange1)) len(ourChange1))
} }
ourInputs2, ourChange2, _ := chanReservation2.OurFunds() ourInputs2, ourChange2 := chanReservation2.OurFunds()
if len(ourInputs2) != 2 { if len(ourInputs2) != 2 {
t.Fatalf("outputs for funding tx not properly selected, have %v "+ t.Fatalf("outputs for funding tx not properly selected, have %v "+
"outputs should have 2", len(ourInputs2)) "outputs should have 2", len(ourInputs2))
@ -561,7 +571,7 @@ func TestCancelNonExistantReservation(t *testing.T) {
// Attempt to cancel this reservation. This should fail, we know // Attempt to cancel this reservation. This should fail, we know
// nothing of it. // nothing of it.
if err = res.Cancel(); err == nil { if err := res.Cancel(); err == nil {
t.Fatalf("cancelled non-existant reservation") t.Fatalf("cancelled non-existant reservation")
} }
} }