Add some tests

This commit is contained in:
Jonathan Kim 2018-01-05 17:44:26 +00:00 committed by Jonathan Kim
parent a78d2fe0ad
commit 17c225fda2

View File

@ -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():