mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 13:52:43 +08:00
bcc1718cc1
- Applied ruff formatting fixes to test files - Fixed import ordering and line length issues - All pre-commit hooks now pass successfully - Tests continue to pass after formatting changes Co-authored-by: openhands <openhands@all-hands.dev>
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from unittest.mock import MagicMock
|
|
from uuid import UUID, uuid4
|
|
|
|
import pytest
|
|
|
|
from openhands.app_server.app_conversation.app_conversation_models import (
|
|
AppConversationStartRequest,
|
|
)
|
|
from openhands.app_server.event_callback.event_callback_models import (
|
|
EventCallback,
|
|
EventCallbackProcessor,
|
|
)
|
|
from openhands.app_server.event_callback.event_callback_result_models import (
|
|
EventCallbackResult,
|
|
EventCallbackResultStatus,
|
|
)
|
|
from openhands.sdk import Event
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_app_conversation_start_request_polymorphism():
|
|
class MyCallbackProcessor(EventCallbackProcessor):
|
|
async def __call__(
|
|
self,
|
|
conversation_id: UUID,
|
|
callback: EventCallback,
|
|
event: Event,
|
|
) -> EventCallbackResult | None:
|
|
return EventCallbackResult(
|
|
status=EventCallbackResultStatus.SUCCESS,
|
|
event_callback_id=callback.id,
|
|
event_id=event.id,
|
|
conversation_id=conversation_id,
|
|
detail='Live long and prosper!',
|
|
)
|
|
|
|
req = AppConversationStartRequest(processors=[MyCallbackProcessor()])
|
|
assert len(req.processors) == 1
|
|
processor = req.processors[0]
|
|
result = await processor(uuid4(), MagicMock(id=uuid4()), MagicMock(id=str(uuid4())))
|
|
assert result.detail == 'Live long and prosper!'
|