92 lines
3.1 KiB
Python
92 lines
3.1 KiB
Python
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
import src.chat.replyer.maisaka_expression_selector as selector_module
|
|
from src.chat.replyer.maisaka_expression_selector import MaisakaExpressionSelector
|
|
from src.common.utils.utils_session import SessionUtils
|
|
|
|
|
|
def _build_target(platform: str, item_id: str, rule_type: str = "group") -> SimpleNamespace:
|
|
return SimpleNamespace(platform=platform, item_id=item_id, rule_type=rule_type)
|
|
|
|
|
|
def test_resolve_expression_group_scope_returns_related_sessions(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
current_session_id = SessionUtils.calculate_session_id("qq", group_id="10001")
|
|
related_session_id = SessionUtils.calculate_session_id("qq", group_id="10002")
|
|
|
|
monkeypatch.setattr(
|
|
selector_module,
|
|
"global_config",
|
|
SimpleNamespace(
|
|
expression=SimpleNamespace(
|
|
expression_groups=[
|
|
SimpleNamespace(
|
|
expression_groups=[
|
|
_build_target("qq", "10001"),
|
|
_build_target("qq", "10002"),
|
|
]
|
|
)
|
|
]
|
|
)
|
|
),
|
|
)
|
|
|
|
selector = MaisakaExpressionSelector()
|
|
related_session_ids, has_global_share = selector._resolve_expression_group_scope(current_session_id)
|
|
|
|
assert related_session_ids == {current_session_id, related_session_id}
|
|
assert has_global_share is False
|
|
|
|
|
|
def test_resolve_expression_group_scope_uses_star_as_global_share(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
current_session_id = SessionUtils.calculate_session_id("qq", group_id="10001")
|
|
|
|
monkeypatch.setattr(
|
|
selector_module,
|
|
"global_config",
|
|
SimpleNamespace(
|
|
expression=SimpleNamespace(
|
|
expression_groups=[
|
|
SimpleNamespace(
|
|
expression_groups=[
|
|
_build_target("*", "*"),
|
|
]
|
|
)
|
|
]
|
|
)
|
|
),
|
|
)
|
|
|
|
selector = MaisakaExpressionSelector()
|
|
related_session_ids, has_global_share = selector._resolve_expression_group_scope(current_session_id)
|
|
|
|
assert related_session_ids == {current_session_id}
|
|
assert has_global_share is True
|
|
|
|
|
|
def test_resolve_expression_group_scope_does_not_treat_empty_target_as_global(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
current_session_id = SessionUtils.calculate_session_id("qq", group_id="10001")
|
|
|
|
monkeypatch.setattr(
|
|
selector_module,
|
|
"global_config",
|
|
SimpleNamespace(
|
|
expression=SimpleNamespace(
|
|
expression_groups=[
|
|
SimpleNamespace(
|
|
expression_groups=[
|
|
_build_target("", ""),
|
|
]
|
|
)
|
|
]
|
|
)
|
|
),
|
|
)
|
|
|
|
selector = MaisakaExpressionSelector()
|
|
related_session_ids, has_global_share = selector._resolve_expression_group_scope(current_session_id)
|
|
|
|
assert related_session_ids == {current_session_id}
|
|
assert has_global_share is False
|