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

iwa/plugins/olas/importer.py

@@ -181,9 +181,7 @@ class OlasServiceImporter:
             if service.service_id:
                 key = f"{service.chain_name}:{service.service_id}"
                 if key in seen_keys:
-                    logger.debug(
-                        f"Skipping duplicate service {key} from {service.source_folder}"
-                    )
+                    logger.debug(f"Skipping duplicate service {key} from {service.source_folder}")
                     duplicates += 1
                     continue
                 seen_keys.add(key)
@@ -482,9 +480,7 @@ class OlasServiceImporter:
                     staking_program_id, chain_name
                 )
 
-    def _resolve_staking_contract(
-        self, staking_program_id: str, chain_name: str
-    ) -> Optional[str]:
+    def _resolve_staking_contract(self, staking_program_id: str, chain_name: str) -> Optional[str]:
         """Resolve a staking program ID to a contract address."""
         address = STAKING_PROGRAM_MAP.get(staking_program_id)
         if address:
@@ -540,8 +536,10 @@ class OlasServiceImporter:
 
                 # Check for "safes" entry which indicates the owner is a Safe
                 # Structure: "safes": { "gnosis": "0x..." }
-                if "safes" in data and FLAGS_OWNER_SAFE in data["safes"]: # Need to detect chain dynamically or iterate
-                     pass
+                if (
+                    "safes" in data and FLAGS_OWNER_SAFE in data["safes"]
+                ):  # Need to detect chain dynamically or iterate
+                    pass
 
                 # Logic update:
                 # 1. Capture EOA address always (it's the signer)
@@ -557,9 +555,13 @@ class OlasServiceImporter:
                 if safe_owner_address:
                     # CASE: Owner is Safe
                     service.service_owner_multisig_address = safe_owner_address
-                    service.service_owner_eoa_address = eoa_address # The EOA is the signer/controller
+                    service.service_owner_eoa_address = (
+                        eoa_address  # The EOA is the signer/controller
+                    )
 
-                    logger.debug(f"Extracted Safe owner address: {safe_owner_address} (Signer: {eoa_address})")
+                    logger.debug(
+                        f"Extracted Safe owner address: {safe_owner_address} (Signer: {eoa_address})"
+                    )
                 elif eoa_address:
                     # CASE: Owner is EOA
                     service.service_owner_eoa_address = eoa_address
@@ -767,8 +769,8 @@ class OlasServiceImporter:
             safe_result = self._import_safe(
                 address=service.safe_address,
                 signers=self._get_agent_signers(service),
-                tag_suffix="multisig", # e.g. trader_zeta_safe
-                service_name=service.service_name
+                tag_suffix="multisig",  # e.g. trader_zeta_safe
+                service_name=service.service_name,
             )
             if safe_result[0]:
                 result.imported_safes.append(service.safe_address)
@@ -778,19 +780,22 @@ class OlasServiceImporter:
                 result.errors.append(f"Safe {service.safe_address}: {safe_result[1]}")
 
         # 2. Import Owner Safe (if it exists and is different)
-        if service.service_owner_multisig_address and service.service_owner_multisig_address != service.safe_address:
-             # Signer for Owner Safe is the EOA owner key
+        if (
+            service.service_owner_multisig_address
+            and service.service_owner_multisig_address != service.safe_address
+        ):
+            # Signer for Owner Safe is the EOA owner key
             owner_signers = self._get_owner_signers(service)
 
             safe_result = self._import_safe(
                 address=service.service_owner_multisig_address,
                 signers=owner_signers,
-                tag_suffix="owner_multisig", # e.g. trader_zeta_owner_safe
-                service_name=service.service_name
+                tag_suffix="owner_multisig",  # e.g. trader_zeta_owner_safe
+                service_name=service.service_name,
             )
             if safe_result[0]:
