Browse Source
This is the first example of the py.test style fixtures which should allow us to write much cleaner and nicer tests. Signed-off-by: Christian Decker <decker.christian@gmail.com>ppa-0.6.1
Christian Decker
7 years ago
committed by
Rusty Russell
Rusty Russell
4 changed files with 207 additions and 54 deletions
@ -0,0 +1,144 @@ |
|||
from concurrent import futures |
|||
from test_lightningd import NodeFactory |
|||
|
|||
import logging |
|||
import os |
|||
import pytest |
|||
import re |
|||
import tempfile |
|||
import utils |
|||
|
|||
|
|||
TEST_DIR = tempfile.mkdtemp(prefix='ltests-') |
|||
VALGRIND = os.getenv("NO_VALGRIND", "0") == "0" |
|||
DEVELOPER = os.getenv("DEVELOPER", "0") == "1" |
|||
TEST_DEBUG = os.getenv("TEST_DEBUG", "0") == "1" |
|||
|
|||
|
|||
@pytest.fixture |
|||
def directory(test_name): |
|||
"""Return a per-test specific directory |
|||
""" |
|||
global TEST_DIR |
|||
yield os.path.join(TEST_DIR, test_name) |
|||
|
|||
|
|||
@pytest.fixture |
|||
def test_name(request): |
|||
yield request.function.__name__ |
|||
|
|||
|
|||
@pytest.fixture |
|||
def bitcoind(directory): |
|||
bitcoind = utils.BitcoinD(bitcoin_dir=directory, rpcport=28332) |
|||
try: |
|||
bitcoind.start() |
|||
except Exception: |
|||
bitcoind.stop() |
|||
raise |
|||
|
|||
info = bitcoind.rpc.getnetworkinfo() |
|||
|
|||
if info['version'] < 160000: |
|||
bitcoind.rpc.stop() |
|||
raise ValueError("bitcoind is too old. At least version 16000 (v0.16.0)" |
|||
" is needed, current version is {}".format(info['version'])) |
|||
|
|||
info = bitcoind.rpc.getblockchaininfo() |
|||
# Make sure we have some spendable funds |
|||
if info['blocks'] < 101: |
|||
bitcoind.generate_block(101 - info['blocks']) |
|||
elif bitcoind.rpc.getwalletinfo()['balance'] < 1: |
|||
logging.debug("Insufficient balance, generating 1 block") |
|||
bitcoind.generate_block(1) |
|||
|
|||
yield bitcoind |
|||
|
|||
try: |
|||
bitcoind.rpc.stop() |
|||
except Exception: |
|||
bitcoind.proc.kill() |
|||
bitcoind.proc.wait() |
|||
|
|||
|
|||
@pytest.fixture |
|||
def node_factory(directory, test_name, bitcoind, executor): |
|||
nf = NodeFactory(test_name, bitcoind, executor, directory=directory) |
|||
yield nf |
|||
err_count = 0 |
|||
ok = nf.killall([not n.may_fail for n in nf.nodes]) |
|||
if VALGRIND: |
|||
for node in nf.nodes: |
|||
err_count += printValgrindErrors(node) |
|||
if err_count: |
|||
raise ValueError("{} nodes reported valgrind errors".format(err_count)) |
|||
|
|||
for node in nf.nodes: |
|||
err_count += printCrashLog(node) |
|||
if err_count: |
|||
raise ValueError("{} nodes had crash.log files".format(err_count)) |
|||
for node in nf.nodes: |
|||
err_count += checkReconnect(node) |
|||
if err_count: |
|||
raise ValueError("{} nodes had unexpected reconnections".format(err_count)) |
|||
|
|||
if not ok: |
|||
raise Exception("At least one lightning exited with unexpected non-zero return code") |
|||
|
|||
|
|||
def getValgrindErrors(node): |
|||
for error_file in os.listdir(node.daemon.lightning_dir): |
|||
if not re.fullmatch("valgrind-errors.\d+", error_file): |
|||
continue |
|||
with open(os.path.join(node.daemon.lightning_dir, error_file), 'r') as f: |
|||
errors = f.read().strip() |
|||
if errors: |
|||
return errors, error_file |
|||
return None, None |
|||
|
|||
|
|||
def printValgrindErrors(node): |
|||
errors, fname = getValgrindErrors(node) |
|||
if errors: |
|||
print("-" * 31, "Valgrind errors", "-" * 32) |
|||
print("Valgrind error file:", fname) |
|||
print(errors) |
|||
print("-" * 80) |
|||
return 1 if errors else 0 |
|||
|
|||
|
|||
def getCrashLog(node): |
|||
if node.may_fail: |
|||
return None, None |
|||
try: |
|||
crashlog = os.path.join(node.daemon.lightning_dir, 'crash.log') |
|||
with open(crashlog, 'r') as f: |
|||
return f.readlines(), crashlog |
|||
except Exception: |
|||
return None, None |
|||
|
|||
|
|||
def printCrashLog(node): |
|||
errors, fname = getCrashLog(node) |
|||
if errors: |
|||
print("-" * 10, "{} (last 50 lines)".format(fname), "-" * 10) |
|||
for l in errors[-50:]: |
|||
print(l, end='') |
|||
print("-" * 80) |
|||
return 1 if errors else 0 |
|||
|
|||
|
|||
def checkReconnect(node): |
|||
# Without DEVELOPER, we can't suppress reconnection. |
|||
if node.may_reconnect or not DEVELOPER: |
|||
return 0 |
|||
if node.daemon.is_in_log('Peer has reconnected'): |
|||
return 1 |
|||
return 0 |
|||
|
|||
|
|||
@pytest.fixture |
|||
def executor(): |
|||
ex = futures.ThreadPoolExecutor(max_workers=20) |
|||
yield ex |
|||
ex.shutdown(wait=False) |
|||
@ -0,0 +1,60 @@ |
|||
from fixtures import * # noqa: F401,F403 |
|||
from test_lightningd import wait_for |
|||
|
|||
import os |
|||
import time |
|||
import unittest |
|||
|
|||
|
|||
DEVELOPER = os.getenv("DEVELOPER", "0") == "1" |
|||
|
|||
|
|||
@unittest.skipIf(not DEVELOPER, "needs DEVELOPER=1 for --dev-broadcast-interval") |
|||
def test_gossip_pruning(node_factory, bitcoind): |
|||
""" Create channel and see it being updated in time before pruning |
|||
""" |
|||
opts = {'channel-update-interval': 5} |
|||
l1, l2, l3 = node_factory.get_nodes(3, opts) |
|||
|
|||
l1.rpc.connect(l2.info['id'], 'localhost', l2.info['port']) |
|||
l2.rpc.connect(l3.info['id'], 'localhost', l3.info['port']) |
|||
|
|||
scid1 = l1.fund_channel(l2, 10**6) |
|||
scid2 = l2.fund_channel(l3, 10**6) |
|||
|
|||
bitcoind.rpc.generate(6) |
|||
|
|||
# Channels should be activated locally |
|||
wait_for(lambda: [c['active'] for c in l1.rpc.listchannels()['channels']] == [True] * 4) |
|||
wait_for(lambda: [c['active'] for c in l2.rpc.listchannels()['channels']] == [True] * 4) |
|||
wait_for(lambda: [c['active'] for c in l3.rpc.listchannels()['channels']] == [True] * 4) |
|||
|
|||
# All of them should send a keepalive message |
|||
l1.daemon.wait_for_logs([ |
|||
'Sending keepalive channel_update for {}'.format(scid1), |
|||
]) |
|||
l2.daemon.wait_for_logs([ |
|||
'Sending keepalive channel_update for {}'.format(scid1), |
|||
'Sending keepalive channel_update for {}'.format(scid2), |
|||
]) |
|||
l3.daemon.wait_for_logs([ |
|||
'Sending keepalive channel_update for {}'.format(scid2), |
|||
]) |
|||
|
|||
# Now kill l3, so that l2 and l1 can prune it from their view after 10 seconds |
|||
|
|||
# FIXME: This sleep() masks a real bug: that channeld sends a |
|||
# channel_update message (to disable the channel) with same |
|||
# timestamp as the last keepalive, and thus is ignored. The minimal |
|||
# fix is to backdate the keepalives 1 second, but maybe we should |
|||
# simply have gossipd generate all updates? |
|||
time.sleep(1) |
|||
l3.stop() |
|||
|
|||
l1.daemon.wait_for_log("Pruning channel {} from network view".format(scid2)) |
|||
l2.daemon.wait_for_log("Pruning channel {} from network view".format(scid2)) |
|||
|
|||
assert scid2 not in [c['short_channel_id'] for c in l1.rpc.listchannels()['channels']] |
|||
assert scid2 not in [c['short_channel_id'] for c in l2.rpc.listchannels()['channels']] |
|||
assert l3.info['id'] not in [n['nodeid'] for n in l1.rpc.listnodes()['nodes']] |
|||
assert l3.info['id'] not in [n['nodeid'] for n in l2.rpc.listnodes()['nodes']] |
|||
Loading…
Reference in new issue