1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
|
from ..resolver import (
attr_resolver,
dict_resolver,
dict_or_attr_resolver,
get_default_resolver,
set_default_resolver,
)
args = {}
context = None
info = None
demo_dict = {"attr": "value"}
class demo_obj:
attr = "value"
def test_attr_resolver():
resolved = attr_resolver("attr", None, demo_obj, info, **args)
assert resolved == "value"
def test_attr_resolver_default_value():
resolved = attr_resolver("attr2", "default", demo_obj, info, **args)
assert resolved == "default"
def test_dict_resolver():
resolved = dict_resolver("attr", None, demo_dict, info, **args)
assert resolved == "value"
def test_dict_resolver_default_value():
resolved = dict_resolver("attr2", "default", demo_dict, info, **args)
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() == dict_or_attr_resolver
def test_set_default_resolver_workd():
default_resolver = get_default_resolver()
set_default_resolver(dict_resolver)
assert get_default_resolver() == dict_resolver
set_default_resolver(default_resolver)
|