Added some tests.

taskiq-python · Sep 29, 2024 · 5618873 · 5618873
1 parent a5c2948
commit 5618873
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/test_graph.py b/tests/test_graph.py
@@ -334,13 +334,15 @@ def dep(info: ParamInfo = Depends()) -> ParamInfo:
     def target(my_test_param: ParamInfo = Depends(dep)) -> None:
         return None
 
-    with DependencyGraph(target=target).sync_ctx() as g:
+    graph = DependencyGraph(target=target)
+    with graph.sync_ctx() as g:
         kwargs = g.resolve_kwargs()
 
     info: ParamInfo = kwargs["my_test_param"]
     assert info.name == "my_test_param"
     assert info.definition
     assert info.definition.annotation == ParamInfo
+    assert info.graph == graph
 
 
 def test_param_info_no_dependant() -> None:
@@ -349,12 +351,14 @@ def test_param_info_no_dependant() -> None:
     def target(info: ParamInfo = Depends()) -> None:
         return None
 
-    with DependencyGraph(target=target).sync_ctx() as g:
+    graph = DependencyGraph(target=target)
+    with graph.sync_ctx() as g:
         kwargs = g.resolve_kwargs()
 
     info: ParamInfo = kwargs["info"]
     assert info.name == ""
     assert info.definition is None
+    assert info.graph == graph
 
 
 def test_class_based_dependencies() -> None: