iwa 0.1.3__py3-none-any.whl → 0.1.6__py3-none-any.whl

iwa/web/tests/test_wallet_injection.py

@@ -0,0 +1,204 @@
+"""Tests for wallet injection in web dependencies."""
+
+from unittest.mock import MagicMock, patch
+
+
+class TestWalletInjection:
+    """Tests for get_wallet, set_wallet, and _WalletProxy."""
+
+    def setup_method(self):
+        """Reset the global wallet state before each test."""
+        # Import fresh to reset state
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+    def test_get_wallet_lazy_initialization(self):
+        """get_wallet() creates a Wallet on first call if none injected."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            mock_wallet_cls.return_value = MagicMock(name="LazyWallet")
+
+            result = deps.get_wallet()
+
+            mock_wallet_cls.assert_called_once()
+            assert result == mock_wallet_cls.return_value
+
+    def test_get_wallet_returns_same_instance(self):
+        """get_wallet() returns the same instance on subsequent calls."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            mock_wallet_cls.return_value = MagicMock(name="SingletonWallet")
+
+            first = deps.get_wallet()
+            second = deps.get_wallet()
+
+            # Should only create once
+            mock_wallet_cls.assert_called_once()
+            assert first is second
+
+    def test_set_wallet_injects_external_wallet(self):
+        """set_wallet() allows injecting an external wallet instance."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        external_wallet = MagicMock(name="ExternalWallet")
+        deps.set_wallet(external_wallet)
+
+        result = deps.get_wallet()
+
+        assert result is external_wallet
+
+    def test_set_wallet_prevents_lazy_init(self):
+        """set_wallet() prevents Wallet() from being called."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            external_wallet = MagicMock(name="InjectedWallet")
+            deps.set_wallet(external_wallet)
+
+            result = deps.get_wallet()
+
+            # Wallet() should NOT be called
+            mock_wallet_cls.assert_not_called()
+            assert result is external_wallet
+
+    def test_set_wallet_overrides_existing(self):
+        """set_wallet() can override a previously set wallet."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        wallet1 = MagicMock(name="Wallet1")
+        wallet2 = MagicMock(name="Wallet2")
+
+        deps.set_wallet(wallet1)
+        assert deps.get_wallet() is wallet1
+
+        deps.set_wallet(wallet2)
+        assert deps.get_wallet() is wallet2
+
+
+class TestWalletProxy:
+    """Tests for _WalletProxy backwards compatibility."""
+
+    def setup_method(self):
+        """Reset the global wallet state before each test."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+    def test_wallet_proxy_forwards_attribute_access(self):
+        """wallet.attribute forwards to get_wallet().attribute."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        mock_wallet = MagicMock()
+        mock_wallet.balance_service = MagicMock(name="BalanceService")
+        deps.set_wallet(mock_wallet)
+
+        # Access through the proxy
+        result = deps.wallet.balance_service
+
+        assert result is mock_wallet.balance_service
+
+    def test_wallet_proxy_forwards_method_calls(self):
+        """wallet.method() forwards to get_wallet().method()."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        mock_wallet = MagicMock()
+        mock_wallet.get_accounts_balances.return_value = ({"0x1": {}}, {"0x1": {}})
+        deps.set_wallet(mock_wallet)
+
+        # Call method through proxy
+        result = deps.wallet.get_accounts_balances("gnosis", ["native"])
+
+        mock_wallet.get_accounts_balances.assert_called_once_with("gnosis", ["native"])
+        assert result == ({"0x1": {}}, {"0x1": {}})
+
+    def test_wallet_proxy_triggers_lazy_init(self):
+        """Accessing wallet.attr when no wallet set triggers lazy init."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+        with patch.object(deps, "Wallet") as mock_wallet_cls:
+            mock_instance = MagicMock()
+            mock_instance.key_storage = MagicMock(name="KeyStorage")
+            mock_wallet_cls.return_value = mock_instance
+
+            # Access through proxy should trigger lazy init
+            result = deps.wallet.key_storage
+
+            mock_wallet_cls.assert_called_once()
+            assert result is mock_instance.key_storage
+
+
+class TestIntegrationWithRouters:
+    """Integration tests for wallet injection with routers."""
+
+    def setup_method(self):
+        """Reset the global wallet state before each test."""
+        import iwa.web.dependencies as deps
+
+        deps._wallet = None
+
+    def test_injected_wallet_used_by_routers(self):
+        """Routers using 'wallet' get the injected instance."""
+        import iwa.web.dependencies as deps
+
+        # Inject a mock wallet BEFORE routers access it
+        mock_wallet = MagicMock()
+        mock_wallet.key_storage = MagicMock()
+        mock_wallet.key_storage.accounts = {}
+        deps.set_wallet(mock_wallet)
+
+        # Simulate what a router does
+        from iwa.web.dependencies import wallet
+
+        # Access through the module-level wallet (proxy)
+        accounts = wallet.key_storage.accounts
+
+        assert accounts == {}
+        # Verify it's using our injected wallet
+        assert deps.get_wallet() is mock_wallet
+
+
+class TestServerAsync:
+    """Tests for run_server_async function."""
+
+    def test_run_server_async_source_has_localhost_default(self):
+        """run_server_async defaults to localhost for security (source check)."""
+        import pathlib
+
+        # Read the source file directly to verify default
+        server_path = pathlib.Path(__file__).parent.parent / "server.py"
+        source = server_path.read_text()
+
+        # Verify the function signature has localhost default
+        assert 'async def run_server_async(host: str = "127.0.0.1"' in source
+
+    def test_run_server_source_has_async_function(self):
+        """run_server_async function exists in server.py source."""
+        import pathlib
+
+        server_path = pathlib.Path(__file__).parent.parent / "server.py"
+        source = server_path.read_text()
+
+        # Verify the async function exists
+        assert "async def run_server_async" in source
+        assert "uvicorn.Config" in source
+        assert "uvicorn.Server" in source
+        assert "await server.serve()" in source

{iwa-0.1.3.dist-info → iwa-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iwa
-Version: 0.1.3
+Version: 0.1.6
 Summary: A secure, modular, and plugin-based framework for crypto agents and ops
 Requires-Python: <4.0,>=3.12
 Description-Content-Type: text/markdown

{iwa-0.1.3.dist-info → iwa-0.1.6.dist-info}/RECORD

@@ -92,11 +92,11 @@ iwa/plugins/olas/contracts/abis/staking_token.json,sha256=cuUOmi1s4Z6VSIX0an_IxK
 iwa/plugins/olas/scripts/test_full_mech_flow.py,sha256=Fqoq5bn7Z_3YyRrnuqNAZy9cwQDLiXP6Vf3EIeWPo2I,9024
 iwa/plugins/olas/scripts/test_simple_lifecycle.py,sha256=8T50tOZx3afeECSfCNAb0rAHNtYOsBaeXlMwKXElCk8,2099
 iwa/plugins/olas/service_manager/__init__.py,sha256=GXiThMEY3nPgHUl1i-DLrF4h96z9jPxxI8Jepo2E1PM,1926
-iwa/plugins/olas/service_manager/base.py,sha256=EBPg0ymqgtAb7ZvVSfTt31QYgv_6gp4UAc6je00NLAg,5009
-iwa/plugins/olas/service_manager/drain.py,sha256=tJLqvMEPOymxukFuxt-AdgOB4bIIciKcXoxOPx9j-LM,13983
-iwa/plugins/olas/service_manager/lifecycle.py,sha256=Jz2WTsBE_kGjFnsnpQlUBmMM2csOuqEfYdvj4cLrNuU,50586
+iwa/plugins/olas/service_manager/base.py,sha256=1a5qzG02mCYoUFxNRUVRf6HZ-HuUuXatQGJesqr14vQ,5736
+iwa/plugins/olas/service_manager/drain.py,sha256=67BRhIPjaCDqS4kUI7yVYCMAHjmvAk96hjuNLNJ35Zc,14480
+iwa/plugins/olas/service_manager/lifecycle.py,sha256=1RRlRl0bZhcdQ53SS7TVpUQL75JJcwhpjWD1pgfwW1g,50995
 iwa/plugins/olas/service_manager/mech.py,sha256=NVzVbEmyOe3wK92VEzCCOSuy3HDkEP1MSoVt7Av8Psk,27949
-iwa/plugins/olas/service_manager/staking.py,sha256=Vl7NGzxfNhlajec1kUvqODiq-DBXyd86at5jKrwHDuc,29808
+iwa/plugins/olas/service_manager/staking.py,sha256=qhZB5yM4YMJulfmFoX6zTAukbZxBw8xtcik7bu8dGjE,31264
 iwa/plugins/olas/tests/conftest.py,sha256=4vM7EI00SrTGyeP0hNzsGSQHEj2-iznVgzlNh2_OGfo,739
 iwa/plugins/olas/tests/test_importer.py,sha256=i9LKov7kNRECB3hmRnhKBwcfx3uxtjWe4BB77bOOpeo,4282
 iwa/plugins/olas/tests/test_importer_error_handling.py,sha256=GeXu4Par3_FAUL9hT6Sn5PdRg2_EU2gf3iaL73atoYo,12256
@@ -112,12 +112,12 @@ iwa/plugins/olas/tests/test_plugin.py,sha256=RVgU-Cq6t_3mOh90xFAGwlJOV7ZIgp0VNaK
 iwa/plugins/olas/tests/test_plugin_full.py,sha256=55EBa07JhJLVG3IMi6QKlR_ivWLYCdLQTySP66qbEXo,8584
 iwa/plugins/olas/tests/test_service_lifecycle.py,sha256=sOCtpz8T9s55AZe9AoqP1h3XrXw5NDSjDqwLgYThvU4,5559
 iwa/plugins/olas/tests/test_service_manager.py,sha256=_mFRptssimITHhjvZA5jUPU2bInUIPCKll5wlj9elcA,40905
-iwa/plugins/olas/tests/test_service_manager_errors.py,sha256=-qpLmU4Uiqqtre59L2wXpO4WPMs4ej_K_gAL3naEvRg,8554
+iwa/plugins/olas/tests/test_service_manager_errors.py,sha256=1rgM-gUrpo0y3IHMVhsiAC2JMRH5YrOpzRqTP5NPwfg,8608
 iwa/plugins/olas/tests/test_service_manager_flows.py,sha256=ZSmBJNa18d_MyAaLQRoPpfFYRwzmk9k-5AhSAGd7WeI,20737
 iwa/plugins/olas/tests/test_service_manager_mech.py,sha256=qG6qu5IPRNypXUsblU2OEkuiuwDJ0TH8RXZbibmTFcQ,4937
 iwa/plugins/olas/tests/test_service_manager_rewards.py,sha256=2YCrXBU5bEkPuhBoGBhjnO1nA2qwHxn5Ivrror18FHM,12248
 iwa/plugins/olas/tests/test_service_manager_validation.py,sha256=ajlfH5uc4mAHf8A7GLE5cW7X8utM2vUilM0JdGDdlVg,5382
-iwa/plugins/olas/tests/test_service_staking.py,sha256=exxWsile_wG_0rz_cGbCPG-_Ubq01Ofl4D_pi0plj5Y,18332
+iwa/plugins/olas/tests/test_service_staking.py,sha256=BZW0bpBmekP2tkD64K-DM9Xd2jATrF6QcY7byjDQf0U,18350
 iwa/plugins/olas/tests/test_staking_integration.py,sha256=q7zLQLrUyhtcnZf6MMymx2cX0Gmqaa7i1mRh1clnyj4,30198
 iwa/plugins/olas/tests/test_staking_validation.py,sha256=uug64jFcXYJ3Nw_lNa3O4fnhNr5wAWHHIrchSbR2MVE,4020
 iwa/plugins/olas/tui/__init__.py,sha256=5ZRsbC7J3z1xfkZRiwr4bLEklf78rNVjdswe2p7SlS8,28
@@ -146,10 +146,11 @@ iwa/tui/tests/test_wallets_refactor.py,sha256=71G3HLbhTtgDy3ffVbYv0MFYRgdYd-NWGB
 iwa/tui/tests/test_widgets.py,sha256=C9UgIGeWRaQ459JygFEQx-7hOi9mWrSUDDIMZH1ge50,3994
 iwa/tui/widgets/__init__.py,sha256=UzD6nJbwv9hOtkWl9I7faXm1a-rcu4xFRxrf4KBwwY4,161
 iwa/tui/widgets/base.py,sha256=Z8FigMhsfD76PkFVERqMaotd-xwXfuFZm_8TmCMOsl4,3381
-iwa/web/dependencies.py,sha256=0_dAJlRh6gKrUDRPKUe92eshFsg572yx_H0lQgSqGDA,2103
+iwa/web/cache.py,sha256=cRrt0uy2bbSL3qBXzZqx3u3S6rRYCduy9FqpfNnzgTI,4149
+iwa/web/dependencies.py,sha256=iTvdCSuETFLeQPtFydi21s1EKA_a80rbA57s-994fMQ,2980
 iwa/web/models.py,sha256=MSD9WPy_Nz_amWgoo2KSDTn4ZLv_AV0o0amuNtSf-68,3035
-iwa/web/server.py,sha256=4ZLVFEKoGs_NoCcXMeyYzDNdxUXazjwHQaX7CR1pwHE,5239
-iwa/web/routers/accounts.py,sha256=VhCHrwzRWqZcSW-tTEqFWT5hFl-IYEWpqXeuN8xM3-4,3922
+iwa/web/server.py,sha256=kqQrZhuc3sr_CBnC6xdyeFUu-2QKGCIbWul-sdjJ5xM,5787
+iwa/web/routers/accounts.py,sha256=zCkRDCIfL6PjVNtEmsU2GSL5Hmk8abebugisShgHoCU,4894
 iwa/web/routers/state.py,sha256=wsBAOIWZeMWjMwLiiWVhuEXHAceI0IIq6CPpmh7SbIc,2469
 iwa/web/routers/swap.py,sha256=8xycAytquR29ELxW3vx428W8s9bI_w_x2kpRhhJ0KXY,22630
 iwa/web/routers/transactions.py,sha256=bRjfD7zcuX3orVIHzOMVsIkEtcE_pM_KCom4cdAKV6Q,5602
@@ -157,16 +158,18 @@ iwa/web/routers/olas/__init__.py,sha256=Jo6Dm1e8fHafCD800fbxsxeFz3tvuEEKXEf5-9tj
 iwa/web/routers/olas/admin.py,sha256=PMRdNelqYgQ1xbqh3floFV5xrVtBRQiwZPd8J9_ffxg,5785
 iwa/web/routers/olas/funding.py,sha256=f8fADNtbZEBFl-vuVKfas6os38Vot6K5tJBTenZmCD0,4832
 iwa/web/routers/olas/general.py,sha256=dPsBQppTGoQY1RztliUhseOHOZGeeCR10lhThD9kyXo,803
-iwa/web/routers/olas/services.py,sha256=jDbxLUJBN48vs2-fXrIpxhPSM6quqYxggmMOkUVEIaI,18200
-iwa/web/routers/olas/staking.py,sha256=jktJ2C1Q9X4aC0tWJByN3sHpEXY0EIvr3rr4N0MtXXc,14081
+iwa/web/routers/olas/services.py,sha256=cKMjy7jExgu_X7idVF--l4o_sYc3ve-aNs0aONFHFcQ,19883
+iwa/web/routers/olas/staking.py,sha256=RCoYCzIfVuhjUckfbC_eZLKEYTw1VToaU6jMWPN-6MI,14369
 iwa/web/static/app.js,sha256=VCm9zPJERb9pX6zbFQ_7D47UnztGdibrkZ1dmwhsvdc,114949
 iwa/web/static/index.html,sha256=Qg06MFK__2KxwWTr8hhjX_GwsoN53QsLCTypu4fBR-Q,28516
 iwa/web/static/style.css,sha256=7i6T96pS7gXSLDZfyp_87gRlyB9rpsFWJEHJ-dRY1ug,24371
+iwa/web/tests/test_response_cache.py,sha256=o1s4shq4N72xjMBawx_mhnqDTY1GbNy6FTgulY2nvM4,25435
+iwa/web/tests/test_wallet_injection.py,sha256=YjoQh1FMwroswF6O_XZOwYXzanv-JYElNTDNKxAS77g,6751
 iwa/web/tests/test_web_endpoints.py,sha256=vA25YghHNB23sbmhD4ciesn_f_okSq0tjlkrSiKZ0rs,24007
 iwa/web/tests/test_web_olas.py,sha256=GunKEAzcbzL7FoUGMtEl8wqiqwYwA5lB9sOhfCNj0TA,16312
 iwa/web/tests/test_web_swap.py,sha256=7A4gBJFL01kIXPtW1E1J17SCsVc_0DmUn-R8kKrnnVA,2974
 iwa/web/tests/test_web_swap_coverage.py,sha256=zGNrzlhZ_vWDCvWmLcoUwFgqxnrp_ACbo49AtWBS_Kw,5584
-iwa-0.1.3.dist-info/licenses/LICENSE,sha256=eIubm_IlBHPYRQlLNZKbBNKhJUUP3JH0A2miZUhAVfI,1078
+iwa-0.1.6.dist-info/licenses/LICENSE,sha256=eIubm_IlBHPYRQlLNZKbBNKhJUUP3JH0A2miZUhAVfI,1078
 tests/legacy_cow.py,sha256=oOkZvIxL70ReEoD9oHQbOD5GpjIr6AGNHcOCgfPlerU,8389
 tests/legacy_safe.py,sha256=AssM2g13E74dNGODu_H0Q0y412lgqsrYnEzI97nm_Ts,2972
 tests/legacy_transaction_retry_logic.py,sha256=D9RqZ7DBu61Xr2djBAodU2p9UE939LL-DnQXswX5iQk,1497
@@ -224,8 +227,8 @@ tests/test_utils.py,sha256=vkP49rYNI8BRzLpWR3WnKdDr8upeZjZcs7Rx0pjbQMo,1292
 tests/test_workers.py,sha256=MInwdkFY5LdmFB3o1odIaSD7AQZb3263hNafO1De5PE,2793
 tools/create_and_stake_service.py,sha256=1xwy_bJQI1j9yIQ968Oc9Db_F6mk1659LuuZntTASDE,3742
 tools/verify_drain.py,sha256=PkMjblyOOAuQge88FwfEzRtCYeEtJxXhPBmtQYCoQ-8,6743
-iwa-0.1.3.dist-info/METADATA,sha256=uKemLrcWBfz_raFNxn3okTiQWiwF-lLQok838PpZXtI,7336
-iwa-0.1.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-iwa-0.1.3.dist-info/entry_points.txt,sha256=nwB6kscrfA7M00pYmL2j-sBH6eF6h2ga9IK1BZxdiyQ,241
-iwa-0.1.3.dist-info/top_level.txt,sha256=kedS9cRUbm4JE2wYeabIXilhHjN8KCw0IGbqqqsw0Bs,16
-iwa-0.1.3.dist-info/RECORD,,
+iwa-0.1.6.dist-info/METADATA,sha256=KUGGOVmGvGxhLDLF4tbDcu73WjStPOe_FCRiOqmf-nI,7336
+iwa-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+iwa-0.1.6.dist-info/entry_points.txt,sha256=nwB6kscrfA7M00pYmL2j-sBH6eF6h2ga9IK1BZxdiyQ,241
+iwa-0.1.6.dist-info/top_level.txt,sha256=kedS9cRUbm4JE2wYeabIXilhHjN8KCw0IGbqqqsw0Bs,16
+iwa-0.1.6.dist-info/RECORD,,