feat: 08

ApeWorX · May 15, 2024 · adfd6e6 · adfd6e6
1 parent a393408
commit adfd6e6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/ape_polygon/__init__.py b/ape_polygon/__init__.py
@@ -1,6 +1,6 @@
 from ape import plugins
 from ape.api.networks import LOCAL_NETWORK_NAME, ForkedNetworkAPI, NetworkAPI, create_network_type
-from ape_geth import GethProvider
+from ape_node import Node
 from ape_test import LocalProvider
 
 from .ecosystem import NETWORKS, Polygon, PolygonConfig
@@ -29,6 +29,6 @@ def networks():
 @plugins.register(plugins.ProviderPlugin)
 def providers():
     for network_name in NETWORKS:
-        yield "polygon", network_name, GethProvider
+        yield "polygon", network_name, Node
 
     yield "polygon", LOCAL_NETWORK_NAME, LocalProvider
diff --git a/tests/test_integration.py b/tests/test_integration.py
@@ -5,13 +5,13 @@
 EXPECTED_OUTPUT = """
 polygon
 ├── amoy
-│   └── geth  (default)
+│   └── node  (default)
 ├── local  (default)
 │   └── test  (default)
 ├── mainnet
-│   └── geth  (default)
+│   └── node  (default)
 └── mumbai
-    └── geth  (default)
+    └── node  (default)
 """.strip()
 
 
@@ -48,9 +48,9 @@ def assert_rich_text(actual: str, expected: str):
 
 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")
+    polygon.mainnet.set_default_provider("node")
+    polygon.mumbai.set_default_provider("node")
+    polygon.amoy.set_default_provider("node")
 
-    result = runner.invoke(cli, ["networks", "list"])
+    result = runner.invoke(cli, ("networks", "list"))
     assert_rich_text(result.output, EXPECTED_OUTPUT)