mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
32 lines
669 B
Python
32 lines
669 B
Python
|
"""api.schemas.websocket coverage."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from uuid import uuid4
|
||
|
|
|
||
|
|
from roboco.api.schemas.websocket import (
|
||
|
|
WSAgentStream,
|
||
|
|
WSMessage,
|
||
|
|
WSNotification,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ws_message_base() -> None:
|
||
|
|
msg = WSMessage(type="custom")
|
||
|
|
assert msg.type == "custom"
|
||
|
|
|
||
|
|
|
||
|
|
def test_ws_agent_stream() -> None:
|
||
|
|
msg = WSAgentStream(agent_id=uuid4(), chunk="thinking...")
|
||
|
|
assert msg.type == "agent.stream"
|
||
|
|
|
||
|
|
|
||
|
|
def test_ws_notification() -> None:
|
||
|
|
msg = WSNotification(
|
||
|
|
notification_id=uuid4(),
|
||
|
|
notification_type="MENTION",
|
||
|
|
subject="hi",
|
||
|
|
priority="normal",
|
||
|
|
)
|
||
|
|
assert msg.type == "notification"
|