aljaz/plugins
1
0
Fork 0
mirror of https://github.com/aljazceru/plugins.git synced 2025-12-21 15:14:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

drain: adds __init__.py and cleanups flake8 nits

Browse Source
This commit is contained in:
Michael Schmoock
2020-11-30 10:02:17 +01:00
committed by Christian Decker
parent b7abc6f188
commit 04254693ab
4 changed files with 37 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
drain/__init__.py Normal file
View File

4
drain/drain.py
View File

@@ -1,6 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from pyln.client import Plugin, Millisatoshi, RpcError from pyln.client import Plugin, Millisatoshi, RpcError
from utils import * from utils import get_ours, wait_ours
import re import re
import time import time
import uuid import uuid
@@ -157,7 +157,7 @@ def test_or_set_chunks(plugin, payload):
continue continue
try: try:
spend, recv = spendable_from_scid(plugin, payload, channel['short_channel_id'], True) spend, recv = spendable_from_scid(plugin, payload, channel['short_channel_id'], True)
except RpcError as e: except RpcError:
continue continue
channels[channel['short_channel_id']] = { channels[channel['short_channel_id']] = {
'spendable': spend, 'spendable': spend,

8
drain/test_drain.py
View File

@@ -1,14 +1,16 @@
from pyln.testing.fixtures import * from pyln.testing.fixtures import * # noqa: F401,F403
from pyln.testing.utils import DEVELOPER from pyln.testing.utils import DEVELOPER
from pyln.client import RpcError, Millisatoshi from pyln.client import RpcError
from utils import * from .utils import get_ours, get_theirs, wait_ours, wait_for_all_htlcs
import os import os
import unittest import unittest
import pytest
pluginopt = {'plugin': os.path.join(os.path.dirname(__file__), "drain.py")} pluginopt = {'plugin': os.path.join(os.path.dirname(__file__), "drain.py")}
EXPERIMENTAL_FEATURES = int(os.environ.get("EXPERIMENTAL_FEATURES", "0")) EXPERIMENTAL_FEATURES = int(os.environ.get("EXPERIMENTAL_FEATURES", "0"))
@unittest.skipIf(not DEVELOPER, "slow gossip, needs DEVELOPER=1") @unittest.skipIf(not DEVELOPER, "slow gossip, needs DEVELOPER=1")
def test_drain_and_refill(node_factory, bitcoind): def test_drain_and_refill(node_factory, bitcoind):
# Scenario: first drain then refill # Scenario: first drain then refill