You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
73 lines
2.0 KiB
73 lines
2.0 KiB
import os |
|
import sys |
|
import unittest |
|
import subprocess |
|
|
|
class TestLightning(unittest.TestCase): |
|
|
|
@staticmethod |
|
def run_shell(args, timeout=30): |
|
process = subprocess.Popen(['electrum/tests/regtest/regtest.sh'] + args, stderr=subprocess.STDOUT, stdout=subprocess.PIPE, universal_newlines=True) |
|
for line in iter(process.stdout.readline, ''): |
|
sys.stdout.write(line) |
|
sys.stdout.flush() |
|
process.wait(timeout=timeout) |
|
process.stdout.close() |
|
assert process.returncode == 0 |
|
|
|
def setUp(self): |
|
test_name = self.id().split('.')[-1] |
|
sys.stdout.write("***** %s ******\n" % test_name) |
|
# initialize and get funds |
|
for agent in self.agents: |
|
self.run_shell(['init', agent]) |
|
# mine a block so that funds are confirmed |
|
self.run_shell(['new_block']) |
|
# extra configuration (optional) |
|
self.run_shell(['configure_' + test_name]) |
|
# start daemons |
|
for agent in self.agents: |
|
self.run_shell(['start', agent]) |
|
|
|
def tearDown(self): |
|
for agent in self.agents: |
|
self.run_shell(['stop', agent]) |
|
|
|
|
|
class TestUnixSockets(TestLightning): |
|
agents = [] |
|
|
|
def test_unixsockets(self): |
|
self.run_shell(['unixsockets']) |
|
|
|
|
|
class TestLightningAB(TestLightning): |
|
agents = ['alice', 'bob'] |
|
|
|
def test_collaborative_close(self): |
|
self.run_shell(['collaborative_close']) |
|
|
|
def test_backup(self): |
|
self.run_shell(['backup']) |
|
|
|
def test_breach(self): |
|
self.run_shell(['breach']) |
|
|
|
def test_extract_preimage(self): |
|
self.run_shell(['extract_preimage']) |
|
|
|
def test_redeem_htlcs(self): |
|
self.run_shell(['redeem_htlcs']) |
|
|
|
def test_breach_with_unspent_htlc(self): |
|
self.run_shell(['breach_with_unspent_htlc']) |
|
|
|
def test_breach_with_spent_htlc(self): |
|
self.run_shell(['breach_with_spent_htlc']) |
|
|
|
|
|
class TestLightningABC(TestLightning): |
|
agents = ['alice', 'bob', 'carol'] |
|
|
|
def test_watchtower(self): |
|
self.run_shell(['watchtower'])
|
|
|