Coverage for node / src / stigmem_node / plugins / testing.py: 100%

1"""Testing helpers for controlled plugin registry state.""" 

2 

3from __future__ import annotations 

4 

5from collections.abc import Iterator 

6from contextlib import contextmanager 

7 

8from .manifest import PluginManifest 

9from .registry import HookRegistry, set_registry 

10 

11 

12class TestPluginRegistry(HookRegistry): 

13 """Named test subclass for fixtures and downstream plugin tests.""" 

14 

15 

16@contextmanager 

17def stigmem_plugins(manifests: list[PluginManifest]) -> Iterator[TestPluginRegistry]: 

18 """Temporarily replace the process registry with test plugin manifests.""" 

19 registry = TestPluginRegistry() 

20 for manifest in manifests: 

21 registry.register_plugin(manifest) 

22 previous = set_registry(registry) 

23 try: 

24 yield registry 

25 finally: 

26 set_registry(previous)