74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
import unittest
|
|
from pathlib import Path
|
|
from tempfile import TemporaryDirectory
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
|
|
from app.danmu_queue import BgiLogMonitor, WebServer
|
|
|
|
|
|
class BgiLogDisplayFormatTests(unittest.TestCase):
|
|
def test_formats_multi_instance_log_as_single_readable_line(self):
|
|
raw_lines = [
|
|
"[23:55:25.160] [INF] [Primary:S1:P16720:T1785426828552] "
|
|
"BetterGenshinImpact.GameTask.TaskTriggerDispatcher\n",
|
|
"游戏已退出，BetterGI 自动停止截图器\n",
|
|
"\n",
|
|
]
|
|
|
|
self.assertEqual(
|
|
BgiLogMonitor.format_display_lines(raw_lines),
|
|
["[23:55:25] 游戏已退出，BetterGI 自动停止截图器"],
|
|
)
|
|
|
|
def test_formats_legacy_log_header_the_same_way(self):
|
|
raw_lines = [
|
|
"[16:39:12.328] [INF] BetterGenshinImpact.GameTask.TaskTriggerDispatcher\n",
|
|
"游戏已退出，BetterGI 自动停止截图器\n",
|
|
]
|
|
|
|
self.assertEqual(
|
|
BgiLogMonitor.format_display_lines(raw_lines),
|
|
["[16:39:12] 游戏已退出，BetterGI 自动停止截图器"],
|
|
)
|
|
|
|
def test_preserves_multiline_message_inside_one_log_entry(self):
|
|
raw_lines = [
|
|
"[10:00:00.001] [ERR] [Primary:S1:P1:T2] BetterGI.Component\n",
|
|
"第一行\n",
|
|
"第二行\n",
|
|
"[10:00:01.999] [INF] BetterGI.OtherComponent\n",
|
|
"下一条\n",
|
|
]
|
|
|
|
self.assertEqual(
|
|
BgiLogMonitor.format_display_lines(raw_lines),
|
|
["[10:00:00] 第一行\n第二行", "[10:00:01] 下一条"],
|
|
)
|
|
|
|
def test_frontend_reader_falls_back_to_latest_log_after_midnight(self):
|
|
with TemporaryDirectory() as temp_dir:
|
|
log_dir = Path(temp_dir) / "log"
|
|
log_dir.mkdir()
|
|
(log_dir / "better-genshin-impact20260730.log").write_text(
|
|
"[23:55:25.160] [INF] [Primary:S1:P16720:T1785426828552] "
|
|
"BetterGenshinImpact.GameTask.TaskTriggerDispatcher\n"
|
|
"游戏已退出，BetterGI 自动停止截图器\n",
|
|
encoding="utf-8",
|
|
)
|
|
server = WebServer.__new__(WebServer)
|
|
server.config = SimpleNamespace(bettergi_work_dir=temp_dir)
|
|
|
|
with patch("app.danmu_queue.datetime") as mocked_datetime:
|
|
mocked_datetime.now.return_value.strftime.return_value = "20260731"
|
|
result = server._read_bgi_log(50)
|
|
|
|
self.assertEqual(
|
|
result,
|
|
["[23:55:25] 游戏已退出，BetterGI 自动停止截图器"],
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|