From 044d6cd28244567a642fa5a311b7f8bb0dc1678c Mon Sep 17 00:00:00 2001 From: kmochocki <kmochocki@syncad.com> Date: Tue, 11 Feb 2025 14:51:37 +0000 Subject: [PATCH] Use StrictOverseer --- package/test_tools/__private/base_node.py | 6 ++---- package/test_tools/__private/wallet/wallet.py | 5 ++++- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/package/test_tools/__private/base_node.py b/package/test_tools/__private/base_node.py index 404b68595..bce5c7d32 100644 --- a/package/test_tools/__private/base_node.py +++ b/package/test_tools/__private/base_node.py @@ -6,6 +6,7 @@ from typing import TYPE_CHECKING, Any, cast from beekeepy import Settings from helpy import Hived +from helpy._communication.overseers import StrictOverseer from test_tools.__private.scope import context from test_tools.__private.user_handles.implementation import Implementation as UserHandleImplementation @@ -19,10 +20,7 @@ class BaseNode(UserHandleImplementation, Hived): self.__name = context.names.register_numbered_name(name) super().__init__( handle=handle, - settings=Settings( - period_between_retries=timedelta(seconds=0.5), - max_retries=8, - ), + settings=Settings(period_between_retries=timedelta(seconds=0.5), max_retries=8, overseer=StrictOverseer), ) def __str__(self) -> str: diff --git a/package/test_tools/__private/wallet/wallet.py b/package/test_tools/__private/wallet/wallet.py index 165e00068..90e6cdc91 100644 --- a/package/test_tools/__private/wallet/wallet.py +++ b/package/test_tools/__private/wallet/wallet.py @@ -10,6 +10,7 @@ from beekeepy import Beekeeper, Settings import wax from helpy import Hf26Asset as Asset from helpy import wax as wax_helpy +from helpy._communication.overseers import StrictOverseer from helpy.exceptions import ErrorInResponseError from schemas.fields.basic import PublicKey from schemas.fields.hex import Hex @@ -157,7 +158,9 @@ class Wallet(UserHandleImplementation, ScopedObject): if self.connected_node is not None and not self.connected_node.is_running(): raise exceptions.NodeIsNotRunningError("Before attaching wallet you have to run node") - self.__beekeeper = Beekeeper.factory(settings=Settings(working_directory=self.directory)) + self.__beekeeper = Beekeeper.factory( + settings=Settings(working_directory=self.directory, overseer=StrictOverseer) + ) self.__beekeeper_session = self.__beekeeper.create_session() try: -- GitLab