jarviscore-framework 0.3.1__py3-none-any.whl → 0.3.2__py3-none-any.whl

{jarviscore_framework-0.3.1.dist-info → jarviscore_framework-0.3.2.dist-info}/RECORD

@@ -3,13 +3,13 @@ examples/calculator_agent_example.py,sha256=x7TrzE45WT_1DqwEnw8U3Fw56WpR9jBe3SLZ
 examples/cloud_deployment_example.py,sha256=R0WVtta6hi_yuhWVnnQ5X_wiztPSRLC2gfhoDMgUakw,5038
 examples/custom_profile_decorator.py,sha256=4EAgqXrT9an1f5AQ3cEWAhPvI_EeXbQq7O8ByunXb_M,4099
 examples/custom_profile_wrap.py,sha256=0NoQECVBEJOiL7Syr2QL73jnvj2XhGQZUhOvisqEQXw,5088
-examples/customagent_cognitive_discovery_example.py,sha256=3A5ibpTL5GCwJtnvx2sbp_Da59uK_xhBf83HvxjdQy0,12952
-examples/customagent_distributed_example.py,sha256=_vt_2sqSCIdHgwuV4VQFzHzmC3bgemPrHFifeA5tRNE,13677
+examples/customagent_cognitive_discovery_example.py,sha256=UfdQm9j11G38IfHPyBguQSmxODU7A5Niod95aydNb7s,14835
+examples/customagent_distributed_example.py,sha256=RA-zvRjPq2YrGfKU6_dP4ZuEVwivKdA3NNPty6j_GmA,18951
 examples/customagent_p2p_example.py,sha256=tWywC5abc1aYIyW2V6Xgr_XkjZdqzcOeLut3PnzUpB8,30669
-examples/fastapi_integration_example.py,sha256=e87ewAFQzTCrckZzkTpSl3Ajx30qL6Ws7ArfqussbSo,22462
+examples/fastapi_integration_example.py,sha256=qWob72OD2BLf-SlpvwKjfzgyTZo97DL2ttp-Ox56280,24790
 examples/multi_agent_workflow.py,sha256=Sygx3iEBM9WzorVMXqtiwn4rLYrW9BsxsiQSKceuzsE,4303
 examples/research_agent_example.py,sha256=phJ5AHNnZ_pxCfiKvHoTp_IFwOAW7VD1fRNHlXvfgj4,2287
-jarviscore/__init__.py,sha256=S7nug9_fRGeOh5nNYBrlb-ty-uDt3cll9dvLbwCUJuY,3402
+jarviscore/__init__.py,sha256=MFMKSUN3tqYyUAXHKrHWp6GyptOeg_wbg7CgbEfqZwA,3402
 jarviscore/adapter/__init__.py,sha256=fipq2XzgW3Us35tbFhs9B_ypoeEZeaP7bTsNPFFZVRk,1092
 jarviscore/adapter/decorator.py,sha256=ZY-WCF16EvtQWkra_6HNBKdTOJ0G5yvrD5PYKa6bcqk,12040
 jarviscore/adapter/wrapper.py,sha256=kU9nBiezU1M1dq9x5EjnAVHs_QjFFOFx7dR4TNyFdPk,9666
@@ -26,7 +26,7 @@ jarviscore/context/jarvis_context.py,sha256=6ai2TjDE5PRiBxF5lTdyBMoK3b8wv6cr0a6q
 jarviscore/context/memory.py,sha256=taonyl24ZUe-NZ5VtvrxljNv6f_BebuH6VE7l0B9S7A,4442
 jarviscore/core/__init__.py,sha256=30K2aqZckYTRZupn6X-mGV2QDSqWCgJ1cpN6Zk1gqlQ,177
 jarviscore/core/agent.py,sha256=j76BGNwN6ATnjWujvBSz8Dvojz7W9AEzC7gxfs9xde4,15561
