xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-12 16:56:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(agent-runner): audit steering injection

Browse Source
This commit is contained in:
huanghuoguoguo
2026-06-12 00:28:03 +08:00
parent 6c186661e6
commit 14c9a3a8c6
7 changed files with 166 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tests/unit_tests/agent/test_event_log_transcript.py
View File

@@ -95,6 +95,41 @@ class TestEventLogStore:
)
assert event_id == "evt_1"
stored_event = mock_session.add.call_args.args[0]
assert stored_event.metadata_json is None
@pytest.mark.asyncio
async def test_append_event_stores_metadata_json(self, mock_db_engine):
"""EventLog metadata records steering dispatch/audit facts."""
from unittest.mock import AsyncMock, MagicMock, patch
store = EventLogStore(mock_db_engine)
mock_session = AsyncMock()
mock_session.add = MagicMock()
mock_session.commit = AsyncMock()
with patch.object(store, '_session_factory') as mock_factory:
mock_factory.return_value.__aenter__.return_value = mock_session
event_id = await store.append_event(
event_id="evt_steering",
event_type="message.received",
source="platform",
run_id="run_1",
runner_id="plugin:test/plugin/runner",
metadata={
"steering": {
"status": "queued",
"claimed_by_run_id": "run_1",
}
},
)
assert event_id == "evt_steering"
stored_event = mock_session.add.call_args.args[0]
assert '"status": "queued"' in stored_event.metadata_json
assert '"claimed_by_run_id": "run_1"' in stored_event.metadata_json
@pytest.mark.asyncio
async def test_append_event_truncates_input_summary(self, mock_db_engine):

45
tests/unit_tests/agent/test_session_registry.py
View File

@@ -213,6 +213,51 @@ class TestSessionRegistryBasic:
assert await registry.get('old_run') is None
assert await registry.get('new_run') is not None
@pytest.mark.asyncio
async def test_pull_steering_all_preserves_queue_order(self):
"""Default all-mode steering returns every queued item in FIFO order."""
registry = AgentRunSessionRegistry()
await registry.register(
run_id='run_steering',
runner_id='plugin:test/my-runner/default',
query_id=1,
plugin_identity='test/my-runner',
resources=make_resources(),
conversation_id='conv_1',
available_apis={'steering_pull': True},
)
await registry.enqueue_steering('run_steering', {'event': {'event_id': 'event_1'}, 'input': {'text': 'first'}})
await registry.enqueue_steering('run_steering', {'event': {'event_id': 'event_2'}, 'input': {'text': 'second'}})
await registry.enqueue_steering('run_steering', {'event': {'event_id': 'event_3'}, 'input': {'text': 'third'}})
items = await registry.pull_steering('run_steering', mode='all')
assert [item['event']['event_id'] for item in items] == ['event_1', 'event_2', 'event_3']
assert await registry.pull_steering('run_steering', mode='all') == []
@pytest.mark.asyncio
async def test_pull_steering_one_at_a_time_leaves_remaining_items(self):
"""one-at-a-time is an explicit runner-side throttling mode."""
registry = AgentRunSessionRegistry()
await registry.register(
run_id='run_steering_one',
runner_id='plugin:test/my-runner/default',
query_id=1,
plugin_identity='test/my-runner',
resources=make_resources(),
conversation_id='conv_1',
available_apis={'steering_pull': True},
)
await registry.enqueue_steering('run_steering_one', {'event': {'event_id': 'event_1'}})
await registry.enqueue_steering('run_steering_one', {'event': {'event_id': 'event_2'}})
first = await registry.pull_steering('run_steering_one', mode='one-at-a-time')
second = await registry.pull_steering('run_steering_one', mode='one-at-a-time')
assert [item['event']['event_id'] for item in first] == ['event_1']
assert [item['event']['event_id'] for item in second] == ['event_2']
class TestIsResourceAllowed:
"""Tests for is_resource_allowed validation."""