mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-05-21 07:26:49 +00:00
106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding:utf-8 -*-
|
||
"""
|
||
TemplateContextBuilder 的并发安全单元测试。
|
||
|
||
历史上 builder 持有 ``self._context`` 实例字段，``build()`` 内 ``clear()`` →
|
||
``_add_*`` → 推导式返回这一序列在 ``TRANSFER_THREADS > 1`` 下会被多线程相互
|
||
覆盖，导致同一 builder 实例并发调用产生互相串味的 rename_dict。本测试在多
|
||
线程下连续调用 ``build()``，校验每个线程拿到的字典只反映自己的入参。
|
||
"""
|
||
import threading
|
||
import unittest
|
||
|
||
from app.helper.message import TemplateContextBuilder
|
||
|
||
|
||
class TemplateContextBuilderConcurrencyTest(unittest.TestCase):
|
||
"""
|
||
使用 8 个线程并发调用同一 TemplateContextBuilder 实例的 build()，
|
||
确保各自的 file_extension / 自定义 kwargs 不会被其它线程覆盖。
|
||
"""
|
||
|
||
THREAD_COUNT = 8
|
||
ITERATIONS_PER_THREAD = 200
|
||
|
||
def test_concurrent_build_no_cross_contamination(self):
|
||
builder = TemplateContextBuilder()
|
||
errors = []
|
||
|
||
def worker(tag: int) -> None:
|
||
try:
|
||
for _ in range(self.ITERATIONS_PER_THREAD):
|
||
ctx = builder.build(
|
||
file_extension=f".{tag}",
|
||
marker=tag,
|
||
)
|
||
self.assertEqual(ctx.get("fileExt"), f".{tag}")
|
||
self.assertEqual(ctx.get("marker"), tag)
|
||
except AssertionError as exc:
|
||
errors.append(exc)
|
||
|
||
threads = [
|
||
threading.Thread(target=worker, args=(i,), name=f"builder-{i}")
|
||
for i in range(self.THREAD_COUNT)
|
||
]
|
||
for t in threads:
|
||
t.start()
|
||
for t in threads:
|
||
t.join()
|
||
|
||
self.assertFalse(
|
||
errors,
|
||
msg=f"检测到并发串味，共 {len(errors)} 条；首个错误：{errors[0] if errors else ''}",
|
||
)
|
||
|
||
def test_build_returns_independent_dicts(self):
|
||
"""
|
||
即便不开线程，连续两次 build() 也应当返回相互独立的 dict 实例，
|
||
避免无状态化后调用方误以为返回的还是 builder 内部共享对象。
|
||
"""
|
||
builder = TemplateContextBuilder()
|
||
first = builder.build(file_extension=".a", marker=1)
|
||
second = builder.build(file_extension=".b", marker=2)
|
||
self.assertIsNot(first, second)
|
||
self.assertEqual(first.get("fileExt"), ".a")
|
||
self.assertEqual(second.get("fileExt"), ".b")
|
||
# 第二次调用不应反向污染第一次的结果
|
||
self.assertEqual(first.get("marker"), 1)
|
||
|
||
def test_build_exposes_video_bit_from_meta(self):
|
||
"""
|
||
模板上下文应提供独立 videoBit 字段，避免用户只能从 videoCodec 中手工拆位深。
|
||
"""
|
||
meta = type("FakeMeta", (), {})()
|
||
meta.begin_episode = None
|
||
meta.title = "Movie.2024.1080p.x265.10bit.mkv"
|
||
meta.name = "Movie"
|
||
meta.en_name = "Movie"
|
||
meta.year = "2024"
|
||
meta.season_seq = ""
|
||
meta.season = ""
|
||
meta.episode_seqs = ""
|
||
meta.episode = ""
|
||
meta.part = None
|
||
meta.customization = None
|
||
meta.fps = None
|
||
meta.resource_type = None
|
||
meta.resource_effect = None
|
||
meta.edition = ""
|
||
meta.resource_pix = "1080p"
|
||
meta.resource_term = "1080p"
|
||
meta.resource_team = None
|
||
meta.video_encode = "x265 10bit"
|
||
meta.video_bit = "10bit"
|
||
meta.audio_encode = "AAC"
|
||
meta.web_source = None
|
||
|
||
context = TemplateContextBuilder().build(meta=meta)
|
||
|
||
self.assertEqual(context.get("videoCodec"), "x265 10bit")
|
||
self.assertEqual(context.get("videoBit"), "10bit")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|