Selector contexts

.gitignore

@@ -21,3 +21,4 @@ dist/
 .venv
 uv.lock
 .agents
+site

tests/providers/test_base.py

@@ -88,6 +88,25 @@ def test_register_with_mixed_items() -> None:
     assert parent in child_1._children, "Expected child_1._children to contain parent"
 
 
+def test_get_resolution_dependencies_registers_provider_arguments() -> None:
+    dependency = DummyProvider()
+    provider = Singleton(lambda value: value, dependency)
+
+    resolution_dependencies = provider.get_resolution_dependencies()
+
+    assert resolution_dependencies == frozenset({dependency})
+    assert isinstance(resolution_dependencies, frozenset)
+
+
+async def test_default_resolution_contexts_have_no_runtime_dependencies() -> None:
+    provider = DummyProvider()
+
+    async with provider.resolution_context() as async_dependencies:
+        assert async_dependencies == ()
+    with provider.resolution_context_sync() as sync_dependencies:
+        assert sync_dependencies == ()
+
+
 def test_invalidate_scope_init_order_handles_duplicate_descendants() -> None:
     root = DummyProvider()
     left = DummyProvider()
@@ -100,13 +119,11 @@ def test_invalidate_scope_init_order_handles_duplicate_descendants() -> None:
     right.add_child_provider(shared)
 
     for provider in (root, left, right, shared):
-        provider._scope_context_init_order = ()
         provider._scope_init_order = ()
 
     root._invalidate_scope_init_order()
 
     for provider in (root, left, right, shared):
-        assert provider._scope_context_init_order is None
         assert provider._scope_init_order is None
 
 

tests/providers/test_selector.py

@@ -140,6 +140,45 @@ async def test_selector_with_provider_selector_async() -> None:
     assert (await StringProviderSelectorContainer.selector.resolve()) == "Provider 1"
 
 
+def test_selector_exposes_only_its_key_provider_as_static_dependency() -> None:
+    def _selector_key() -> typing.Iterator[str]:  # pragma: no cover
+        yield "selected"
+
+    selector_key = providers.ContextResource(_selector_key)
+    selected = providers.Object("value")
+    selector = providers.Selector(selector_key, selected=selected)
+
+    assert selector.get_resolution_dependencies() == frozenset({selector_key})
+    assert selector.get_resolution_dependencies() == frozenset({selector_key})
+
+    selector._deregister_arguments()
+
+    assert selector.get_resolution_dependencies() == frozenset({selector_key})
+
+
+def test_selector_resolution_context_exposes_and_pins_selected_provider() -> None:
+    expected_selection_count = 2
+    selected_key = "one"
+    selection_count = 0
+    one = providers.Object("one")
+    two = providers.Object("two")
+
+    def _select() -> str:
+        nonlocal selection_count
+        selection_count += 1
+        return selected_key
+
+    selector = providers.Selector(_select, one=one, two=two)
+
+    with selector.resolution_context_sync() as dependencies:
+        assert dependencies == (one,)
+        selected_key = "two"
+        assert selector.resolve_sync() == "one"
+
+    assert selector.resolve_sync() == "two"
+    assert selection_count == expected_selection_count
+
+
 class InvalidSelectorContainer(BaseContainer):
     selector = providers.Selector(
         None,  # type: ignore[arg-type]