test: import graph

Author: pinheadmz

Diff for: test/graph_test.py

@@ -1,14 +1,13 @@
 #!/usr/bin/env python3
 
+import json
 import os
 import tempfile
 import uuid
 from pathlib import Path
-
+from warnet.utils import channel_match
 from test_base import TestBase
-from warnet.utils import DEFAULT_TAG
 
-json_file_path = Path(os.path.dirname(__file__)) / "data" / "LN_10.json"
 
 base = TestBase()
 base.start_server()
@@ -41,6 +40,10 @@
     base.wait_for_all_tanks_status(target="stopped")
 
 print("Import graph")
+
+json_file_path = Path(os.path.dirname(__file__)) / "data" / "LN_10.json"
+NUM_NODES = 10
+
 with tempfile.TemporaryDirectory() as dir:
     tf = f"{dir}/{str(uuid.uuid4())}.graphml"
     if base.backend == "compose":
@@ -66,6 +69,18 @@
     base.warcli("scenarios run ln_init")
     base.wait_for_all_scenarios()
 
-    print(base.warcli("lncli 0 describegraph"))
+    print("Ensuring warnet LN channel policies match imported JSON description")
+    with open(json_file_path, "r") as file:
+        actual = json.loads(base.warcli(f"lncli 0 describegraph"))["edges"]
+        expected = json.loads(file.read())["edges"]
+        expected = sorted(expected, key=lambda chan: int(chan['channel_id']))
+        for chan_index, actual_chan in enumerate(actual):
+            expected_chan = expected[chan_index]
+            if not channel_match(actual_chan, expected_chan, allow_flip=True):
+                raise Exception(
+                    f"Channel policy doesn't match source: {actual_chan['channel_id']}\n" +
+                    "Actual:\n" + json.dumps(actual_chan, indent=2) +
+                    "Expected:\n" + json.dumps(expected_chan, indent=2)
+                )
 
 base.stop_server()