feat: add start of Universal Router encoder/decoder algorithm

ApeWorX · Mar 15, 2024 · 8ecb277 · 8ecb277
1 parent e166794
commit 8ecb277
Show file tree

Hide file tree

Showing 2 changed files with 439 additions and 0 deletions.
diff --git a/tests/functional/test_universal_router.py b/tests/functional/test_universal_router.py
@@ -0,0 +1,26 @@
+import pytest
+from hexbytes import HexBytes
+
+from uniswap_sdk.universal_router import Plan
+
+
+@pytest.mark.parametrize(
+    "plan_commands,plan_inputs,expected_plan",
+    [
+        (
+            "0b",
+            [
+                "000000000000000000000000c02aaa39b223fe8d0a0e5c4f27ead9083c756cc2"
+                "000000000000000000000000000000000000000000000000016345785d8a0000"
+            ],
+            Plan().wrap_eth("0xC02aaA39b223FE8D0A0e5C4F27eAD9083C756Cc2", int(10**17)),
+        ),
+    ],
+)
+def test_decode_plan(plan_commands, plan_inputs, expected_plan):
+    plan_commands, plan_inputs = HexBytes(plan_commands), list(HexBytes(i) for i in plan_inputs)
+    actual_plan = Plan.decode(plan_commands, plan_inputs)
+
+    assert actual_plan == expected_plan
+    assert actual_plan.encoded_commands == plan_commands
+    assert actual_plan.encode_inputs() == plan_inputs