rgb-cln/contrib/pyln-spec/bolt1/tests/test_bolt1.py

62 lines
2.9 KiB
Python

#! /usr/bin/python3
from pyln.proto.message import Message, MessageNamespace
import pyln.spec.bolt1 as bolt1
import io
def test_bolt_01_csv_tlv():
# FIXME: Test failure cases too!
for t in [['0x', ''],
['0x21 00', '33='],
['0xfd0201 00', '513='],
['0xfd00fd 00', '253='],
['0xfd00ff 00', '255='],
['0xfe02000001 00', '33554433='],
['0xff0200000000000001 00', '144115188075855873='],
['0x01 00', 'tlv1={amount_msat=0}'],
['0x01 01 01', 'tlv1={amount_msat=1}'],
['0x01 02 0100', 'tlv1={amount_msat=256}'],
['0x01 03 010000', 'tlv1={amount_msat=65536}'],
['0x01 04 01000000', 'tlv1={amount_msat=16777216}'],
['0x01 05 0100000000', 'tlv1={amount_msat=4294967296}'],
['0x01 06 010000000000', 'tlv1={amount_msat=1099511627776}'],
['0x01 07 01000000000000', 'tlv1={amount_msat=281474976710656}'],
['0x01 08 0100000000000000', 'tlv1={amount_msat=72057594037927936}'],
['0x02 08 0000000000000226', 'tlv2={scid=0x0x550}'],
['0x03 31 023da092f6980e58d2c037173180e9a465476026ee50f96695963e8efe436f54eb00000000000000010000000000000002', 'tlv3={node_id=023da092f6980e58d2c037173180e9a465476026ee50f96695963e8efe436f54eb,amount_msat_1=1,amount_msat_2=2}'],
['0xfd00fe 02 0226', 'tlv4={cltv_delta=550}']]:
msg = io.BytesIO(bytes.fromhex(t[0][2:].replace(' ', '')))
val = bolt1.n1.read(msg, None)
assert len(msg.read()) == 0
assert bolt1.n1.val_to_str(val, None) == '{' + t[1] + '}'
def test_bolt_01_csv():
# We can create a namespace from the csv.
ns = MessageNamespace(bolt1.csv)
# string [expected string]
for t in [['init globalfeatures= features=80',
'init globalfeatures= features=80 tlvs={}'],
['init globalfeatures= features=80 tlvs={}'],
['init globalfeatures= features=80 tlvs={networks={chains=[6fe28c0ab6f1b372c1a6a246ae63f74f931e8365e15a089c68d6190000000000]}}'],
['init globalfeatures= features=80 tlvs={networks={chains=[6fe28c0ab6f1b372c1a6a246ae63f74f931e8365e15a089c68d6190000000000,1fe28c0ab6f1b372c1a6a246ae63f74f931e8365e15a089c68d6190000000000]}}'],
['error channel_id=0000000000000000000000000000000000000000000000000000000000000000 data=00'],
['ping num_pong_bytes=0 ignored='],
['ping num_pong_bytes=3 ignored=0000'],
['pong ignored='],
['pong ignored=000000']]:
m = Message.from_str(bolt1.namespace, t[0])
b = io.BytesIO()
m.write(b)
# Works with our manually-made namespace, and the builtin one.
b.seek(0)
m2 = Message.read(bolt1.namespace, b)
assert m2.to_str() == t[-1]
b.seek(0)
m2 = Message.read(ns, b)
assert m2.to_str() == t[-1]