Add tests for the middleware utils.

Author: jhgg

graphql/core/execution/middlewares/utils.py

@@ -22,7 +22,7 @@ def resolver_has_tag(f, tag):
     return tag in f._resolver_tags
 
 
-def merge_resolver_resolver_tags(source_resolver, target_resolver):
+def merge_resolver_tags(source_resolver, target_resolver):
     if not hasattr(source_resolver, '_resolver_tags'):
         return target_resolver
 

tests/core_execution/test_middleware.py

@@ -0,0 +1,49 @@
+from graphql.core.execution.middlewares.utils import tag_resolver, resolver_has_tag, merge_resolver_tags
+
+
+def test_tag_resolver():
+    resolver = lambda: None
+
+    tag_resolver(resolver, 'test')
+    assert resolver_has_tag(resolver, 'test')
+    assert not resolver_has_tag(resolver, 'not test')
+
+
+def test_merge_resolver_tags():
+    a = lambda: None
+    b = lambda: None
+
+    tag_resolver(a, 'a')
+    tag_resolver(b, 'b')
+
+    merge_resolver_tags(a, b)
+
+    assert resolver_has_tag(a, 'a')
+    assert not resolver_has_tag(a, 'b')
+
+    assert resolver_has_tag(b, 'a')
+    assert resolver_has_tag(b, 'b')
+
+
+def test_resolver_has_tag_with_untagged_resolver():
+    a = lambda: None
+
+    assert not resolver_has_tag(a, 'anything')
+
+
+def test_merge_resolver_from_untagged_source():
+    a = lambda: None
+    b = lambda: None
+
+    merge_resolver_tags(a, b)
+    assert not hasattr(b, '_resolver_tags')
+
+
+def test_merge_resolver_to_untagged_target():
+    a = lambda: None
+    b = lambda: None
+
+    tag_resolver(a, 'test')
+    merge_resolver_tags(a, b)
+
+    assert resolver_has_tag(b, 'test')