mirror of
https://github.com/aljazceru/lightning.git
synced 2026-01-05 23:24:21 +01:00
pytest: add blackbox tests for reckless
A canned lightningd/plugins is used to test against. This allows faster and more deterministic outcomes. Changelog-None
This commit is contained in:
committed by
Rusty Russell
parent
cf203369bc
commit
2f050621b0
188
tests/test_reckless.py
Normal file
188
tests/test_reckless.py
Normal file
@@ -0,0 +1,188 @@
|
||||
from fixtures import * # noqa: F401,F403
|
||||
import subprocess
|
||||
from pathlib import PosixPath, Path
|
||||
import socket
|
||||
import pytest
|
||||
import os
|
||||
import shutil
|
||||
import time
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def canned_github_server(directory):
|
||||
global NETWORK
|
||||
NETWORK = os.environ.get('TEST_NETWORK')
|
||||
if NETWORK is None:
|
||||
NETWORK = 'regtest'
|
||||
FILE_PATH = Path(os.path.dirname(os.path.realpath(__file__)))
|
||||
if os.environ.get('LIGHTNING_CLI') is None:
|
||||
os.environ['LIGHTNING_CLI'] = str(FILE_PATH.parent / 'cli/lightning-cli')
|
||||
print('LIGHTNING_CALL: ', os.environ.get('LIGHTNING_CLI'))
|
||||
# Use socket to provision a random free port
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
sock.bind(('localhost', 0))
|
||||
free_port = str(sock.getsockname()[1])
|
||||
sock.close()
|
||||
global my_env
|
||||
my_env = os.environ.copy()
|
||||
# This tells reckless to redirect to the canned server rather than github.
|
||||
my_env['REDIR_GITHUB_API'] = f'http://127.0.0.1:{free_port}/api'
|
||||
my_env['REDIR_GITHUB'] = directory
|
||||
my_env['FLASK_RUN_PORT'] = free_port
|
||||
my_env['FLASK_APP'] = str(FILE_PATH / 'rkls_github_canned_server')
|
||||
server = subprocess.Popen(["python3", "-m", "flask", "run"],
|
||||
env=my_env)
|
||||
|
||||
# Generate test plugin repository to test reckless against.
|
||||
repo_dir = os.path.join(directory, "lightningd")
|
||||
os.mkdir(repo_dir, 0o777)
|
||||
plugins_path = str(FILE_PATH / 'data/recklessrepo/lightningd')
|
||||
# This lets us temporarily set .gitconfig user info in order to commit
|
||||
my_env['HOME'] = directory
|
||||
with open(os.path.join(directory, '.gitconfig'), 'w') as conf:
|
||||
conf.write(("[user]\n"
|
||||
"\temail = reckless@example.com\n"
|
||||
"\tname = reckless CI\n"
|
||||
"\t[init]\n"
|
||||
"\tdefaultBranch = master"))
|
||||
|
||||
with open(os.path.join(directory, '.gitconfig'), 'r') as conf:
|
||||
print(conf.readlines())
|
||||
|
||||
# Bare repository must be initialized prior to setting other git env vars
|
||||
subprocess.check_output(['git', 'init', '--bare', 'plugins'], cwd=repo_dir,
|
||||
env=my_env)
|
||||
|
||||
my_env['GIT_DIR'] = os.path.join(repo_dir, 'plugins')
|
||||
my_env['GIT_WORK_TREE'] = repo_dir
|
||||
my_env['GIT_INDEX_FILE'] = os.path.join(repo_dir, 'scratch-index')
|
||||
repo_initialization = (f'cp -r {plugins_path}/* .;'
|
||||
'git add --all;'
|
||||
'git commit -m "initial commit - autogenerated by test_reckless.py";')
|
||||
subprocess.check_output([repo_initialization], env=my_env, shell=True,
|
||||
cwd=repo_dir)
|
||||
del my_env['HOME']
|
||||
del my_env['GIT_DIR']
|
||||
del my_env['GIT_WORK_TREE']
|
||||
del my_env['GIT_INDEX_FILE']
|
||||
# We also need the github api data for the repo which will be served via http
|
||||
shutil.copyfile(str(FILE_PATH / 'data/recklessrepo/rkls_api_lightningd_plugins.json'), os.path.join(directory, 'rkls_api_lightningd_plugins.json'))
|
||||
yield
|
||||
server.terminate()
|
||||
|
||||
|
||||
def reckless(cmds: list, dir: PosixPath = None,
|
||||
autoconfirm=True, timeout: int = 15):
|
||||
'''Call the reckless executable, optionally with a directory.'''
|
||||
if dir is not None:
|
||||
cmds.insert(0, "-l")
|
||||
cmds.insert(1, str(dir))
|
||||
cmds.insert(0, "tools/reckless")
|
||||
r = subprocess.run(cmds, capture_output=True, encoding='utf-8', env=my_env,
|
||||
input='Y\n')
|
||||
print(" ".join(r.args), "\n")
|
||||
print("***RECKLESS STDOUT***")
|
||||
for l in r.stdout.splitlines():
|
||||
print(l)
|
||||
print('\n')
|
||||
print("***RECKLESS STDERR***")
|
||||
for l in r.stderr.splitlines():
|
||||
print(l)
|
||||
print('\n')
|
||||
return r
|
||||
|
||||
|
||||
def get_reckless_node(node_factory):
|
||||
'''This may be unnecessary, but a preconfigured lightning dir
|
||||
is useful for reckless testing.'''
|
||||
node = node_factory.get_node(options={}, start=False)
|
||||
return node
|
||||
|
||||
|
||||
def check_stderr(stderr):
|
||||
def output_okay(out):
|
||||
for warning in ['[notice]', 'npm WARN', 'npm notice']:
|
||||
if out.startswith(warning):
|
||||
return True
|
||||
return False
|
||||
for e in stderr.splitlines():
|
||||
if len(e) < 1:
|
||||
continue
|
||||
# Don't err on verbosity from pip, npm
|
||||
assert output_okay(e)
|
||||
|
||||
|
||||
def test_basic_help():
|
||||
'''Validate that argparse provides basic help info.
|
||||
This requires no config options passed to reckless.'''
|
||||
r = reckless(["-h"])
|
||||
assert r.returncode == 0
|
||||
assert "positional arguments:" in r.stdout.splitlines()
|
||||
assert "options:" in r.stdout.splitlines() or "optional arguments:" in r.stdout.splitlines()
|
||||
|
||||
|
||||
def test_contextual_help(node_factory):
|
||||
n = get_reckless_node(node_factory)
|
||||
for subcmd in ['install', 'uninstall', 'search',
|
||||
'enable', 'disable', 'source']:
|
||||
r = reckless([subcmd, "-h"], dir=n.lightning_dir)
|
||||
assert r.returncode == 0
|
||||
assert "positional arguments:" in r.stdout.splitlines()
|
||||
|
||||
|
||||
def test_sources(node_factory):
|
||||
"""add additional sources and search through them"""
|
||||
n = get_reckless_node(node_factory)
|
||||
r = reckless(["source", "-h"], dir=n.lightning_dir)
|
||||
assert r.returncode == 0
|
||||
|
||||
|
||||
def test_search(node_factory):
|
||||
"""add additional sources and search through them"""
|
||||
n = get_reckless_node(node_factory)
|
||||
r = reckless([f"--network={NETWORK}", "search", "testplugpass"], dir=n.lightning_dir)
|
||||
assert r.returncode == 0
|
||||
assert 'found testplugpass in repo: https://github.com/lightningd/plugins' in r.stdout
|
||||
|
||||
|
||||
def test_install(node_factory):
|
||||
"""test search, git clone, and installation to folder."""
|
||||
n = get_reckless_node(node_factory)
|
||||
r = reckless([f"--network={NETWORK}", "-v", "install", "testplugpass"], dir=n.lightning_dir)
|
||||
assert r.returncode == 0
|
||||
assert 'dependencies installed successfully' in r.stdout
|
||||
assert 'plugin installed:' in r.stdout
|
||||
assert 'testplugpass enabled' in r.stdout
|
||||
check_stderr(r.stderr)
|
||||
plugin_path = Path(n.lightning_dir) / 'reckless/testplugpass'
|
||||
print(plugin_path)
|
||||
assert os.path.exists(plugin_path)
|
||||
|
||||
|
||||
def test_disable_enable(node_factory):
|
||||
"""test search, git clone, and installation to folder."""
|
||||
n = get_reckless_node(node_factory)
|
||||
r = reckless([f"--network={NETWORK}", "-v", "install", "testplugpass"],
|
||||
dir=n.lightning_dir)
|
||||
assert r.returncode == 0
|
||||
assert 'dependencies installed successfully' in r.stdout
|
||||
assert 'plugin installed:' in r.stdout
|
||||
assert 'testplugpass enabled' in r.stdout
|
||||
check_stderr(r.stderr)
|
||||
plugin_path = Path(n.lightning_dir) / 'reckless/testplugpass'
|
||||
print(plugin_path)
|
||||
assert os.path.exists(plugin_path)
|
||||
r = reckless([f"--network={NETWORK}", "-v", "disable", "testplugpass"],
|
||||
dir=n.lightning_dir)
|
||||
assert r.returncode == 0
|
||||
n.start()
|
||||
# Should find it with or without the file extension
|
||||
r = reckless([f"--network={NETWORK}", "-v", "enable", "testplugpass.py"],
|
||||
dir=n.lightning_dir)
|
||||
assert r.returncode == 0
|
||||
assert 'testplugpass.py enabled' in r.stdout
|
||||
test_plugin = {'name': str(plugin_path / 'testplugpass.py'),
|
||||
'active': True, 'dynamic': True}
|
||||
time.sleep(1)
|
||||
print(n.rpc.plugin_list()['plugins'])
|
||||
assert(test_plugin in n.rpc.plugin_list()['plugins'])
|
||||
Reference in New Issue
Block a user