Improved test coverage

Author: syrusakbary

graphene/utils/resolver_from_annotations.py

@@ -0,0 +1,65 @@
+import pytest
+from .. import deprecated
+from ..deprecated import deprecated as deprecated_decorator, warn_deprecation
+
+
+def test_warn_deprecation(mocker):
+    mocker.patch.object(deprecated.warnings, 'warn')
+    
+    warn_deprecation("OH!")
+    deprecated.warnings.warn.assert_called_with('OH!', stacklevel=2, category=DeprecationWarning)
+
+
+def test_deprecated_decorator(mocker):
+    mocker.patch.object(deprecated, 'warn_deprecation')
+    
+    @deprecated_decorator
+    def my_func():
+        return True
+
+    result = my_func()
+    assert result
+    deprecated.warn_deprecation.assert_called_with("Call to deprecated function my_func.")
+
+
+def test_deprecated_class(mocker):
+    mocker.patch.object(deprecated, 'warn_deprecation')
+    
+    @deprecated_decorator
+    class X:
+        pass
+
+    result = X()
+    assert result
+    deprecated.warn_deprecation.assert_called_with("Call to deprecated class X.")
+
+
+def test_deprecated_decorator_text(mocker):
+    mocker.patch.object(deprecated, 'warn_deprecation')
+    
+    @deprecated_decorator("Deprecation text")
+    def my_func():
+        return True
+
+    result = my_func()
+    assert result
+    deprecated.warn_deprecation.assert_called_with("Call to deprecated function my_func (Deprecation text).")
+
+
+def test_deprecated_class_text(mocker):
+    mocker.patch.object(deprecated, 'warn_deprecation')
+    
+    @deprecated_decorator("Deprecation text")
+    class X:
+        pass
+
+    result = X()
+    assert result
+    deprecated.warn_deprecation.assert_called_with("Call to deprecated class X (Deprecation text).")
+
+
+def test_deprecated_other_object(mocker):
+    mocker.patch.object(deprecated, 'warn_deprecation')
+    
+    with pytest.raises(TypeError) as exc_info:
+        deprecated_decorator({})

graphene/utils/tests/test_deprecated.py

@@ -4,10 +4,12 @@
 
 def test_resolve_only_args(mocker):
     mocker.patch.object(deprecated, 'warn_deprecation')
-    def resolver(*args, **kwargs):
-        return kwargs
+    def resolver(root, **args):
+        return root, args
 
     my_data = {'one': 1, 'two': 2}
 
-    wrapped = resolve_only_args(resolver)
+    wrapped_resolver = resolve_only_args(resolver)
     assert deprecated.warn_deprecation.called
+    result = wrapped_resolver(1, 2, a=3)
+    assert result == (1, {'a': 3})