签入版本

2026-09-12 18:24:41 +08:00
commit fb2b7d9538
852 changed files with 115466 additions and 0 deletions
@@ -0,0 +1,102 @@
+"""dict_util 字典读取工具：列表/映射/单值读取，进程内缓存与失效行为。"""
+from types import SimpleNamespace
+
+from utils import dict_util
+from utils.dict_util import (
+    get_dict_items,
+    get_dict_label,
+    get_dict_map,
+    invalidate_dict,
+    invalidate_dicts_by_ids,
+)
+
+
+def _clear_cache():
+    dict_util._dict_cache.clear()
+
+
+def _rows():
+    # 刻意乱序：女(sort=5)、男(sort=1)、保密(sort=10)，校验按 sort+id 升序输出
+    return [
+        SimpleNamespace(id=2, name='女', value='2', sort=5),
+        SimpleNamespace(id=1, name='男', value='1', sort=1),
+        SimpleNamespace(id=3, name='保密', value='3', sort=10),
+    ]
+
+
+class _FakeQuery:
+    def __init__(self, rows):
+        self._rows = rows
+
+    def all(self):
+        return self._rows
+
+
+def _patch_repo(monkeypatch, dict_obj):
+    monkeypatch.setattr(dict_util.dict_repo, 'get_one', lambda **kw: dict_obj)
+    monkeypatch.setattr(dict_util.dictitem_repo, 'filter_by', lambda **kw: _FakeQuery(_rows()))
+
+
+# ============ 读取 ============
+
+def test_get_dict_map_maps_value_to_label(monkeypatch):
+    _clear_cache()
+    _patch_repo(monkeypatch, SimpleNamespace(id=1, code='gender', name='性别'))
+    assert get_dict_map('gender') == {'1': '男', '2': '女', '3': '保密'}
+
+
+def test_get_dict_items_sorted_by_sort_then_id(monkeypatch):
+    _clear_cache()
+    _patch_repo(monkeypatch, SimpleNamespace(id=1, code='gender', name='性别'))
+    assert [it['value'] for it in get_dict_items('gender')] == ['1', '2', '3']
+
+
+def test_get_dict_label_normalizes_value_to_str(monkeypatch):
+    _clear_cache()
+    _patch_repo(monkeypatch, SimpleNamespace(id=1, code='gender', name='性别'))
+    # 用户存整数 1，字典项 value 是字符串 '1'，需自动转字符串匹配
+    assert get_dict_label('gender', 1) == '男'
+    assert get_dict_label('gender', '2') == '女'
+    assert get_dict_label('gender', 99) is None
+    assert get_dict_label('gender', 99, '未知') == '未知'
+
+
+def test_get_dict_map_unknown_code_returns_empty(monkeypatch):
+    _clear_cache()
+    _patch_repo(monkeypatch, None)
+    assert get_dict_items('nope') == []
+    assert get_dict_map('nope') == {}
+
+
+# ============ 缓存 ============
+
+def test_cache_hit_skips_db(monkeypatch):
+    _clear_cache()
+    _patch_repo(monkeypatch, SimpleNamespace(id=1, code='gender', name='性别'))
+    get_dict_map('gender')
+    calls = []
+    monkeypatch.setattr(dict_util.dict_repo, 'get_one', lambda **kw: calls.append(1) or None)
+    get_dict_map('gender')  # 命中缓存，不再访问 DB
+    assert calls == []
+
+
+def test_invalidate_dict_clears_cache(monkeypatch):
+    _clear_cache()
+    _patch_repo(monkeypatch, SimpleNamespace(id=1, code='gender', name='性别'))
+    get_dict_map('gender')
+    invalidate_dict('gender')
+    assert dict_util._dict_cache == {}
+
+
+def test_invalidate_dicts_by_ids_clears_cached_codes(monkeypatch):
+    _clear_cache()
+    _patch_repo(monkeypatch, SimpleNamespace(id=1, code='gender', name='性别'))
+    get_dict_map('gender')  # 先填充缓存
+
+    class _FakeRows:
+        def all(self):
+            return [SimpleNamespace(code='gender')]
+
+    monkeypatch.setattr(dict_util.dict_repo, 'filter', lambda *criteria: _FakeRows())
+    invalidate_dicts_by_ids({1})
+    assert dict_util._dict_cache == {}