Skip to content

Add sauron tests env fix #16

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 31 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion .ci/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -241,13 +241,14 @@ def run_one(p: Plugin, workflow: str) -> bool:

logging.info(f"Virtualenv at {vpath}")

num_workers = 1 if p.name == "sauron" else 5
cmd = [
str(pytest_path),
"-vvv",
"--timeout=600",
"--timeout-method=thread",
"--color=yes",
"-n=5",
f"-n={num_workers}",
]

logging.info(f"Running `{' '.join(cmd)}` in directory {p.path.resolve()}")
Expand Down
2 changes: 1 addition & 1 deletion poncho
1 change: 1 addition & 0 deletions sauron/.gitignore
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
.venv/
8 changes: 8 additions & 0 deletions sauron/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,14 @@ Here is a fully reptilian example running against [blockstream.info](https://blo
lightningd --mainnet --disable-plugin bcli --plugin $PWD/sauron.py --sauron-api-endpoint https://blockstream.info/api/
```


Here is an example running against [mutinynet.com](https://mutinynet.com/):

```
lightningd --signet --disable-plugin bcli --plugin $PWD/sauron.py --sauron-api-endpoint https://mutinynet.com/api/
```


You can use also proxy your requests through [Tor](https://www.torproject.org/) by
specifying a SOCKS proxy to use with the `--sauron-tor-proxy` startup option, in
the form `address:port`.
Expand Down
2 changes: 1 addition & 1 deletion sauron/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
pyln-client>=23.2
pyln-client>=23.2,<=24.5
requests[socks]>=2.23.0
33 changes: 25 additions & 8 deletions sauron/sauron.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,12 +40,23 @@ def fetch(url):


@plugin.init()
def init(plugin, options, configuration, **kwargs):
plugin.api_endpoint = options["sauron-api-endpoint"]
def init(plugin, options, **kwargs):
plugin.api_endpoint = options.get("sauron-api-endpoint", None)
if not plugin.api_endpoint:
raise SauronError("You need to specify the sauron-api-endpoint option.")
sys.exit(1)

# Testing for Esplora or mempool.space API
try:
# MutinyNet API
feerate_url = "{}/v1/fees/recommended".format(plugin.api_endpoint)
feerate_req = fetch(feerate_url)
assert feerate_req.status_code == 200
plugin.is_mempoolspace = True
except AssertionError:
# Esplora API
plugin.is_mempoolspace = False

if options["sauron-tor-proxy"]:
address, port = options["sauron-tor-proxy"].split(":")
socks5_proxy = "socks5h://{}:{}".format(address, port)
Expand All @@ -55,7 +66,8 @@ def init(plugin, options, configuration, **kwargs):
}
plugin.log("Using proxy {} for requests".format(socks5_proxy))

plugin.log("Sauron plugin initialized")
api = "mempool.space" if plugin.is_mempoolspace else "Esplora"
plugin.log(f"Sauron plugin initialized using {api} API")
plugin.log(sauron_eye)


Expand Down Expand Up @@ -188,12 +200,17 @@ def getutxout(plugin, txid, vout, **kwargs):

@plugin.method("estimatefees")
def estimatefees(plugin, **kwargs):
feerate_url = "{}/fee-estimates".format(plugin.api_endpoint)
if plugin.is_mempoolspace:
# MutinyNet API
feerate_url = "{}/v1/fees/recommended".format(plugin.api_endpoint)
else:
# Blockstream API
feerate_url = "{}/fee-estimates".format(plugin.api_endpoint)

feerate_req = fetch(feerate_url)
assert feerate_req.status_code == 200
feerates = feerate_req.json()
if plugin.sauron_network == "test" or plugin.sauron_network == "signet":
if plugin.sauron_network in ["test", "signet"]:
# FIXME: remove the hack if the test API is "fixed"
feerate = feerates.get("144", 1)
slow = normal = urgent = very_urgent = int(feerate * 10**3)
Expand All @@ -204,7 +221,7 @@ def estimatefees(plugin, **kwargs):
urgent = int(feerates["6"] * 10**3)
very_urgent = int(feerates["2"] * 10**3)

feerate_floor = int(feerates["1008"] * 10**3)
feerate_floor = int(feerates.get("1008", slow) * 10**3)
feerates = [
{"blocks": 2, "feerate": very_urgent},
{"blocks": 6, "feerate": urgent},
Expand All @@ -229,15 +246,15 @@ def estimatefees(plugin, **kwargs):
plugin.add_option(
"sauron-api-endpoint",
"",
"The URL of the esplora instance to hit (including '/api').",
"The URL of the esplora or mempool.space instance to hit (including '/api').",
)

plugin.add_option(
"sauron-tor-proxy",
"",
"Tor's SocksPort address in the form address:port, don't specify the"
" protocol. If you didn't modify your torrc you want to put"
"'localhost:9050' here.",
" 'localhost:9050' here.",
)


Expand Down
159 changes: 159 additions & 0 deletions sauron/tests/test_sauron_esplora_bitcoin.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,159 @@
#!/usr/bin/python

import os

import pyln
import pytest
from pyln.testing import utils
from pyln.testing.fixtures import * # noqa: F403
from util import LightningD

pyln.testing.fixtures.network_daemons["bitcoin"] = utils.BitcoinD


class LightningNode(utils.LightningNode):
def __init__(self, *args, **kwargs):
pyln.testing.utils.TEST_NETWORK = "bitcoin"
utils.LightningNode.__init__(self, *args, **kwargs)
lightning_dir = args[1]

self.daemon = LightningD(lightning_dir, None) # noqa: F405
options = {
"disable-plugin": "bcli",
"network": "bitcoin",
"plugin": os.path.join(os.path.dirname(__file__), "../sauron.py"),
"sauron-api-endpoint": "https://blockstream.info/api",
}
self.daemon.opts.update(options)

# Monkey patch
def set_feerates(self, feerates, wait_for_effect=True):
return None


@pytest.fixture
def node_cls(monkeypatch):
monkeypatch.setenv("TEST_NETWORK", "bitcoin")
yield LightningNode


def test_rpc_getchaininfo(node_factory):
"""
Test getchaininfo
"""
ln_node = node_factory.get_node()

response = ln_node.rpc.call("getchaininfo")

assert ln_node.daemon.is_in_log("Sauron plugin initialized using Esplora API")

expected_response_keys = ["chain", "blockcount", "headercount", "ibd"]
assert list(response.keys()) == expected_response_keys
assert response["chain"] == "main"
assert not response["ibd"]


def test_rpc_getrawblockbyheight(node_factory):
"""
Test getrawblockbyheight
"""
ln_node = node_factory.get_node()

response = ln_node.rpc.call("getrawblockbyheight", {"height": 0})

expected_response = {
"block": "0100000000000000000000000000000000000000000000000000000000000000000000003ba3edfd7a7b12b27ac72c3e67768f617fc81bc3888a51323a9fb8aa4b1e5e4a29ab5f49ffff001d1dac2b7c0101000000010000000000000000000000000000000000000000000000000000000000000000ffffffff4d04ffff001d0104455468652054696d65732030332f4a616e2f32303039204368616e63656c6c6f72206f6e206272696e6b206f66207365636f6e64206261696c6f757420666f722062616e6b73ffffffff0100f2052a01000000434104678afdb0fe5548271967f1a67130b7105cd6a828e03909a67962e0ea1f61deb649f6bc3f4cef38c4f35504e51ec112de5c384df7ba0b8d578a4c702b6bf11d5fac00000000",
"blockhash": "000000000019d6689c085ae165831e934ff763ae46a2a6c172b3f1b60a8ce26f",
}
assert response == expected_response


def test_rpc_sendrawtransaction_invalid(node_factory):
"""
Test sendrawtransaction
"""
ln_node = node_factory.get_node()

expected_response = {
"errmsg": 'sendrawtransaction RPC error: {"code":-22,"message":"TX decode failed. Make sure the tx has at least one input."}',
"success": False,
}
response = ln_node.rpc.call(
"sendrawtransaction",
{"tx": "invalid-raw-tx"},
)

assert response == expected_response


def test_rpc_getutxout(node_factory):
"""
Test getutxout
"""
ln_node = node_factory.get_node()

expected_response = {
"amount": 1000000000,
"script": "4104b5abd412d4341b45056d3e376cd446eca43fa871b51961330deebd84423e740daa520690e1d9e074654c59ff87b408db903649623e86f1ca5412786f61ade2bfac",
}
response = ln_node.rpc.call(
"getutxout",
{
# block 181
"txid": "a16f3ce4dd5deb92d98ef5cf8afeaf0775ebca408f708b2146c4fb42b41e14be",
"vout": 0,
},
)
assert response == expected_response


def test_rpc_estimatefees(node_factory):
"""
Test estimatefees
"""
ln_node = node_factory.get_node()

# Sample response
# {
# "opening": 4477,
# "mutual_close": 4477,
# "unilateral_close": 11929,
# "delayed_to_us": 4477,
# "htlc_resolution": 5652,
# "penalty": 5652,
# "min_acceptable": 1060,
# "max_acceptable": 119290,
# "feerate_floor": 1520,
# "feerates": [
# {"blocks": 2, "feerate": 11929},
# {"blocks": 6, "feerate": 5652},
# {"blocks": 12, "feerate": 4477},
# {"blocks": 144, "feerate": 2120}
# ]
# }
response = ln_node.rpc.call("estimatefees")

expected_response_keys = [
"opening",
"mutual_close",
"unilateral_close",
"delayed_to_us",
"htlc_resolution",
"penalty",
"min_acceptable",
"max_acceptable",
"feerate_floor",
"feerates",
]
assert list(response.keys()) == expected_response_keys

expected_feerates_keys = ("blocks", "feerate")
assert (
list(set([tuple(entry.keys()) for entry in response["feerates"]]))[0]
== expected_feerates_keys
)

expected_feerates_blocks = [2, 6, 12, 144]
assert [
entry["blocks"] for entry in response["feerates"]
] == expected_feerates_blocks
Loading
Loading