Add "clear" method.

maldoinc · Dec 28, 2023 · 631647d · 631647d
1 parent 7ebae4e
commit 631647d
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/test/test_container_override.py b/test/test_container_override.py
@@ -6,6 +6,7 @@
 
 from typing_extensions import Annotated
 from wireup import DependencyContainer, ParameterBag, Wire
+from wireup.ioc.override_manager import OverrideManager
 from wireup.ioc.types import ServiceOverride
 
 
@@ -84,3 +85,13 @@ def target(foo: FooBase):
                 self.assertEqual(svc.foo, "mock")
 
                 target()
+
+    def test_clear_services_removes_all(self):
+        overrides = {}
+        mock1 = MagicMock()
+        override_mgr = OverrideManager(overrides)
+        override_mgr.set(RandomService, new=mock1)
+        self.assertEqual(overrides, {(RandomService, None): mock1})
+
+        override_mgr.clear()
+        self.assertEqual(overrides, {})
diff --git a/wireup/ioc/override_manager.py b/wireup/ioc/override_manager.py
@@ -30,6 +30,10 @@ def delete(self, target: type, qualifier: ContainerProxyQualifierValue = None) -
         if (target, qualifier) in self.__active_overrides:
             del self.__active_overrides[target, qualifier]
 
+    def clear(self) -> None:
+        """Clear all active service overrides."""
+        self.__active_overrides.clear()
+
     @contextmanager
     def service(self, target: type, new: Any, qualifier: ContainerProxyQualifierValue = None) -> Iterator[None]:
         """Override the target service with new for the duration of the context manager.