refactor: add equipment registry

This commit is contained in:
2026-06-04 19:46:09 +08:00
parent 6c109708b8
commit 53b9f48f7d
3 changed files with 55 additions and 8 deletions
@@ -0,0 +1,30 @@
from core.basic.equipment import get_equipment
from core.basic.equipment_registry import EquipmentRegistry
def test_get_equipment_returns_cached_default_version():
first = get_equipment('0')
second = get_equipment('0')
assert first is second
assert first.version == '0'
def test_get_equipment_falls_back_to_default_version():
equipment = get_equipment('missing-version')
assert equipment.version == '0'
def test_equipment_registry_caches_by_version():
calls = []
class Equipment:
def __init__(self, version):
calls.append(version)
self.version = version
registry = EquipmentRegistry(Equipment)
assert registry.get('0') is registry.get('0')
assert calls == ['0']