mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2024-11-19 09:53:54 +01:00
Code Issues Projects Releases Wiki Activity

test: modify waitForTxInMempool to expect only one tx in the mempool

Browse Source
This commit is contained in:
Wilmer Paulino 2018-05-24 17:57:20 -04:00
parent d8325e29fd
commit 11c315cffe
No known key found for this signature in database
GPG Key ID: 6DF57B9F9514972F
1 changed files with 15 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
lnd_test.go
View File

@ -4009,38 +4009,23 @@ func copyFile(dest, src string) error {
return d.Close() return d.Close()
} }
// waitForTxInMempool polls until finding one transaction in the provided
// miner's mempool. An error is returned if *one* transaction isn't found within
// the given timeout.
func waitForTxInMempool(miner *rpcclient.Client, func waitForTxInMempool(miner *rpcclient.Client,
timeout time.Duration) (*chainhash.Hash, error) { timeout time.Duration) (*chainhash.Hash, error) {
var txid *chainhash.Hash txs, err := waitForNTxsInMempool(miner, 1, timeout)
breakTimeout := time.After(timeout) if err != nil {
ticker := time.NewTicker(50 * time.Millisecond) return nil, err
defer ticker.Stop()
poll:
for {
select {
case <-breakTimeout:
return nil, errors.New("no tx found in mempool")
case <-ticker.C:
mempool, err := miner.GetRawMempool()
if err != nil {
return nil, err
}
if len(mempool) == 0 {
continue
}
txid = mempool[0]
break poll
}
} }
return txid, nil
return txs[0], err
} }
// waitForNTxsInMempool polls until finding the desired number of transactions // waitForNTxsInMempool polls until finding the desired number of transactions
// in the provided miner's mempool. An error is returned if the this number is // in the provided miner's mempool. An error is returned if this number is not
// not met after the given timeout. // met after the given timeout.
func waitForNTxsInMempool(miner *rpcclient.Client, n int, func waitForNTxsInMempool(miner *rpcclient.Client, n int,
timeout time.Duration) ([]*chainhash.Hash, error) { timeout time.Duration) ([]*chainhash.Hash, error) {
@ -4054,7 +4039,7 @@ func waitForNTxsInMempool(miner *rpcclient.Client, n int,
select { select {
case <-breakTimeout: case <-breakTimeout:
return nil, fmt.Errorf("wanted %v, found %v txs "+ return nil, fmt.Errorf("wanted %v, found %v txs "+
"in mempool", n, len(mempool)) "in mempool: %v", n, len(mempool), mempool)
case <-ticker.C: case <-ticker.C:
mempool, err = miner.GetRawMempool() mempool, err = miner.GetRawMempool()
if err != nil { if err != nil {
@ -6441,7 +6426,7 @@ func testMultiHopHtlcLocalTimeout(net *lntest.NetworkHarness, t *harnessTest) {
} }
_, err = waitForTxInMempool(net.Miner.Node, time.Second*10) _, err = waitForTxInMempool(net.Miner.Node, time.Second*10)
if err != nil { if err != nil {
t.Fatalf("unable to find bob's sweeping transaction") t.Fatalf("unable to find bob's sweeping transaction: %v", err)
} }
// Next, we'll mine a final block that should confirm the second-layer // Next, we'll mine a final block that should confirm the second-layer
@ -6669,7 +6654,7 @@ func testMultiHopReceiverChainClaim(net *lntest.NetworkHarness, t *harnessTest)
// We should have a new transaction in the mempool. // We should have a new transaction in the mempool.
_, err = waitForTxInMempool(net.Miner.Node, time.Second*10) _, err = waitForTxInMempool(net.Miner.Node, time.Second*10)
if err != nil { if err != nil {
t.Fatalf("unable to find bob's sweeping transaction") t.Fatalf("unable to find bob's sweeping transaction: %v", err)
} }
// Finally, if we mine an additional block to confirm these two sweep // Finally, if we mine an additional block to confirm these two sweep
@ -7355,7 +7340,7 @@ func testMultiHopHtlcLocalChainClaim(net *lntest.NetworkHarness, t *harnessTest)
_, err = waitForTxInMempool(net.Miner.Node, time.Second*10) _, err = waitForTxInMempool(net.Miner.Node, time.Second*10)
if err != nil { if err != nil {
t.Fatalf("unable to find bob's sweeping transaction") t.Fatalf("unable to find bob's sweeping transaction: %v", err)
} }
// At this point, Bob should detect that he has no pending channels // At this point, Bob should detect that he has no pending channels
@ -7542,7 +7527,7 @@ func testMultiHopHtlcRemoteChainClaim(net *lntest.NetworkHarness, t *harnessTest
_, err = waitForNTxsInMempool(net.Miner.Node, 1, time.Second*15) _, err = waitForNTxsInMempool(net.Miner.Node, 1, time.Second*15)
if err != nil { if err != nil {
t.Fatalf("unable to find bob's sweeping transaction") t.Fatalf("unable to find bob's sweeping transaction: %v", err)
} }
// We'll now mine another block, this should confirm the sweep // We'll now mine another block, this should confirm the sweep