mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-05-23 23:16:46 +00:00
73 lines
2.6 KiB
Python
73 lines
2.6 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)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|