Kris
69fe97a90d
1. 移除Telegram格式化测试中未使用的导入项 2. 修复Teams测试用例，添加monkeypatch参数并配置通配符开关 3. 更新钉钉适配器测试，替换弃用的流属性检查 4. 修正Twitch规范化测试，更新ROOMSTATE测试逻辑 5. 重构会话映射测试，完善数据库执行结果模拟 6. 格式化Slack块构建测试的长参数调用 7. 修复LINE适配器测试，更新能力断言和异步锁使用 8. 修正Slack会话解析测试，修复聊天类型判断错误 9. 更新能力测试，补充缺失的字段检查 10. 修复Matrix适配器测试，修正位置参数和配置校验逻辑 11. 为飞书分析模块测试添加跳过标记 12. 新增微信能力、限流、链接格式、会话路由等模块的单元测试 13. 修复Twitch适配器导入路径和测试断言 14. 新增Discord Webhook、Nextcloud Talk、Signal多账户等模块的单元测试 15. 修复Manager阶段测试的导入路径 16. 新增iMessage异常和命令处理的单元测试 17. 新增Nostr健康检查和相关模块的单元测试 18. 新增Signal守护进程和SSE重连相关测试
136 lines
4.7 KiB
Python
136 lines
4.7 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import time
|
|
|
|
from yuxi.channels.adapters.nextcloudtalk.dedup import NextcloudTalkDedupGuard
|
|
|
|
|
|
class TestDedupGuardGC:
|
|
def test_gc_removes_expired_entries(self):
|
|
guard = NextcloudTalkDedupGuard(ttl=-1, max_entries=1000)
|
|
guard.commit("token", "msg-1")
|
|
assert guard.is_duplicate("token", "msg-1") is False
|
|
|
|
def test_gc_enforces_max_entries(self):
|
|
guard = NextcloudTalkDedupGuard(ttl=3600, max_entries=5)
|
|
for i in range(10):
|
|
guard.commit("token", f"msg-{i}")
|
|
stats = guard.stats()
|
|
assert stats["committed"] <= 5
|
|
|
|
|
|
class TestDedupGuardMakeKey:
|
|
def test_make_key_format(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
key = guard._make_key("token-abc", "msg-123")
|
|
assert key == "token-abc:msg-123"
|
|
|
|
|
|
class TestDedupGuardClaimCommitCycle:
|
|
def test_claim_then_commit(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
assert guard.claim("token", "msg-1") is True
|
|
assert guard.claim("token", "msg-1") is False
|
|
guard.commit("token", "msg-1")
|
|
assert guard.is_duplicate("token", "msg-1") is True
|
|
assert guard.claim("token", "msg-1") is False
|
|
|
|
def test_claim_then_release(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
assert guard.claim("token", "msg-1") is True
|
|
guard.release("token", "msg-1")
|
|
assert guard.claim("token", "msg-1") is True
|
|
|
|
def test_claim_then_mark_invalid(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
assert guard.claim("token", "msg-1") is True
|
|
guard.mark_invalid("token", "msg-1")
|
|
assert guard.is_invalid("token", "msg-1") is True
|
|
assert guard.claim("token", "msg-1") is True
|
|
|
|
|
|
class TestDedupGuardInvalid:
|
|
def test_mark_and_check_invalid(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
guard.mark_invalid("token", "msg-1")
|
|
assert guard.is_invalid("token", "msg-1") is True
|
|
|
|
def test_invalid_not_duplicate(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
guard.mark_invalid("token", "msg-1")
|
|
assert guard.is_duplicate("token", "msg-1") is False
|
|
|
|
|
|
class TestDedupGuardStats:
|
|
def test_initial_stats(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
stats = guard.stats()
|
|
assert stats["committed"] == 0
|
|
assert stats["pending"] == 0
|
|
assert stats["invalid"] == 0
|
|
|
|
def test_stats_after_operations(self):
|
|
guard = NextcloudTalkDedupGuard()
|
|
guard.commit("token", "msg-1")
|
|
guard.commit("token", "msg-2")
|
|
guard.claim("token", "msg-3")
|
|
guard.mark_invalid("token", "msg-4")
|
|
stats = guard.stats()
|
|
assert stats["committed"] == 2
|
|
assert stats["pending"] == 1
|
|
assert stats["invalid"] == 1
|
|
|
|
|
|
class TestDedupGuardPersist:
|
|
def test_flush_saves_to_file(self, tmp_path, monkeypatch):
|
|
import yuxi.channels.adapters.nextcloudtalk.dedup as dedup_mod
|
|
|
|
persist_file = tmp_path / ".dedup_cache_test.json"
|
|
monkeypatch.setattr(dedup_mod, "_PERSIST_PATH", str(persist_file))
|
|
|
|
guard = NextcloudTalkDedupGuard(ttl=3600, max_entries=1000)
|
|
guard.commit("token", "msg-1")
|
|
guard.commit("token", "msg-2")
|
|
guard.flush()
|
|
|
|
assert persist_file.exists()
|
|
with open(persist_file, encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
assert data["token:msg-1"] > 0
|
|
assert data["token:msg-2"] > 0
|
|
|
|
def test_load_from_persist(self, tmp_path, monkeypatch):
|
|
import yuxi.channels.adapters.nextcloudtalk.dedup as dedup_mod
|
|
|
|
persist_file = tmp_path / ".dedup_cache_load.json"
|
|
now = time.time()
|
|
data = {
|
|
"token:msg-1": now,
|
|
"token:msg-2": now,
|
|
}
|
|
with open(persist_file, "w", encoding="utf-8") as f:
|
|
json.dump(data, f)
|
|
|
|
monkeypatch.setattr(dedup_mod, "_PERSIST_PATH", str(persist_file))
|
|
|
|
guard = NextcloudTalkDedupGuard(ttl=3600, max_entries=1000)
|
|
assert guard.is_duplicate("token", "msg-1") is True
|
|
assert guard.is_duplicate("token", "msg-2") is True
|
|
assert guard.is_duplicate("token", "msg-3") is False
|
|
|
|
def test_load_expired_from_persist(self, tmp_path, monkeypatch):
|
|
import yuxi.channels.adapters.nextcloudtalk.dedup as dedup_mod
|
|
|
|
persist_file = tmp_path / ".dedup_cache_expired.json"
|
|
old_time = time.time() - 99999
|
|
data = {
|
|
"token:msg-old": old_time,
|
|
}
|
|
with open(persist_file, "w", encoding="utf-8") as f:
|
|
json.dump(data, f)
|
|
|
|
monkeypatch.setattr(dedup_mod, "_PERSIST_PATH", str(persist_file))
|
|
|
|
guard = NextcloudTalkDedupGuard(ttl=10, max_entries=1000)
|
|
assert guard.is_duplicate("token", "msg-old") is False |