Update integration test to include amoy as a network.

ApeWorX · Mar 28, 2024 · 79870df · 79870df
1 parent 3f9027d
commit 79870df
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -4,12 +4,14 @@
 
 EXPECTED_OUTPUT = """
 polygon
-├── mainnet
+├── amoy
 │   └── geth  (default)
-├── mumbai
+├── local  (default)
+│   └── test  (default)
+├── mainnet
 │   └── geth  (default)
-└── local  (default)
-    └── test  (default)
+└── mumbai
+    └── geth  (default)
 """.strip()
 
 
@@ -48,6 +50,7 @@ def test_networks(runner, cli, polygon):
     # Do this in case local env changed it.
     polygon.mainnet.set_default_provider("geth")
     polygon.mumbai.set_default_provider("geth")
+    polygon.amoy.set_default_provider("geth")
 
     result = runner.invoke(cli, ["networks", "list"])
     assert_rich_text(result.output, EXPECTED_OUTPUT)