iwa 0.0.58__py3-none-any.whl → 0.0.59__py3-none-any.whl

iwa/plugins/olas/service_manager/lifecycle.py

@@ -429,9 +429,7 @@ class LifecycleManagerMixin:
             )
 
             if balance < bond_amount:
-                logger.error(
-                    f"[ACTIVATE] FAIL: Owner balance {balance} < required {bond_amount}"
-                )
+                logger.error(f"[ACTIVATE] FAIL: Owner balance {balance} < required {bond_amount}")
                 return False
 
             protocol_contracts = OLAS_CONTRACTS.get(self.chain_name.lower(), {})
@@ -682,7 +680,9 @@ class LifecycleManagerMixin:
                 )
                 return True
 
-        logger.info(f"[REGISTER] Service Owner approving Token Utility for bond: {bond_amount_wei} wei")
+        logger.info(
+            f"[REGISTER] Service Owner approving Token Utility for bond: {bond_amount_wei} wei"
+        )
 
         utility_address = str(
             OLAS_CONTRACTS[self.chain_name]["OLAS_SERVICE_REGISTRY_TOKEN_UTILITY"]
@@ -697,9 +697,7 @@ class LifecycleManagerMixin:
         )
 
         if allowance >= bond_amount_wei:
-            logger.debug(
-                f"[REGISTER] Sufficient allowance ({allowance} >= {bond_amount_wei})"
-            )
+            logger.debug(f"[REGISTER] Sufficient allowance ({allowance} >= {bond_amount_wei})")
             return True
 
         # Use service owner which holds the OLAS tokens (not necessarily master)
@@ -800,7 +798,9 @@ class LifecycleManagerMixin:
             )
             return False
 