-                 result.imported_safes.append(service.service_owner_multisig_address)
-                 logger.info(f"Imported Owner Safe {service.service_owner_multisig_address}")
+                result.imported_safes.append(service.service_owner_multisig_address)
+                logger.info(f"Imported Owner Safe {service.service_owner_multisig_address}")
 
     def _get_agent_signers(self, service: DiscoveredService) -> List[str]:
         """Get list of signers for the agent safe."""
@@ -926,7 +931,7 @@ class OlasServiceImporter:
         address: str,
         signers: List[str] = None,
         tag_suffix: str = "multisig",
-        service_name: Optional[str] = None
+        service_name: Optional[str] = None,
     ) -> Tuple[bool, str]:
         """Import a generic Safe."""
         if not address:
@@ -1062,4 +1067,5 @@ class OlasServiceImporter:
             key.signature_failed = True
             logger.warning(f"Error verifying signature for key {key.address}: {e}")
 
-FLAGS_OWNER_SAFE="deprecated"
+
+FLAGS_OWNER_SAFE = "deprecated"

iwa/plugins/olas/plugin.py

@@ -151,7 +151,9 @@ class OlasPlugin(Plugin):
 
                     is_signer = key_addr in [s.lower() for s in on_chain_signers]
                     if not is_signer:
-                        safe_text += f"\n[bold red]⚠ Agent {agent_key.address} - NOT A SIGNER![/bold red]"
+                        safe_text += (
+                            f"\n[bold red]⚠ Agent {agent_key.address} - NOT A SIGNER![/bold red]"
+                        )
                     else:
                         safe_text += f" (Signer: {agent_key.address[:6]}...)"
 
@@ -182,21 +184,21 @@ class OlasPlugin(Plugin):
         # 1. Display Signer/EOA Owner
         owner_key = next((k for k in service.keys if k.role == "owner"), None)
         if owner_key:
-             val = owner_key.address
-             if not val.startswith("0x"):
-                 val = "0x" + val
-
-             if owner_key.signature_verified:
-                 val = f"[green]{val}[/green]"
-             elif not owner_key.is_encrypted:
-                 val = f"[red]{val}[/red]"
-             status = "🔒 encrypted" if owner_key.is_encrypted else "🔓 plaintext"
-             table.add_row("Owner (EOA)", f"{val} {status}")
+            val = owner_key.address
+            if not val.startswith("0x"):
+                val = "0x" + val
+
+            if owner_key.signature_verified:
+                val = f"[green]{val}[/green]"
+            elif not owner_key.is_encrypted:
+                val = f"[red]{val}[/red]"
+            status = "🔒 encrypted" if owner_key.is_encrypted else "🔓 plaintext"
+            table.add_row("Owner (EOA)", f"{val} {status}")
         elif service.service_owner_eoa_address:
-             # Fallback if we have an address but no key object
-             table.add_row("Owner (EOA)", service.service_owner_eoa_address)
+            # Fallback if we have an address but no key object
+            table.add_row("Owner (EOA)", service.service_owner_eoa_address)
         else:
-             table.add_row("Owner (EOA)", "[yellow]N/A[/yellow]")
+            table.add_row("Owner (EOA)", "[yellow]N/A[/yellow]")
 
         # 2. Display Safe Owner
         if service.service_owner_multisig_address:

iwa/plugins/olas/service_manager/drain.py

@@ -135,9 +135,7 @@ class DrainManagerMixin:
         withdrawal_tag = self.wallet.get_tag_by_address(withdrawal_address) or withdrawal_address
         multisig_tag = self.wallet.get_tag_by_address(multisig_address) or multisig_address
 
-        logger.info(
-            f"Withdrawing {olas_amount:.4f} OLAS from {multisig_tag} to {withdrawal_tag}"
-        )
+        logger.info(f"Withdrawing {olas_amount:.4f} OLAS from {multisig_tag} to {withdrawal_tag}")
 
         # Transfer from Safe to withdrawal address
         tx_hash = self.wallet.send(

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