Handle functools.partial in wireup.ioc.util.get_globals (#92)

* fix wireup.ioc.util.get_globals

* PR remarks

* PR remarks

* fix lint

* fix ruff

---------

Co-authored-by: MuriloScarpaSitonio <[email protected]>

Author: MuriloScarpaSitonio

test/unit/test_util.py

@@ -1,12 +1,20 @@
+import functools
 import inspect
 import unittest
+from typing import Callable
 
 import pytest
 from typing_extensions import Annotated
 from wireup import Inject
 from wireup.errors import WireupError
-from wireup.ioc.types import AnnotatedParameter, InjectableType, ParameterWrapper, ServiceQualifier
+from wireup.ioc.types import (
+    AnnotatedParameter,
+    InjectableType,
+    ParameterWrapper,
+    ServiceQualifier,
+)
 from wireup.ioc.util import (
+    get_globals,
     param_get_annotation,
 )
 
@@ -29,10 +37,14 @@ def inner(
         ): ...
 
         params = inspect.signature(inner)
-        self.assertEqual(param_get_annotation(params.parameters["_a"], globalns=globals()), AnnotatedParameter(str, d1))
+        self.assertEqual(
+            param_get_annotation(params.parameters["_a"], globalns=globals()),
+            AnnotatedParameter(str, d1),
+        )
         self.assertEqual(param_get_annotation(params.parameters["_b"], globalns=globals()), None)
         self.assertEqual(
-            param_get_annotation(params.parameters["_c"], globalns=globals()), AnnotatedParameter(str, None)
+            param_get_annotation(params.parameters["_c"], globalns=globals()),
+            AnnotatedParameter(str, None),
         )
         self.assertEqual(
             param_get_annotation(params.parameters["_d"], globalns=globals()),
@@ -65,3 +77,36 @@ def inner(_a: Annotated[str, Inject(), Inject(param="foo")]): ...
 
 class MyCustomClass:
     pass
+
+
+def _sample_function():
+    pass
+
+
+def test_returns_globals_for_class():
+    # GIVEN
+    cls = MyCustomClass
+
+    # WHEN
+    result = get_globals(cls)
+
+    # THEN
+    assert "_sample_function" in result
+    assert "MyCustomClass" in result
+
+
+@pytest.mark.parametrize(
+    "partial_func",
+    (
+        _sample_function,
+        functools.partial(_sample_function),
+        functools.partial(functools.partial(functools.partial(_sample_function))),
+    ),
+)
+def test_unwraps_functools_partial(partial_func: Callable):
+    # GIVEN
+    # WHEN
+    result = get_globals(partial_func)
+
+    # THEN
+    assert result is _sample_function.__globals__

wireup/ioc/util.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import functools
 import importlib
 import inspect
 import types
@@ -82,6 +83,10 @@ def get_globals(obj: type[Any] | Callable[..., Any]) -> dict[str, Any]:
     if isinstance(obj, type):
         return importlib.import_module(obj.__module__).__dict__
 
+    # Unwrap nested functools.partial to get the underlying function
+    while isinstance(obj, functools.partial):
+        obj = obj.func
+
     return obj.__globals__