From 17c225fda23fdd554911c692e41bc7ea087df0e5 Mon Sep 17 00:00:00 2001 From: Jonathan Kim Date: Fri, 5 Jan 2018 17:44:26 +0000 Subject: [PATCH] Add some tests --- graphene/types/tests/test_resolver.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/graphene/types/tests/test_resolver.py b/graphene/types/tests/test_resolver.py index 3be9a492..7ba530ae 100644 --- a/graphene/types/tests/test_resolver.py +++ b/graphene/types/tests/test_resolver.py @@ -2,6 +2,7 @@ from ..resolver import ( attr_resolver, dict_resolver, + dict_or_attr_resolver, get_default_resolver, set_default_resolver, ) @@ -37,8 +38,16 @@ def test_dict_resolver_default_value(): assert resolved == "default" +def test_dict_or_attr_resolver(): + resolved = dict_or_attr_resolver('attr', None, demo_dict, info, **args) + assert resolved == 'value' + + resolved = dict_or_attr_resolver('attr', None, demo_obj, info, **args) + assert resolved == 'value' + + def test_get_default_resolver_is_attr_resolver(): - assert get_default_resolver() == attr_resolver + assert get_default_resolver() == dict_or_attr_resolver def test_set_default_resolver_workd():