139 lines
3.9 KiB
Python
139 lines
3.9 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from datetime import datetime, timedelta
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from pydantic import ValidationError
|
||
|
|
|
||
|
|
from yuxi.channels.models import (
|
||
|
|
FetchOptions,
|
||
|
|
HistoricalMessage,
|
||
|
|
SessionScope,
|
||
|
|
ThreadContext,
|
||
|
|
ThreadType,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class TestThreadType:
|
||
|
|
def test_native(self):
|
||
|
|
assert ThreadType.NATIVE == "native"
|
||
|
|
|
||
|
|
def test_topic(self):
|
||
|
|
assert ThreadType.TOPIC == "topic"
|
||
|
|
|
||
|
|
def test_simulated(self):
|
||
|
|
assert ThreadType.SIMULATED == "simulated"
|
||
|
|
|
||
|
|
def test_all_values_strings(self):
|
||
|
|
for member in ThreadType:
|
||
|
|
assert isinstance(member.value, str)
|
||
|
|
|
||
|
|
|
||
|
|
class TestSessionScope:
|
||
|
|
def test_direct(self):
|
||
|
|
assert SessionScope.DIRECT == "dm"
|
||
|
|
|
||
|
|
def test_group(self):
|
||
|
|
assert SessionScope.GROUP == "group"
|
||
|
|
|
||
|
|
def test_thread(self):
|
||
|
|
assert SessionScope.THREAD == "thread"
|
||
|
|
|
||
|
|
def test_all_values_strings(self):
|
||
|
|
for member in SessionScope:
|
||
|
|
assert isinstance(member.value, str)
|
||
|
|
|
||
|
|
|
||
|
|
class TestThreadContext:
|
||
|
|
def test_minimal(self):
|
||
|
|
ctx = ThreadContext(thread_id="t1")
|
||
|
|
assert ctx.thread_id == "t1"
|
||
|
|
assert ctx.thread_type == ThreadType.DIRECT
|
||
|
|
assert ctx.parent_id is None
|
||
|
|
assert ctx.participants == []
|
||
|
|
assert ctx.metadata == {}
|
||
|
|
|
||
|
|
def test_full(self):
|
||
|
|
ctx = ThreadContext(
|
||
|
|
thread_id="t2",
|
||
|
|
thread_type=ThreadType.NATIVE,
|
||
|
|
parent_id="p1",
|
||
|
|
root_message_id="root_1",
|
||
|
|
participants=["u1", "u2"],
|
||
|
|
created_at=datetime.now(),
|
||
|
|
metadata={"key": "val"},
|
||
|
|
)
|
||
|
|
assert ctx.thread_type == ThreadType.NATIVE
|
||
|
|
assert len(ctx.participants) == 2
|
||
|
|
assert ctx.metadata["key"] == "val"
|
||
|
|
|
||
|
|
def test_thread_id_required(self):
|
||
|
|
with pytest.raises(ValidationError):
|
||
|
|
ThreadContext()
|
||
|
|
|
||
|
|
|
||
|
|
class TestHistoricalMessage:
|
||
|
|
def test_minimal(self):
|
||
|
|
msg = HistoricalMessage(
|
||
|
|
message_id="m1",
|
||
|
|
sender_id="s1",
|
||
|
|
sender_name="Alice",
|
||
|
|
content="hello",
|
||
|
|
timestamp=datetime.now(),
|
||
|
|
)
|
||
|
|
assert msg.message_id == "m1"
|
||
|
|
assert msg.sender_name == "Alice"
|
||
|
|
assert msg.is_from_bot is False
|
||
|
|
assert msg.reply_to_id is None
|
||
|
|
|
||
|
|
def test_bot_message(self):
|
||
|
|
msg = HistoricalMessage(
|
||
|
|
message_id="m2",
|
||
|
|
sender_id="bot1",
|
||
|
|
sender_name="Bot",
|
||
|
|
content="hello",
|
||
|
|
timestamp=datetime.now(),
|
||
|
|
is_from_bot=True,
|
||
|
|
)
|
||
|
|
assert msg.is_from_bot is True
|
||
|
|
|
||
|
|
def test_with_reply(self):
|
||
|
|
msg = HistoricalMessage(
|
||
|
|
message_id="m3",
|
||
|
|
sender_id="s3",
|
||
|
|
sender_name="Bob",
|
||
|
|
content="reply",
|
||
|
|
timestamp=datetime.now(),
|
||
|
|
reply_to_id="m2",
|
||
|
|
)
|
||
|
|
assert msg.reply_to_id == "m2"
|
||
|
|
|
||
|
|
def test_required_fields(self):
|
||
|
|
with pytest.raises(ValidationError):
|
||
|
|
HistoricalMessage(sender_id="s1", sender_name="Alice", content="hi", timestamp=datetime.now())
|
||
|
|
|
||
|
|
def test_sender_id_required(self):
|
||
|
|
with pytest.raises(ValidationError):
|
||
|
|
HistoricalMessage(message_id="m1", sender_name="Alice", content="hi", timestamp=datetime.now())
|
||
|
|
|
||
|
|
|
||
|
|
class TestFetchOptions:
|
||
|
|
def test_defaults(self):
|
||
|
|
opts = FetchOptions()
|
||
|
|
assert opts.max_messages == 50
|
||
|
|
assert opts.max_chars == 4000
|
||
|
|
assert opts.include_bot_messages is True
|
||
|
|
assert opts.before_message_id is None
|
||
|
|
assert opts.after_message_id is None
|
||
|
|
|
||
|
|
def test_custom(self):
|
||
|
|
opts = FetchOptions(
|
||
|
|
max_messages=10,
|
||
|
|
max_chars=1000,
|
||
|
|
include_bot_messages=False,
|
||
|
|
before_message_id="mid_123",
|
||
|
|
)
|
||
|
|
assert opts.max_messages == 10
|
||
|
|
assert opts.max_chars == 1000
|
||
|
|
assert opts.include_bot_messages is False
|
||
|
|
assert opts.before_message_id == "mid_123"
|