Fix support for factory functions that return a different type (#37)

elfjes · Jan 19, 2023 · d981c9c · d981c9c
1 parent 7004413
commit d981c9c
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/src/gimme/__init__.py b/src/gimme/__init__.py
@@ -16,7 +16,7 @@
 from .resolvers import AttributeResolver, TypeHintingResolver
 from .types import DependencyInfo
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 __all__ = [
     "add",

diff --git a/src/gimme/repository.py b/src/gimme/repository.py
@@ -78,7 +78,7 @@ def create(self, key: t.Type[T], repo=None, kwargs=None) -> T:
 
         inst = self.resolve(info.factory, key=key, repo=repo, kwargs=combined_kwargs)
         if do_store:
-            self.add(inst)
+            self.add(inst, cls=info.cls)
         return inst
 
     def resolve(self, factory, key, repo=None, kwargs=None):

diff --git a/tests/test_gimme.py b/tests/test_gimme.py
@@ -255,7 +255,8 @@ class MyDummy:
             pass
 
         repo.register(MyType, factory=MyDummy)
-        assert isinstance(repo.get(MyType), MyDummy)
+        repo.create(MyType)
+        assert MyType in repo
 
     def test_can_lookup_obj_by_string(self, repo):
         obj = object()