diff --git a/ape_polygon/__init__.py b/ape_polygon/__init__.py index b010db5..10ca4d7 100644 --- 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 index b8bf481..b54773d 100644 --- 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)