mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 14:07:55 +00:00
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
"""Tests for the WebChatChannel adapter."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from openjarvis.channels._stubs import ChannelStatus
|
|
from openjarvis.channels.webchat import WebChatChannel
|
|
from openjarvis.core.events import EventBus, EventType
|
|
from openjarvis.core.registry import ChannelRegistry
|
|
from tests.channels.channel_test_helpers import make_common_channel_tests
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _register_webchat():
|
|
"""Re-register after any registry clear."""
|
|
if not ChannelRegistry.contains("webchat"):
|
|
ChannelRegistry.register_value("webchat", WebChatChannel)
|
|
|
|
|
|
TestCommonChannel = make_common_channel_tests(WebChatChannel, "webchat")
|
|
|
|
|
|
class TestInit:
|
|
def test_defaults(self):
|
|
ch = WebChatChannel()
|
|
assert ch._messages == []
|
|
assert ch._handlers == []
|
|
assert ch._status == ChannelStatus.DISCONNECTED
|
|
|
|
|
|
class TestSend:
|
|
def test_send_success(self):
|
|
ch = WebChatChannel()
|
|
result = ch.send("user", "Hello!")
|
|
assert result is True
|
|
|
|
def test_send_publishes_event(self):
|
|
bus = EventBus(record_history=True)
|
|
ch = WebChatChannel(bus=bus)
|
|
ch.send("user", "Hello!")
|
|
|
|
event_types = [e.event_type for e in bus.history]
|
|
assert EventType.CHANNEL_MESSAGE_SENT in event_types
|
|
|
|
def test_get_messages(self):
|
|
ch = WebChatChannel()
|
|
ch.send("user1", "Hello!")
|
|
ch.send("user2", "World!")
|
|
ch.send("user1", "Again!")
|
|
messages = ch.get_messages()
|
|
assert len(messages) == 3
|
|
assert messages[0].content == "Hello!"
|
|
assert messages[1].content == "World!"
|
|
assert messages[2].content == "Again!"
|
|
|
|
def test_clear_messages(self):
|
|
ch = WebChatChannel()
|
|
ch.send("user1", "Hello!")
|
|
ch.send("user2", "World!")
|
|
assert len(ch.get_messages()) == 2
|
|
ch.clear_messages()
|
|
assert len(ch.get_messages()) == 0
|
|
|
|
|
|
class TestStatus:
|
|
def test_connected_after_connect(self):
|
|
ch = WebChatChannel()
|
|
ch.connect()
|
|
assert ch.status() == ChannelStatus.CONNECTED
|