-        logger.debug(f"[DEPLOY] Preparing deploy tx for owner {self._get_label(self.service.service_owner_address)}")
+        logger.debug(
+            f"[DEPLOY] Preparing deploy tx for owner {self._get_label(self.service.service_owner_address)}"
+        )
         deploy_tx = self.manager.prepare_deploy_tx(
             from_address=self.service.service_owner_address,
             service_id=self.service.service_id,
@@ -855,7 +855,7 @@ class LifecycleManagerMixin:
             _, agent_instances = self.registry.call("getAgentInstances", self.service.service_id)
             service_info = self.registry.get_service(self.service.service_id)
             threshold = service_info["threshold"]
-           # Store the multisig in the wallet with tag
+            # Store the multisig in the wallet with tag
             multisig_tag = f"{self.service.service_name}_multisig"
 
             # ARCHIVING LOGIC: If tag is already taken by a different address, rename the old one

iwa/plugins/olas/service_manager/staking.py

@@ -90,6 +90,7 @@ class StakingManagerMixin:
         # Try token/contract names
         try:
             from iwa.core.chain import ChainInterfaces
+
             chain_interface = ChainInterfaces().get(self.chain_name)
             token_name = chain_interface.chain.get_token_name(address)
             if token_name:
@@ -407,7 +408,9 @@ class StakingManagerMixin:
         # NOTE: We don't check OLAS balance here because OLAS was already
         # deposited to the Token Utility during activation (min_staking_deposit)
         # and registration (agent_bond). The staking contract pulls from there.
-        logger.debug("[STAKE] OLAS already deposited to Token Utility during activation/registration")
+        logger.debug(
+            "[STAKE] OLAS already deposited to Token Utility during activation/registration"
+        )
 
         return {"min_deposit": min_deposit, "staking_token": staking_token}
 
@@ -437,7 +440,9 @@ class StakingManagerMixin:
         owner_address = self.service.service_owner_address or self.wallet.master_account.address
 
         # Approve service NFT - this is an ERC-721 approval, not ERC-20
-        logger.debug(f"[STAKE] Approving service NFT for staking contract from {self._get_label(owner_address)}...")
+        logger.debug(
+            f"[STAKE] Approving service NFT for staking contract from {self._get_label(owner_address)}..."
+        )
         approve_tx = self.registry.prepare_approve_tx(
             from_address=owner_address,
             spender=staking_contract.address,
@@ -483,7 +488,9 @@ class StakingManagerMixin:
         # Use service owner which holds the NFT (not necessarily master)
         owner_address = self.service.service_owner_address or self.wallet.master_account.address
 
-        logger.debug(f"[STAKE] Preparing stake transaction from {self._get_label(owner_address)}...")
+        logger.debug(
+            f"[STAKE] Preparing stake transaction from {self._get_label(owner_address)}..."
+        )
         stake_tx = staking_contract.prepare_stake_tx(
             from_address=owner_address,
             service_id=self.service.service_id,
@@ -601,9 +608,7 @@ class StakingManagerMixin:
             return False
 
         tx_hash = get_tx_hash(receipt)
-        logger.info(
-            f"Unstake transaction sent: {tx_hash if receipt else 'No Receipt'}"
-        )
+        logger.info(f"Unstake transaction sent: {tx_hash if receipt else 'No Receipt'}")
 
         events = staking_contract.extract_events(receipt)
 

iwa/plugins/olas/tests/test_olas_archiving.py

@@ -20,10 +20,12 @@ class TestOlasArchiving(unittest.TestCase):
         self.mock_wallet.key_storage = self.mock_key_storage
 
         # Initialize ServiceManager
-        with patch("iwa.core.models.Config"), \
-             patch("iwa.plugins.olas.service_manager.base.ChainInterfaces"), \
-             patch("iwa.plugins.olas.service_manager.base.ContractCache"), \
-             patch("iwa.plugins.olas.service_manager.base.ServiceRegistryContract"):
+        with (
+            patch("iwa.core.models.Config"),
+            patch("iwa.plugins.olas.service_manager.base.ChainInterfaces"),
+            patch("iwa.plugins.olas.service_manager.base.ContractCache"),
+            patch("iwa.plugins.olas.service_manager.base.ServiceRegistryContract"),
+        ):
             self.sm = ServiceManager(self.mock_wallet)
             self.sm.service = Service(service_name="trader_psi", chain_name="gnosis", service_id=1)
             self.sm.chain_name = "gnosis"
@@ -40,23 +42,30 @@ class TestOlasArchiving(unittest.TestCase):
 
         # 2. Mock required contract calls for deploy()
         self.sm.registry.get_service.return_value = {
-            "state": ServiceState.FINISHED_REGISTRATION, # DEPLOYED
+            "state": ServiceState.FINISHED_REGISTRATION,  # DEPLOYED
             "security_deposit": 0,
             "multisig": new_address,
             "threshold": 1,
-            "configHash": b"\x00" * 32
+            "configHash": b"\x00" * 32,
         }
-        self.sm.registry.call.return_value = (None, []) # getAgentInstances
+        self.sm.registry.call.return_value = (None, [])  # getAgentInstances
 
         # 3. Trigger deploy (archiving happens here)
-        with patch.object(self.mock_wallet, "sign_and_send_transaction", return_value=(True, {"status": 1})), \
-             patch.object(self.sm.registry, "extract_events", return_value=[
-                 {"name": "DeployService", "args": {}},
-                 {"name": "CreateMultisigWithAgents", "args": {"multisig": new_address}}
-             ]), \
-             patch("iwa.plugins.olas.service_manager.lifecycle.get_tx_hash", return_value="0xhash"), \
-             patch("iwa.core.models.StoredSafeAccount") as mock_safe_cls:
-
+        with (
+            patch.object(
+                self.mock_wallet, "sign_and_send_transaction", return_value=(True, {"status": 1})
+            ),
+            patch.object(
+                self.sm.registry,
+                "extract_events",
+                return_value=[
+                    {"name": "DeployService", "args": {}},
+                    {"name": "CreateMultisigWithAgents", "args": {"multisig": new_address}},
+                ],
+            ),
+            patch("iwa.plugins.olas.service_manager.lifecycle.get_tx_hash", return_value="0xhash"),
+            patch("iwa.core.models.StoredSafeAccount") as mock_safe_cls,
+        ):
             self.sm.deploy(fund_multisig=False)
 
             # 4. Verify rename_account was called for the old address
@@ -69,5 +78,6 @@ class TestOlasArchiving(unittest.TestCase):
             self.assertEqual(kwargs["address"], new_address)
             self.assertEqual(kwargs["tag"], multisig_tag)
 
+
 if __name__ == "__main__":
     unittest.main()

iwa/plugins/olas/tests/test_olas_integration.py

@@ -309,9 +309,11 @@ def test_importer_import_service_config_duplicate(mock_wallet):
 
 def test_sm_create_token_utility_missing(mock_wallet):
     """Cover create() with missing token utility (lines 204-206)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
 
@@ -323,9 +325,11 @@ def test_sm_create_token_utility_missing(mock_wallet):
 
 def test_sm_get_staking_status_staked_info_fail(mock_wallet):
     """Cover get_staking_status with STAKED but get_service_info fails (lines 843-854)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -347,9 +351,11 @@ def test_sm_get_staking_status_staked_info_fail(mock_wallet):
 
 def test_sm_call_checkpoint_prepare_fail(mock_wallet):
     """Cover call_checkpoint prepare failure (lines 1100-1102)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -369,9 +375,11 @@ def test_sm_call_checkpoint_prepare_fail(mock_wallet):
 
 def test_sm_spin_up_no_service(mock_wallet):
     """Cover spin_up with no service (lines 1167-1170)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = None
@@ -382,9 +390,11 @@ def test_sm_spin_up_no_service(mock_wallet):
 
 def test_sm_spin_up_activation_fail(mock_wallet):
     """Cover spin_up activation failure (lines 1181-1183)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="gnosis", service_id=1)
@@ -402,9 +412,11 @@ def test_sm_spin_up_activation_fail(mock_wallet):
 
 def test_sm_wind_down_no_service(mock_wallet):
     """Cover wind_down with no service (lines 1264-1266)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = None
@@ -415,9 +427,11 @@ def test_sm_wind_down_no_service(mock_wallet):
 
 def test_sm_wind_down_nonexistent(mock_wallet):
     """Cover wind_down with non-existent service (lines 1274-1276)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="gnosis", service_id=1)
@@ -432,9 +446,11 @@ def test_sm_wind_down_nonexistent(mock_wallet):
 
 def test_sm_mech_request_no_service(mock_wallet):
     """Cover _send_legacy_mech_request with no service (lines 1502-1504)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = None
@@ -445,9 +461,11 @@ def test_sm_mech_request_no_service(mock_wallet):
 
 def test_sm_mech_request_no_address(mock_wallet):
     """Cover _send_legacy_mech_request missing mech address (lines 1510-1512)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="unknown", service_id=1)
@@ -458,8 +476,10 @@ def test_sm_mech_request_no_address(mock_wallet):
 
 def test_sm_marketplace_mech_no_service(mock_wallet):
     """Cover _send_marketplace_mech_request with no service (lines 1549-1551)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache"):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache"),
+    ):
         sm = ServiceManager(mock_wallet)
         sm.service = None
 

iwa/plugins/olas/tests/test_service_manager.py

@@ -104,8 +104,6 @@ def mock_chain_interfaces():
         yield mock
 
 
-
-
 @pytest.fixture
 def service_manager(
     mock_config,
@@ -113,13 +111,14 @@ def service_manager(
     mock_registry,
     mock_manager_contract,
     mock_chain_interfaces,
-
     mock_olas_config,
     mock_service,
 ):
     """ServiceManager fixture with mocked dependencies."""
-    with patch("iwa.plugins.olas.service_manager.base.Config") as local_mock_config, \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache:
+    with (
+        patch("iwa.plugins.olas.service_manager.base.Config") as local_mock_config,
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+    ):
         instance = local_mock_config.return_value
         instance.plugins = {"olas": mock_olas_config}
         instance.save_config = MagicMock()
@@ -136,7 +135,6 @@ def service_manager(
         yield sm
 
 
-
 def test_init(service_manager):
     """Test initialization."""
     assert service_manager.registry is not None
@@ -179,9 +177,7 @@ def test_create_no_event(service_manager, mock_wallet):
     mock_wallet.sign_and_send_transaction.return_value = (True, {})
     service_manager.registry.extract_events.return_value = []
 
-    res = service_manager.create(
-        token_address_or_tag="0x1111111111111111111111111111111111111111"
-    )
+    res = service_manager.create(token_address_or_tag="0x1111111111111111111111111111111111111111")
     # create() finds no ID, logs error, returns None for service_id.
     assert res is None
 
@@ -911,7 +907,9 @@ def test_activate_registration_token_service_sends_security_deposit_as_value(
 
     # Mock balance check to pass
     mock_wallet.balance_service = MagicMock()
-    mock_wallet.balance_service.get_erc20_balance_wei.return_value = 100 * 10**18  # Plenty of balance
+    mock_wallet.balance_service.get_erc20_balance_wei.return_value = (
+        100 * 10**18
+    )  # Plenty of balance
 
     # Mock allowance to pass check (return an int)
     mock_wallet.transfer_service.get_erc20_allowance.return_value = 10**20  # Plenty of allowance

iwa/plugins/olas/tests/test_service_manager_errors.py

@@ -26,10 +26,11 @@ def mock_wallet():
 @pytest.fixture
 def mock_manager(mock_wallet):
     """Setup a ServiceManager with mocked dependencies."""
-    with patch("iwa.plugins.olas.service_manager.base.Config") as mock_cfg_cls, \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
-
+    with (
+        patch("iwa.plugins.olas.service_manager.base.Config") as mock_cfg_cls,
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
 
         mock_cfg = mock_cfg_cls.return_value

iwa/plugins/olas/tests/test_service_manager_flows.py

@@ -237,7 +237,9 @@ def test_activate_registration(
 @patch("iwa.plugins.olas.service_manager.base.ServiceRegistryContract")
 @patch("iwa.plugins.olas.service_manager.base.ServiceManagerContract")
 @patch("iwa.plugins.olas.service_manager.base.ContractCache")
-def test_register_agent(mock_cache, mock_sm_contract, mock_registry_contract, mock_config_cls, mock_wallet):
+def test_register_agent(
+    mock_cache, mock_sm_contract, mock_registry_contract, mock_config_cls, mock_wallet
+):
     """Test agent registration flow."""
     mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
     mock_config_inst = mock_config_cls.return_value
@@ -354,7 +356,9 @@ def test_deploy(mock_cache, mock_sm_contract, mock_registry_contract, mock_confi
 @patch("iwa.plugins.olas.service_manager.base.ServiceRegistryContract")
 @patch("iwa.plugins.olas.service_manager.base.ServiceManagerContract")
 @patch("iwa.plugins.olas.service_manager.base.ContractCache")
-def test_terminate(mock_cache, mock_sm_contract, mock_registry_contract, mock_config_cls, mock_wallet):
+def test_terminate(
+    mock_cache, mock_sm_contract, mock_registry_contract, mock_config_cls, mock_wallet
+):
     """Test service termination."""
     mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
     # Setup mock service
@@ -457,7 +461,6 @@ def test_stake(mock_cache, mock_sm_contract, mock_registry_contract, mock_config
         "required_agent_bond": 50000000000000000000,
     }
 
-
     success = manager.stake(mock_staking)
     assert success is True
     assert manager.service.staking_contract_address == "0xDDdDddDdDdddDDddDDddDDDDdDdDDdDDdDDDDDDd"
@@ -480,8 +483,6 @@ def test_stake(mock_cache, mock_sm_contract, mock_registry_contract, mock_config
     assert manager.stake(mock_staking) is False
     mock_staking.get_service_ids.return_value = []
 
-
-
     # 4. Approve fail
     mock_wallet.sign_and_send_transaction.return_value = (False, {})
     assert manager.stake(mock_staking) is False

iwa/plugins/olas/tests/test_service_staking.py

@@ -33,9 +33,11 @@ def mock_wallet():
 
 def test_sm_unstake_not_staked(mock_wallet):
     """Cover unstake when not staked (lines 736-738)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -51,9 +53,11 @@ def test_sm_unstake_not_staked(mock_wallet):
 
 def test_sm_unstake_tx_fails(mock_wallet):
     """Cover unstake transaction failure (lines 766-768)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -77,9 +81,11 @@ def test_sm_unstake_tx_fails(mock_wallet):
 
 def test_sm_get_staking_status_no_staking_address(mock_wallet):
     """Cover get_staking_status with no staking address (lines 831)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -97,9 +103,11 @@ def test_sm_get_staking_status_no_staking_address(mock_wallet):
 
 def test_sm_get_staking_status_with_full_info(mock_wallet):
     """Cover get_staking_status with complete info (lines 866-891)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -142,9 +150,11 @@ def test_sm_get_staking_status_with_full_info(mock_wallet):
 
 def test_sm_claim_rewards_no_service(mock_wallet):
     """Cover claim_rewards with no service (lines 936-938)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = None
@@ -156,9 +166,11 @@ def test_sm_claim_rewards_no_service(mock_wallet):
 
 def test_sm_claim_rewards_no_staking_address(mock_wallet):
     """Cover claim_rewards with no staking address (lines 939-943)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -171,9 +183,11 @@ def test_sm_claim_rewards_no_staking_address(mock_wallet):
 
 def test_sm_claim_rewards_tx_fails(mock_wallet):
     """Cover claim_rewards transaction failure (lines 967-968)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(
@@ -199,9 +213,11 @@ def test_sm_claim_rewards_tx_fails(mock_wallet):
 
         mock_wallet.sign_and_send_transaction.return_value = (False, None)
 
-        with patch("iwa.plugins.olas.service_manager.drain.StakingContract", return_value=mock_staking):
-             success, amount = sm.claim_rewards()
-             assert success is False
+        with patch(
+            "iwa.plugins.olas.service_manager.drain.StakingContract", return_value=mock_staking
+        ):
+            success, amount = sm.claim_rewards()
+            assert success is False
 
 
 # === SERVICE MANAGER SPIN_UP STATE TRANSITIONS (lines 1188-1241) ===
@@ -209,9 +225,11 @@ def test_sm_claim_rewards_tx_fails(mock_wallet):
 
 def test_sm_spin_up_state_mismatch_after_activation(mock_wallet):
     """Cover spin_up state mismatch after activation (lines 1188-1191)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="gnosis", service_id=1)
@@ -233,9 +251,11 @@ def test_sm_spin_up_state_mismatch_after_activation(mock_wallet):
 
 def test_sm_spin_up_registration_fails(mock_wallet):
     """Cover spin_up registration failure (lines 1199-1201)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="gnosis", service_id=1)
@@ -253,9 +273,11 @@ def test_sm_spin_up_registration_fails(mock_wallet):
 
 def test_sm_spin_up_deploy_fails(mock_wallet):
     """Cover spin_up deploy failure (lines 1216-1218)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="gnosis", service_id=1)
@@ -273,9 +295,11 @@ def test_sm_spin_up_deploy_fails(mock_wallet):
 
 def test_sm_wind_down_terminate_fails(mock_wallet):
     """Cover wind_down terminate failure (lines 1299-1301)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache, \
-         patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache") as mock_cache,
+        patch("iwa.plugins.olas.service_manager.staking.ContractCache", mock_cache),
+    ):
         mock_cache.return_value.get_contract.side_effect = lambda cls, *a, **k: cls(*a, **k)
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="gnosis", service_id=1)
@@ -293,8 +317,10 @@ def test_sm_wind_down_terminate_fails(mock_wallet):
 
 def test_sm_wind_down_unbond_fails(mock_wallet):
     """Cover wind_down unbond failure (lines 1315-1317)."""
-    with patch("iwa.core.models.Config"), \
-         patch("iwa.plugins.olas.service_manager.base.ContractCache"):
+    with (
+        patch("iwa.core.models.Config"),
+        patch("iwa.plugins.olas.service_manager.base.ContractCache"),
+    ):
         sm = ServiceManager(mock_wallet)
         sm.service = Service(service_name="t", chain_name="gnosis", service_id=1)
 

iwa/tools/drain_accounts.py

@@ -1,5 +1,5 @@
-
 """Tool to drain specific accounts to a master address."""
+
 import argparse
 import sys
 from pathlib import Path
@@ -56,5 +56,6 @@ def main() -> None:
         except Exception:
             logger.exception(f"Error draining {tag}")
 
+
 if __name__ == "__main__":
     main()