54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
|
import asyncio
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from hass_assistant import ConversationSession
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_session_idle_timeout_triggers_end():
|
||
|
|
session = ConversationSession(
|
||
|
|
enabled=True,
|
||
|
|
idle_timeout_seconds=1,
|
||
|
|
end_keywords=["再见", "拜拜", "滚", "退下"],
|
||
|
|
)
|
||
|
|
|
||
|
|
triggered = {"value": False}
|
||
|
|
|
||
|
|
async def on_timeout():
|
||
|
|
triggered["value"] = True
|
||
|
|
|
||
|
|
session.restart_idle_timer(on_timeout)
|
||
|
|
await asyncio.sleep(1.2)
|
||
|
|
|
||
|
|
assert triggered["value"] is True
|
||
|
|
await session.close()
|
||
|
|
|
||
|
|
|
||
|
|
def test_session_end_keyword_contains_match():
|
||
|
|
session = ConversationSession(
|
||
|
|
enabled=True,
|
||
|
|
idle_timeout_seconds=15,
|
||
|
|
end_keywords=["再见", "拜拜", "滚", "退下"],
|
||
|
|
)
|
||
|
|
|
||
|
|
assert session.match_end_keyword("那就先这样吧，再见哈") == "再见"
|
||
|
|
assert session.match_end_keyword("你先退下吧") == "退下"
|
||
|
|
assert session.match_end_keyword("继续聊聊") is None
|
||
|
|
|
||
|
|
|
||
|
|
def test_session_area_mapping_hit_and_miss():
|
||
|
|
session = ConversationSession(
|
||
|
|
enabled=True,
|
||
|
|
idle_timeout_seconds=15,
|
||
|
|
end_keywords=["再见"],
|
||
|
|
)
|
||
|
|
|
||
|
|
mapping = {"192.168.1.101": "客厅"}
|
||
|
|
assert session.resolve_area("192.168.1.101", mapping) == "客厅"
|
||
|
|
assert session.resolve_area("192.168.1.102", mapping) is None
|
||
|
|
assert session.resolve_area(None, mapping) is None
|
||
|
|
|
||
|
|
prefix = session.build_area_prefix("客厅")
|
||
|
|
assert "当前用户对话所在区域：客厅" in prefix
|