-jarviscore/core/mesh.py,sha256=3d7CdWqZhED_t2tWAtB5nBjWuyxcTbMbCIy5tWpvLmU,23370
+jarviscore/core/mesh.py,sha256=wPgrppzIXtqzhUdaCqNL4OvLpSBCUiAwxzcs-19Bpzg,29407
 jarviscore/core/profile.py,sha256=sTrGTxV9mAqbt5l3z0-BSNOeWzq8YDLR3mlaPFSgt1c,2190
 jarviscore/data/.env.example,sha256=TCPdye7tYNDOEpcgaEuzUsQ-H7m9G6rsyzNZV9IYQ9s,5156
 jarviscore/data/__init__.py,sha256=757nsqMkytYV0zXiM_mh3LqtGZZ1lgFuMzvaLrW51PM,151
@@ -35,20 +35,20 @@ jarviscore/data/examples/calculator_agent_example.py,sha256=x7TrzE45WT_1DqwEnw8U
 jarviscore/data/examples/cloud_deployment_example.py,sha256=R0WVtta6hi_yuhWVnnQ5X_wiztPSRLC2gfhoDMgUakw,5038
 jarviscore/data/examples/custom_profile_decorator.py,sha256=4EAgqXrT9an1f5AQ3cEWAhPvI_EeXbQq7O8ByunXb_M,4099
 jarviscore/data/examples/custom_profile_wrap.py,sha256=0NoQECVBEJOiL7Syr2QL73jnvj2XhGQZUhOvisqEQXw,5088
-jarviscore/data/examples/customagent_cognitive_discovery_example.py,sha256=3A5ibpTL5GCwJtnvx2sbp_Da59uK_xhBf83HvxjdQy0,12952
-jarviscore/data/examples/customagent_distributed_example.py,sha256=_vt_2sqSCIdHgwuV4VQFzHzmC3bgemPrHFifeA5tRNE,13677
+jarviscore/data/examples/customagent_cognitive_discovery_example.py,sha256=UfdQm9j11G38IfHPyBguQSmxODU7A5Niod95aydNb7s,14835
+jarviscore/data/examples/customagent_distributed_example.py,sha256=RA-zvRjPq2YrGfKU6_dP4ZuEVwivKdA3NNPty6j_GmA,18951
 jarviscore/data/examples/customagent_p2p_example.py,sha256=tWywC5abc1aYIyW2V6Xgr_XkjZdqzcOeLut3PnzUpB8,30669
-jarviscore/data/examples/fastapi_integration_example.py,sha256=e87ewAFQzTCrckZzkTpSl3Ajx30qL6Ws7ArfqussbSo,22462
+jarviscore/data/examples/fastapi_integration_example.py,sha256=qWob72OD2BLf-SlpvwKjfzgyTZo97DL2ttp-Ox56280,24790
 jarviscore/data/examples/multi_agent_workflow.py,sha256=Sygx3iEBM9WzorVMXqtiwn4rLYrW9BsxsiQSKceuzsE,4303
 jarviscore/data/examples/research_agent_example.py,sha256=phJ5AHNnZ_pxCfiKvHoTp_IFwOAW7VD1fRNHlXvfgj4,2287
-jarviscore/docs/API_REFERENCE.md,sha256=kEhCehRupHmBiJfD0AYpEiyMKXKbUI70dELM__8eNWg,31573
+jarviscore/docs/API_REFERENCE.md,sha256=duRyLZ9uDu4EgyTjOPThHpBdOp_l6K73yDJLtzXivdU,44404
 jarviscore/docs/AUTOAGENT_GUIDE.md,sha256=ftm8dymihs3Y9PZTZmSD9xRhlRFygfTZKm59Mz8zNRA,4618
-jarviscore/docs/CHANGELOG.md,sha256=9nnQEVpfsdIkJKPPrjJsZoJI9VQ4clKGhLE1lrodeTs,4661
-jarviscore/docs/CONFIGURATION.md,sha256=cJwSmegYug2E3W2-dwXT1PwKc05dES6opbUNegSELwc,14910
-jarviscore/docs/CUSTOMAGENT_GUIDE.md,sha256=OYeID56PsOE4T-YktazT1_xfHUD2U2FOap_ZU-0poxE,72539
+jarviscore/docs/CHANGELOG.md,sha256=_NIU7VQHiTB5HLcxw-HyhoDGup8uP2uCTcl6Nw5jQeY,7707
+jarviscore/docs/CONFIGURATION.md,sha256=QiP2RVmZMhbV46S9W0FPKQ2y8P97ZntORtwxH-xPfvc,14910
+jarviscore/docs/CUSTOMAGENT_GUIDE.md,sha256=hMbCD2R7h8gEUwl9usCc7_dNTWRRJ6NoI84Kip3yPX4,82420
 jarviscore/docs/GETTING_STARTED.md,sha256=w_t47M9kQcIG3N6tesMcjQgqpkV7PZOHAOXOSk8yGhE,15860
