pyln-proto: Added a couple of utilities to manage onions and zbase32

contrib/pyln-proto/tests/test_primitives.py

@@ -0,0 +1,30 @@
+from binascii import hexlify, unhexlify
+from pyln.proto import zbase32
+from pyln.proto.primitives import ShortChannelId
+
+
+def test_short_channel_id():
+    num = 618150934845652992
+    b = unhexlify(b'08941d00090d0000')
+    s = '562205x2317x0'
+    s1 = ShortChannelId.from_int(num)
+    s2 = ShortChannelId.from_str(s)
+    s3 = ShortChannelId.from_bytes(b)
+    expected = ShortChannelId(block=562205, txnum=2317, outnum=0)
+
+    assert(s1 == expected)
+    assert(s2 == expected)
+    assert(s3 == expected)
+
+    assert(expected.to_bytes() == b)
+    assert(str(expected) == s)
+    assert(expected.to_int() == num)
+
+
+def test_zbase32():
+    zb32 = b'd75qtmgijm79rpooshmgzjwji9gj7dsdat8remuskyjp9oq1ugkaoj6orbxzhuo4njtyh96e3aq84p1tiuz77nchgxa1s4ka4carnbiy'
+    b = zbase32.decode(zb32)
+    assert(hexlify(b) == b'1f76e8acd54afbf23610b7166ba689afcc9e8ec3c44e442e765012dfc1d299958827d0205f7e4e1a12620e7fc8ce1c7d3651acefde899c33f12b6958d3304106a0')
+
+    enc = zbase32.encode(b)
+    assert(enc == zb32)