From e8db8f4ec148d9cf2282c5797d5bcb9ed0b6bfb2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Radek=20Mas=C5=82owski?= <rmaslowski@syncad.com>
Date: Mon, 10 Jun 2024 13:33:10 +0200
Subject: [PATCH] Refactor node to work correctly with new wallet
---
package/test_tools/__private/init_node.py | 2 +-
package/test_tools/__private/node.py | 4 +---
2 files changed, 2 insertions(+), 4 deletions(-)
diff --git a/package/test_tools/__private/init_node.py b/package/test_tools/__private/init_node.py
index d45e2a440..e891d40d4 100644
--- a/package/test_tools/__private/init_node.py
+++ b/package/test_tools/__private/init_node.py
@@ -41,7 +41,7 @@ class InitNode(WitnessNode):
self.api.debug_node.debug_set_vest_price(vest_price=new_price.as_schema())
def __stabilize_the_price(self, invest: Asset.TestT) -> None:
- from test_tools.__private.wallet import Wallet
+ from test_tools.__private.wallet.wallet import Wallet
wallet = Wallet(attach_to=self)
diff --git a/package/test_tools/__private/node.py b/package/test_tools/__private/node.py
index ec8a6a7ba..b07c43005 100644
--- a/package/test_tools/__private/node.py
+++ b/package/test_tools/__private/node.py
@@ -35,7 +35,7 @@ if TYPE_CHECKING:
from test_tools.__private.alternate_chain_specs import AlternateChainSpecs
from test_tools.__private.user_handles.handles.network_handle import NetworkHandle
from test_tools.__private.user_handles.handles.node_handles.node_handle_base import NodeHandleBase as NodeHandle
- from test_tools.__private.wallet import Wallet
+ from test_tools.__private.wallet.wallet import Wallet
class Node(BaseNode, ScopedObject):
@@ -541,7 +541,6 @@ class Node(BaseNode, ScopedObject):
time_control: TimeControl | str | None = None,
alternate_chain_specs: AlternateChainSpecs | None = None,
) -> None:
- self.__close_wallets()
self.close()
self.run(
wait_for_live=wait_for_live,
@@ -549,7 +548,6 @@ class Node(BaseNode, ScopedObject):
time_control=time_control,
alternate_chain_specs=alternate_chain_specs,
)
- self.__run_wallets()
def __remove_files(self) -> None:
policy = cleanup_policy.get_default() if self.__cleanup_policy is None else self.__cleanup_policy
--
GitLab