-jarviscore/docs/TROUBLESHOOTING.md,sha256=YFXc4pjm3_5Ol38r5fxXh1JFKBAlp14lOhIIe8zg12g,11544
-jarviscore/docs/USER_GUIDE.md,sha256=TpiRtyHG_bC0WT3JiwobbeKYaNVnF0u8UT-yZ30pMz8,20951
+jarviscore/docs/TROUBLESHOOTING.md,sha256=iK4kz4x4AsChBPxeOvW6bzg5Gf1XwoZTQEEJYOivtr0,11544
+jarviscore/docs/USER_GUIDE.md,sha256=uOYXABiOpr1e_4QM_8W2xIJtHbdZE6wOS87-U_srdTs,28426
 jarviscore/execution/__init__.py,sha256=yDAMehMO2dVvdKjxVx7zQV2AaxySmvymA24QF3O9tlY,1754
 jarviscore/execution/code_registry.py,sha256=C3_hAVXIeCG31qwSBUrmBBicmd2vnUrXJhJgj8MKlJw,9213
 jarviscore/execution/generator.py,sha256=zY7IxxDu4xoifeuCGZZN8_l8zQCsB5eUO9HGIiLIttw,8696
@@ -66,16 +66,18 @@ jarviscore/orchestration/engine.py,sha256=Xh9Fn3aVY7iwj7cAXdHcN90ogOiVFJlyJeoKnJ
 jarviscore/orchestration/status.py,sha256=XeKASMNQDwpJ6HpDw3m3eAAMNWsWCj4k9jrvMnLHPbo,2540
 jarviscore/p2p/__init__.py,sha256=k1KkBnbb74jrfWKLiKRsMbepy6xBfQZOGzOlDKvslp0,927
 jarviscore/p2p/broadcaster.py,sha256=Gaj0nCHDBQrVe4ob4GcaFSqnAaVe3ugEILslc5Mq1qE,14414
-jarviscore/p2p/coordinator.py,sha256=UtsSVSigIhs2Rcx1v5BVdLC11P649u71y8Bw4cU1fMk,30087
+jarviscore/p2p/coordinator.py,sha256=MMvV7O30Ro0HSRzAT3mpo9p_0QmQoSJ8IJPRkMySOlo,31521
 jarviscore/p2p/keepalive.py,sha256=it0SyXVBmujfE79thrap-A1UxZW01p9ENEIglIbQoAM,14988
-jarviscore/p2p/messages.py,sha256=nd7ZutZ1Xge2C1KkRSwQK3RJ03ScxqESF3fl48hEkhI,2540
-jarviscore/p2p/peer_client.py,sha256=hC21-ecE6edcaxKXhxTnikt0aV5CV-TDNcCuvhSgAcI,32713
-jarviscore/p2p/peer_tool.py,sha256=qXiF0b2pTo8tJ8bUHxe_XbxE0UFgwu2wCpy-W82ocUo,9098
+jarviscore/p2p/messages.py,sha256=ud856fEQQvZz0ngK8vVHKzJ5PlAZGZIRz3e_-DGyNQ4,3151
+jarviscore/p2p/peer_client.py,sha256=ikyn5cPUKODBunEG1qFtd6b_9gaAXYS4Be0sAkR9UL4,44859
+jarviscore/p2p/peer_tool.py,sha256=tMaMGbocGkluOfMEGQABPm37e8Tgr3_PAWZ0urGfFgw,9596
 jarviscore/p2p/swim_manager.py,sha256=mu9clcJ22gqSOoeWlr4QW9b4vhRvD6QEAlTjAVVZvV4,11126
 jarviscore/profiles/__init__.py,sha256=gg2_T7ONl5se6rCSXLo-OlA3yOzmyHZPaNLzB6zjolE,328
 jarviscore/profiles/autoagent.py,sha256=1nJAVf1oU9lLO47BP1xFGBDZtypXXkwKy6kZjtpdlX0,10424
