Browse Source

pytest: Move wait_for to utils.py and make fund_channel a member

Slowly moving towards a more pythonic approach to the testing code.

Signed-off-by: Christian Decker <decker.christian@gmail.com>
ppa-0.6.1
Christian Decker 7 years ago
committed by Rusty Russell
parent
commit
81b715aae9
  1. 36
      tests/test_lightningd.py
  2. 40
      tests/utils.py

36
tests/test_lightningd.py

@ -1,5 +1,6 @@
from concurrent import futures from concurrent import futures
from decimal import Decimal from decimal import Decimal
from utils import wait_for
import copy import copy
import json import json
@ -66,14 +67,6 @@ def setupBitcoind(directory):
bitcoind.generate_block(1) bitcoind.generate_block(1)
def wait_for(success, timeout=30, interval=0.1):
start_time = time.time()
while not success() and time.time() < start_time + timeout:
time.sleep(interval)
if time.time() > start_time + timeout:
raise ValueError("Error waiting for {}", success)
def wait_forget_channels(node): def wait_forget_channels(node):
"""This node is closing all of its channels, check we are forgetting them """This node is closing all of its channels, check we are forgetting them
""" """
@ -307,31 +300,7 @@ class LightningDTests(BaseLightningDTests):
# Returns the short channel-id: <blocknum>:<txnum>:<outnum> # Returns the short channel-id: <blocknum>:<txnum>:<outnum>
def fund_channel(self, l1, l2, amount): def fund_channel(self, l1, l2, amount):
# Generates a block, so we know next tx will be first in block. return l1.fund_channel(l2, amount)
self.give_funds(l1, amount + 1000000)
num_tx = len(l1.bitcoin.rpc.getrawmempool())
tx = l1.rpc.fundchannel(l2.info['id'], amount)['tx']
# Technically, this is async to fundchannel.
l1.daemon.wait_for_log('sendrawtx exit 0')
wait_for(lambda: len(l1.bitcoin.rpc.getrawmempool()) == num_tx + 1)
l1.bitcoin.generate_block(1)
# We wait until gossipd sees local update, as well as status NORMAL,
# so it can definitely route through.
l1.daemon.wait_for_logs(['update for channel .* now ACTIVE', 'to CHANNELD_NORMAL'])
l2.daemon.wait_for_logs(['update for channel .* now ACTIVE', 'to CHANNELD_NORMAL'])
# Hacky way to find our output.
decoded = bitcoind.rpc.decoderawtransaction(tx)
for out in decoded['vout']:
# Sometimes a float? Sometimes a decimal? WTF Python?!
if out['scriptPubKey']['type'] == 'witness_v0_scripthash':
if out['value'] == Decimal(amount) / 10**8 or out['value'] * 10**8 == amount:
return "{}:1:{}".format(bitcoind.rpc.getblockcount(), out['n'])
# Intermittent decoding failure. See if it decodes badly twice?
decoded2 = bitcoind.rpc.decoderawtransaction(tx)
raise ValueError("Can't find {} payment in {} (1={} 2={})".format(amount, tx, decoded, decoded2))
def pay(self, lsrc, ldst, amt, label=None, async=False): def pay(self, lsrc, ldst, amt, label=None, async=False):
if not label: if not label:
@ -731,7 +700,6 @@ class LightningDTests(BaseLightningDTests):
def test_balance(self): def test_balance(self):
l1, l2 = self.connect() l1, l2 = self.connect()
self.fund_channel(l1, l2, 10**6) self.fund_channel(l1, l2, 10**6)
p1 = l1.rpc.getpeer(peer_id=l2.info['id'], level='info')['channels'][0] p1 = l1.rpc.getpeer(peer_id=l2.info['id'], level='info')['channels'][0]
p2 = l2.rpc.getpeer(l1.info['id'], 'info')['channels'][0] p2 = l2.rpc.getpeer(l1.info['id'], 'info')['channels'][0]

40
tests/utils.py

@ -7,6 +7,7 @@ import threading
import time import time
from bitcoin.rpc import RawProxy as BitcoinProxy from bitcoin.rpc import RawProxy as BitcoinProxy
from decimal import Decimal
BITCOIND_CONFIG = { BITCOIND_CONFIG = {
@ -28,6 +29,14 @@ LIGHTNINGD_CONFIG = {
DEVELOPER = os.getenv("DEVELOPER", "0") == "1" DEVELOPER = os.getenv("DEVELOPER", "0") == "1"
def wait_for(success, timeout=30, interval=0.1):
start_time = time.time()
while not success() and time.time() < start_time + timeout:
time.sleep(interval)
if time.time() > start_time + timeout:
raise ValueError("Error waiting for {}", success)
def write_config(filename, opts): def write_config(filename, opts):
with open(filename, 'w') as f: with open(filename, 'w') as f:
for k, v in opts.items(): for k, v in opts.items():
@ -425,3 +434,34 @@ class LightningNode(object):
self.daemon.stop() self.daemon.stop()
self.daemon.start() self.daemon.start()
def fund_channel(self, l2, amount):
# Give yourself some funds to work with
addr = self.rpc.newaddr()['address']
self.bitcoin.rpc.sendtoaddress(addr, (amount + 1000000) / 10**8)
numfunds = len(self.rpc.listfunds()['outputs'])
self.bitcoin.generate_block(1)
wait_for(lambda: len(self.rpc.listfunds()['outputs']) > numfunds)
# Now go ahead and open a channel
num_tx = len(self.bitcoin.rpc.getrawmempool())
tx = self.rpc.fundchannel(l2.info['id'], amount)['tx']
wait_for(lambda: len(self.bitcoin.rpc.getrawmempool()) == num_tx + 1)
self.bitcoin.generate_block(1)
# We wait until gossipd sees local update, as well as status NORMAL,
# so it can definitely route through.
self.daemon.wait_for_logs(['update for channel .* now ACTIVE', 'to CHANNELD_NORMAL'])
l2.daemon.wait_for_logs(['update for channel .* now ACTIVE', 'to CHANNELD_NORMAL'])
# Hacky way to find our output.
decoded = self.bitcoin.rpc.decoderawtransaction(tx)
for out in decoded['vout']:
# Sometimes a float? Sometimes a decimal? WTF Python?!
if out['scriptPubKey']['type'] == 'witness_v0_scripthash':
if out['value'] == Decimal(amount) / 10**8 or out['value'] * 10**8 == amount:
return "{}:1:{}".format(self.bitcoin.rpc.getblockcount(), out['n'])
# Intermittent decoding failure. See if it decodes badly twice?
decoded2 = self.bitcoin.rpc.decoderawtransaction(tx)
raise ValueError("Can't find {} payment in {} (1={} 2={})".format(amount, tx, decoded, decoded2))

Loading…
Cancel
Save