90 lines
3.5 KiB
Python
90 lines
3.5 KiB
Python
from types import SimpleNamespace
|
|
|
|
from src.chat.message_receive.chat_manager import chat_manager
|
|
from src.common.utils.utils_config import ChatConfigUtils, ExpressionConfigUtils
|
|
from src.common.utils.utils_session import SessionUtils
|
|
from src.config.config import global_config
|
|
|
|
|
|
def test_get_chat_prompt_for_chat_merges_multiple_matching_prompts(monkeypatch):
|
|
session_id = SessionUtils.calculate_session_id("qq", group_id="1036092828")
|
|
monkeypatch.setattr(
|
|
global_config.chat,
|
|
"chat_prompts",
|
|
[
|
|
{"platform": "qq", "item_id": "1036092828", "rule_type": "group", "prompt": "你也是群管理员，可以适当进行管理"},
|
|
{"platform": "qq", "item_id": "1036092828", "rule_type": "group", "prompt": "这个群是技术实验群，请你专心讨论技术"},
|
|
{"platform": "qq", "item_id": "other", "rule_type": "group", "prompt": "不应该生效"},
|
|
],
|
|
)
|
|
monkeypatch.setattr(chat_manager, "get_session_by_session_id", lambda _session_id: None)
|
|
|
|
result = ChatConfigUtils.get_chat_prompt_for_chat(session_id, True)
|
|
|
|
assert result == "你也是群管理员，可以适当进行管理\n这个群是技术实验群，请你专心讨论技术"
|
|
|
|
|
|
def test_get_chat_prompt_for_chat_matches_routed_session_by_chat_stream(monkeypatch):
|
|
session_id = SessionUtils.calculate_session_id("qq", group_id="1036092828", account_id="bot-a")
|
|
monkeypatch.setattr(
|
|
global_config.chat,
|
|
"chat_prompts",
|
|
[
|
|
{"platform": "qq", "item_id": "1036092828", "rule_type": "group", "prompt": "路由会话也应该生效"},
|
|
],
|
|
)
|
|
monkeypatch.setattr(
|
|
chat_manager,
|
|
"get_session_by_session_id",
|
|
lambda _session_id: SimpleNamespace(platform="qq", group_id="1036092828", user_id=None),
|
|
)
|
|
|
|
result = ChatConfigUtils.get_chat_prompt_for_chat(session_id, True)
|
|
|
|
assert result == "路由会话也应该生效"
|
|
|
|
|
|
def test_expression_learning_list_matches_routed_session_by_chat_stream(monkeypatch):
|
|
session_id = SessionUtils.calculate_session_id("qq", group_id="1036092828", account_id="bot-a")
|
|
monkeypatch.setattr(
|
|
global_config.expression,
|
|
"learning_list",
|
|
[
|
|
{
|
|
"platform": "qq",
|
|
"item_id": "1036092828",
|
|
"rule_type": "group",
|
|
"use_expression": False,
|
|
"enable_learning": False,
|
|
"enable_jargon_learning": True,
|
|
}
|
|
],
|
|
)
|
|
monkeypatch.setattr(
|
|
chat_manager,
|
|
"get_session_by_session_id",
|
|
lambda _session_id: SimpleNamespace(platform="qq", group_id="1036092828", user_id=None),
|
|
)
|
|
|
|
assert ExpressionConfigUtils.get_expression_config_for_chat(session_id) == (False, False, True)
|
|
|
|
|
|
def test_talk_value_rules_match_routed_session_by_chat_stream(monkeypatch):
|
|
session_id = SessionUtils.calculate_session_id("qq", group_id="1036092828", account_id="bot-a")
|
|
monkeypatch.setattr(global_config.chat, "talk_value", 0.1)
|
|
monkeypatch.setattr(global_config.chat, "enable_talk_value_rules", True)
|
|
monkeypatch.setattr(
|
|
global_config.chat,
|
|
"talk_value_rules",
|
|
[
|
|
{"platform": "qq", "item_id": "1036092828", "rule_type": "group", "time": "00:00-23:59", "value": 0.7}
|
|
],
|
|
)
|
|
monkeypatch.setattr(
|
|
chat_manager,
|
|
"get_session_by_session_id",
|
|
lambda _session_id: SimpleNamespace(platform="qq", group_id="1036092828", user_id=None),
|
|
)
|
|
|
|
assert ChatConfigUtils.get_talk_value(session_id, True) == 0.7
|