-jarviscore/profiles/customagent.py,sha256=zGt0RMnMbyyWeXtbQkHdvk1-dMweuNVjPTnH0-mZdaU,13906
-jarviscore_framework-0.3.1.dist-info/licenses/LICENSE,sha256=SjsXanvmQJFYz_SVFa17O85-bKIa_aG99wrkPpWtypo,1101
+jarviscore/profiles/customagent.py,sha256=--oI_pgKFXprFJXAFD-7Ot9XSYsucr_1L87TOEsJ06g,14279
+jarviscore/testing/__init__.py,sha256=LdPqVN9xfeTBlc4znGcb6kwcTmUdG4Xt7NT77Uu3qOI,869
+jarviscore/testing/mocks.py,sha256=QO-9k5URcRTBRtGCe9v7U8rYT4UalpiQuIidLb51pOE,18949
+jarviscore_framework-0.3.2.dist-info/licenses/LICENSE,sha256=SjsXanvmQJFYz_SVFa17O85-bKIa_aG99wrkPpWtypo,1101
 test_logs/code_registry/functions/data_generator-558779ed_560ebc37.py,sha256=ua0Lueqe1mWCeMpKTMaumfPS-ZrWBFF_Zx6TU5QVjNo,132
 test_logs/code_registry/functions/data_generator-5ed3609e_560ebc37.py,sha256=ua0Lueqe1mWCeMpKTMaumfPS-ZrWBFF_Zx6TU5QVjNo,132
 test_logs/code_registry/functions/data_generator-66da0356_43970bb9.py,sha256=vOkjCOcfXHLJX2TkR3et9vWSwEhCmOn4DdJHMGfTMVY,733
@@ -130,6 +132,11 @@ tests/test_13_dx_improvements.py,sha256=t3xoWdRBq5o2j90i-kZCJYc9AwW9F0lsB-9Y4VhI
 tests/test_14_cloud_deployment.py,sha256=EaqJ9Cai9Ki61dhmeqNgMfEkWGnL8-aQg52CKZ6335I,16402
 tests/test_15_llm_cognitive_discovery.py,sha256=AvBtuIPgiI5Z4skpoEjBOP6XwJukAJy7nR08HP-Gl5M,26331
 tests/test_16_unified_dx_flow.py,sha256=fRqIwhXfkioEY1Wri2vNuxUvLYDy4FvwdRkOqAnffXs,38249
+tests/test_17_session_context.py,sha256=8fDdv8Ct3NNgoXgutp2Bk2CJMiN0oCMCqqUhBqyMYds,17018
+tests/test_18_mesh_diagnostics.py,sha256=-8cQMdGC7kgi8Cmafe3odz0wlYbDw17CxsJFJEVNClU,14043
+tests/test_19_async_requests.py,sha256=lCWTdynGHSloY73sH4UYw6xUOkeekYu_HqeUasVVipI,17381
+tests/test_20_load_balancing.py,sha256=a321d8PmwuE5EFN7PBoXol7yJPDtz9IeyjLxuQS_QnM,19573
+tests/test_21_mock_testing.py,sha256=JFvS6a7U8Tsz8GtGYIHBGp0jz_LUL71P6iqD8r_-N88,25469
 tests/test_agent.py,sha256=qx9SFDTP4DlcQi6hV8y6LZyEYX6IB8D3VnM7fODnW9s,5182
 tests/test_autoagent.py,sha256=_mzinLdQwskOn6a-yGqdfoOsqw2f52XSyTCmj8hLqlg,4628
 tests/test_autoagent_day4.py,sha256=TTb0kSImF9stMsq4cMlkGahf9UBpYjoNXAkgnkKuuQA,4719
@@ -141,7 +148,7 @@ tests/test_llm_fallback.py,sha256=CNajpKkQ6MO503dRbgaP2cz9kXHwUGKo5381tHKTe4c,57
 tests/test_mesh.py,sha256=JmAAvZaduQ8dHThFDBYgUnjiz8a3r4Kt1atvdH1JO5I,11857
 tests/test_p2p_integration.py,sha256=F9B21eWlwRzSRphm2Kacs9nM1FgSbSzi6RSLPDvvt2U,10995
 tests/test_remote_sandbox.py,sha256=80ebc0pWInauWnywsQ0VSzlk8OexSCgGL7BcJUCPkR8,3268
-jarviscore_framework-0.3.1.dist-info/METADATA,sha256=4dF2CJq_tMCtHyWoIVbr_Z9Igx-PKh_eP86jKMAAyxc,5876
-jarviscore_framework-0.3.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-jarviscore_framework-0.3.1.dist-info/top_level.txt,sha256=aTco8nlqDftlvhB43Je0xXmb-Pw5qYgj-phawjHX4VY,36
-jarviscore_framework-0.3.1.dist-info/RECORD,,
+jarviscore_framework-0.3.2.dist-info/METADATA,sha256=AOWvJ2tgtGxED2sKHjX2AsNsfk-kbhqDS-gZ7KiWBcs,5876
+jarviscore_framework-0.3.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+jarviscore_framework-0.3.2.dist-info/top_level.txt,sha256=aTco8nlqDftlvhB43Je0xXmb-Pw5qYgj-phawjHX4VY,36
+jarviscore_framework-0.3.2.dist-info/RECORD,,

tests/test_17_session_context.py

@@ -0,0 +1,489 @@
+"""
+Test 17: Session Context Propagation (Feature F6)
+
+Tests the context propagation feature:
+- Context parameter in notify(), request(), respond(), broadcast()
+- Auto-propagation of context in respond()
+- Context accessible via IncomingMessage.context
+
+Run with: pytest tests/test_17_session_context.py -v -s
+"""
+import asyncio
+import sys
+import pytest
+import logging
+from unittest.mock import AsyncMock, MagicMock, patch
+
+sys.path.insert(0, '.')
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+# =============================================================================
+# TEST: CONTEXT IN OUTGOING MESSAGES
+# =============================================================================
+
+class TestContextInNotify:
+    """Test context parameter in notify()."""
+
+    @pytest.mark.asyncio
+    async def test_notify_with_context(self):
+        """Test notify() accepts and sends context."""
+        from jarviscore.testing import MockPeerClient
+
+        client = MockPeerClient(
+            agent_id="sender-1",
+            agent_role="sender",
+            mock_peers=[{"role": "receiver", "capabilities": ["receiving"]}]
+        )
+
+        context = {"mission_id": "mission-123", "priority": "high"}
+        result = await client.notify("receiver", {"event": "test"}, context=context)
+
+        assert result is True
+        notifications = client.get_sent_notifications()
+        assert len(notifications) == 1
+        assert notifications[0]["context"] == context
+        assert notifications[0]["message"] == {"event": "test"}
+
+    @pytest.mark.asyncio
+    async def test_notify_without_context(self):
+        """Test notify() works without context (None by default)."""
+        from jarviscore.testing import MockPeerClient
+
+        client = MockPeerClient(
+            agent_id="sender-1",
+            agent_role="sender",
+            mock_peers=[{"role": "receiver", "capabilities": ["receiving"]}]
+        )
+
+        result = await client.notify("receiver", {"event": "test"})
+
+        assert result is True
+        notifications = client.get_sent_notifications()
+        assert len(notifications) == 1
+        assert notifications[0]["context"] is None
+
+
+class TestContextInRequest:
+    """Test context parameter in request()."""
+
+    @pytest.mark.asyncio
+    async def test_request_with_context(self):
+        """Test request() accepts and sends context."""
+        from jarviscore.testing import MockPeerClient
+
+        client = MockPeerClient(
+            agent_id="sender-1",
+            agent_role="sender",
+            mock_peers=[{"role": "analyst", "capabilities": ["analysis"]}]
+        )
+        client.set_mock_response("analyst", {"result": "analyzed"})
+
+        context = {"mission_id": "mission-456", "trace_id": "trace-abc"}
+        response = await client.request("analyst", {"query": "test"}, context=context)
+
+        assert response == {"result": "analyzed"}
+        requests = client.get_sent_requests()
+        assert len(requests) == 1
+        assert requests[0]["context"] == context
+
+    @pytest.mark.asyncio
+    async def test_request_without_context(self):
+        """Test request() works without context."""
+        from jarviscore.testing import MockPeerClient
+
+        client = MockPeerClient(
+            agent_id="sender-1",
+            agent_role="sender",
+            mock_peers=[{"role": "analyst", "capabilities": ["analysis"]}]
+        )
+
+        response = await client.request("analyst", {"query": "test"})
+
+        assert response is not None
+        requests = client.get_sent_requests()
+        assert len(requests) == 1
+        assert requests[0]["context"] is None
+
+
+class TestContextInBroadcast:
+    """Test context parameter in broadcast()."""
+
+    @pytest.mark.asyncio
+    async def test_broadcast_with_context(self):
+        """Test broadcast() accepts and sends context to all peers."""
+        from jarviscore.testing import MockPeerClient
+
+        client = MockPeerClient(
+            agent_id="broadcaster-1",
+            agent_role="broadcaster",
+            mock_peers=[
+                {"role": "peer1", "capabilities": ["cap1"]},
+                {"role": "peer2", "capabilities": ["cap2"]}
+            ]
+        )
+
+        context = {"broadcast_id": "bc-789", "source": "alert_system"}
+        count = await client.broadcast({"alert": "important"}, context=context)
+
+        assert count == 2
+        broadcasts = client.get_sent_broadcasts()
+        assert len(broadcasts) == 1
+        assert broadcasts[0]["context"] == context
+
+
+# =============================================================================
+# TEST: CONTEXT IN INCOMING MESSAGES
+# =============================================================================
+
+class TestContextInIncomingMessage:
+    """Test context accessible in IncomingMessage."""
+
+    def test_incoming_message_has_context_field(self):
+        """Test IncomingMessage dataclass has context field."""
+        from jarviscore.p2p.messages import IncomingMessage, MessageType
+
+        msg = IncomingMessage(
+            sender="sender-1",
+            sender_node="localhost:7946",
+            type=MessageType.NOTIFY,
+            data={"event": "test"},
+            context={"mission_id": "m-123"}
+        )
+
+        assert msg.context == {"mission_id": "m-123"}
+
+    def test_incoming_message_context_default_none(self):
+        """Test IncomingMessage context defaults to None."""
+        from jarviscore.p2p.messages import IncomingMessage, MessageType
+
+        msg = IncomingMessage(
+            sender="sender-1",
+            sender_node="localhost:7946",
+            type=MessageType.NOTIFY,
+            data={"event": "test"}
+        )
+
+        assert msg.context is None
+
+
+class TestContextInOutgoingMessage:
+    """Test context in OutgoingMessage dataclass."""
+
+    def test_outgoing_message_has_context_field(self):
+        """Test OutgoingMessage dataclass has context field."""
+        from jarviscore.p2p.messages import OutgoingMessage, MessageType
+
+        msg = OutgoingMessage(
+            target="receiver",
+            type=MessageType.REQUEST,
+            data={"query": "test"},
+            context={"priority": "high"}
+        )
+
+        assert msg.context == {"priority": "high"}
+
+
+# =============================================================================
+# TEST: CONTEXT AUTO-PROPAGATION IN RESPOND
+# =============================================================================
+
+class TestContextAutoPropagation:
+    """Test context auto-propagation in respond()."""
+
+    @pytest.mark.asyncio
+    async def test_respond_auto_propagates_context(self):
+        """Test respond() auto-propagates request context if not overridden."""
+        from jarviscore.p2p.peer_client import PeerClient
+        from jarviscore.p2p.messages import IncomingMessage, MessageType
+
+        # Create a mock coordinator
+        mock_coordinator = MagicMock()
+        mock_coordinator._send_p2p_message = AsyncMock(return_value=True)
+        mock_coordinator._remote_agent_registry = {}
+
+        # Create mock agent registry with a sender agent
+        class MockAgent:
+            def __init__(self):
+                self.agent_id = "sender-1"
+                self.role = "sender"
+                self.capabilities = ["sending"]
+                self.peers = MagicMock()
+                self.peers._deliver_message = AsyncMock()
+
+        mock_sender = MockAgent()
+        agent_registry = {"sender": [mock_sender]}
+
+        # Create the responder PeerClient
+        client = PeerClient(
+            coordinator=mock_coordinator,
+            agent_id="responder-1",
+            agent_role="responder",
+            agent_registry=agent_registry,
+            node_id="localhost:7946"
+        )
+
+        # Create incoming request with context
+        incoming = IncomingMessage(
+            sender="sender-1",
+            sender_node="localhost:7946",
+            type=MessageType.REQUEST,
+            data={"query": "test"},
+            correlation_id="corr-123",
+            context={"mission_id": "m-999", "trace_id": "t-abc"}
+        )
+
+        # Respond without explicitly providing context
+        result = await client.respond(incoming, {"answer": "42"})
+
+        assert result is True
+
+        # Verify the response was delivered with propagated context
+        mock_sender.peers._deliver_message.assert_called_once()
+        delivered_msg = mock_sender.peers._deliver_message.call_args[0][0]
+        assert delivered_msg.context == {"mission_id": "m-999", "trace_id": "t-abc"}
+
+    @pytest.mark.asyncio
+    async def test_respond_override_context(self):
+        """Test respond() can override context."""
+        from jarviscore.p2p.peer_client import PeerClient
+        from jarviscore.p2p.messages import IncomingMessage, MessageType
+
+        mock_coordinator = MagicMock()
+        mock_coordinator._send_p2p_message = AsyncMock(return_value=True)
+        mock_coordinator._remote_agent_registry = {}
+
+        class MockAgent:
+            def __init__(self):
+                self.agent_id = "sender-1"
+                self.role = "sender"
+                self.capabilities = ["sending"]
+                self.peers = MagicMock()
+                self.peers._deliver_message = AsyncMock()
+
+        mock_sender = MockAgent()
+        agent_registry = {"sender": [mock_sender]}
+
+        client = PeerClient(
+            coordinator=mock_coordinator,
+            agent_id="responder-1",
+            agent_role="responder",
+            agent_registry=agent_registry,
+            node_id="localhost:7946"
+        )
+
+        # Incoming with original context
+        incoming = IncomingMessage(
+            sender="sender-1",
+            sender_node="localhost:7946",
+            type=MessageType.REQUEST,
+            data={"query": "test"},
+            correlation_id="corr-456",
+            context={"original": "context"}
+        )
+
+        # Respond with overridden context
+        result = await client.respond(
+            incoming,
+            {"answer": "42"},
+            context={"overridden": "context", "status": "complete"}
+        )
+
+        assert result is True
+        delivered_msg = mock_sender.peers._deliver_message.call_args[0][0]
+        assert delivered_msg.context == {"overridden": "context", "status": "complete"}
+
+
+# =============================================================================
+# TEST: CONTEXT THROUGH LOCAL DELIVERY
+# =============================================================================
+
+class TestContextLocalDelivery:
+    """Test context flows through local message delivery."""
+
+    @pytest.mark.asyncio
+    async def test_context_in_local_notify(self):
+        """Test context preserved in local notify delivery."""
+        from jarviscore.p2p.peer_client import PeerClient
+        from jarviscore.p2p.messages import MessageType
+
+        mock_coordinator = MagicMock()
+        mock_coordinator._remote_agent_registry = {}
+
+        # Create receiver agent with PeerClient
+        class MockReceiverAgent:
+            def __init__(self):
+                self.agent_id = "receiver-1"
+                self.role = "receiver"
+                self.capabilities = ["receiving"]
+                self.peers = None
+
+        receiver = MockReceiverAgent()
+        receiver_client = PeerClient(
+            coordinator=mock_coordinator,
+            agent_id="receiver-1",
+            agent_role="receiver",
+            agent_registry={},
+            node_id="localhost:7946"
+        )
+        receiver.peers = receiver_client
+
+        # Create sender with receiver in registry
+        agent_registry = {"receiver": [receiver]}
+        sender_client = PeerClient(
+            coordinator=mock_coordinator,
+            agent_id="sender-1",
+            agent_role="sender",
+            agent_registry=agent_registry,
+            node_id="localhost:7946"
+        )
+
+        # Send notify with context
+        context = {"session_id": "sess-123", "user_id": "user-456"}
+        result = await sender_client.notify("receiver", {"event": "test"}, context=context)
+
+        assert result is True
+
+        # Receive and check context
+        msg = await receiver_client.receive(timeout=1)
+        assert msg is not None
+        assert msg.type == MessageType.NOTIFY
+        assert msg.context == context
+
+
+# =============================================================================
+# TEST: CONTEXT IN MOCK PEER CLIENT
+# =============================================================================
+
+class TestMockPeerClientContext:
+    """Test MockPeerClient supports context in all operations."""
+
+    @pytest.mark.asyncio
+    async def test_mock_inject_message_with_context(self):
+        """Test MockPeerClient.inject_message() supports context."""
+        from jarviscore.testing import MockPeerClient
+        from jarviscore.p2p.messages import MessageType
+
+        client = MockPeerClient()
+
+        context = {"injected": "context", "test_id": "inject-1"}
+        client.inject_message(
+            sender="injector",
+            message_type=MessageType.NOTIFY,
+            data={"injected": True},
+            context=context
+        )
+
+        msg = await client.receive(timeout=1)
+        assert msg is not None
+        assert msg.context == context
+        assert msg.data == {"injected": True}
+
+    @pytest.mark.asyncio
+    async def test_mock_tracks_context_in_sent_messages(self):
+        """Test MockPeerClient tracks context in all sent messages."""
+        from jarviscore.testing import MockPeerClient
+
+        client = MockPeerClient(
+            mock_peers=[{"role": "target", "capabilities": []}]
+        )
+
+        # Send notification with context
+        await client.notify("target", {"n": 1}, context={"ctx": "notify"})
+        assert client.get_sent_notifications()[0]["context"] == {"ctx": "notify"}
+
+        # Send request with context
+        await client.request("target", {"r": 2}, context={"ctx": "request"})
+        assert client.get_sent_requests()[0]["context"] == {"ctx": "request"}
+
+        # Send broadcast with context
+        await client.broadcast({"b": 3}, context={"ctx": "broadcast"})
+        assert client.get_sent_broadcasts()[0]["context"] == {"ctx": "broadcast"}
+
+
+# =============================================================================
+# TEST: CONTEXT IN CUSTOM AGENT HANDLERS
+# =============================================================================
+
+class TestContextInCustomAgentHandlers:
+    """Test context accessible in CustomAgent message handlers."""
+
+    @pytest.mark.asyncio
+    async def test_context_in_on_peer_request(self):
+        """Test context is accessible in on_peer_request handler."""
+        from jarviscore.profiles import CustomAgent
+        from jarviscore.p2p.messages import IncomingMessage, MessageType
+
+        received_context = None
+
+        class TestAgent(CustomAgent):
+            role = "context_handler"
+            capabilities = ["handling"]
+
+            async def on_peer_request(self, msg):
+                nonlocal received_context
+                received_context = msg.context
+                return {"received_context": msg.context}
+
+        agent = TestAgent()
+        agent._logger = MagicMock()
+        agent.peers = MagicMock()
+        agent.peers.respond = AsyncMock()
+
+        msg = IncomingMessage(
+            sender="sender",
+            sender_node="localhost:7946",
+            type=MessageType.REQUEST,
+            data={"query": "test"},
+            correlation_id="corr-123",
+            context={"mission_id": "m-handler", "stage": "processing"}
+        )
+
+        await agent._dispatch_message(msg)
+
+        assert received_context == {"mission_id": "m-handler", "stage": "processing"}
+
+    @pytest.mark.asyncio
+    async def test_context_in_on_peer_notify(self):
+        """Test context is accessible in on_peer_notify handler."""
+        from jarviscore.profiles import CustomAgent
+        from jarviscore.p2p.messages import IncomingMessage, MessageType
+
+        received_context = None
+
+        class TestAgent(CustomAgent):
+            role = "context_handler"
+            capabilities = ["handling"]
+
+            async def on_peer_notify(self, msg):
+                nonlocal received_context
+                received_context = msg.context
+
+            async def on_peer_request(self, msg):
+                return {}
+
+        agent = TestAgent()
+        agent._logger = MagicMock()
+
+        msg = IncomingMessage(
+            sender="sender",
+            sender_node="localhost:7946",
+            type=MessageType.NOTIFY,
+            data={"event": "test"},
+            context={"notification_source": "event_system"}
+        )
+
+        await agent._dispatch_message(msg)
+
+        assert received_context == {"notification_source": "event_system"}
+
+
+# =============================================================================
+# RUN TESTS
+# =============================================================================
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v", "-s"])