def test_format_results(self):
"""Test format_results."""
test_dict = {
"nested": {
"bool": True,
"nested_key": "nested_value"
},
"test_key": "test_value",
}
mute_map = MutableMap(**test_dict.copy())
assert LookupHandler.format_results(test_dict) == test_dict
assert LookupHandler.format_results(mute_map) == test_dict
assert (LookupHandler.format_results(
test_dict, get="test_key") == test_dict["test_key"])
assert (LookupHandler.format_results(
mute_map, get="test_key") == mute_map.test_key)
assert (LookupHandler.format_results(
mute_map, get="nested") == mute_map.nested.data)
assert (LookupHandler.format_results(
mute_map, get="nested.nested_key") == mute_map.nested.nested_key)
assert LookupHandler.format_results(mute_map, get="nested.bool")
assert LookupHandler.format_results(mute_map,
transform="str") == json.dumps(
json.dumps(test_dict,
indent=0))
assert LookupHandler.format_results(mute_map,
transform="str",
indent=2) == json.dumps(
json.dumps(test_dict,
indent=2))
assert (LookupHandler.format_results(mute_map,
get="nested.bool",
transform="str") == '"True"')
with pytest.raises(TypeError):
LookupHandler.format_results(["something"], get="key")
def test_format_results(self):
"""Test format_results."""
test_dict = {
'nested': {
'bool': True,
'nested_key': 'nested_value'
},
'test_key': 'test_value'
}
mute_map = MutableMap(**test_dict.copy())
assert LookupHandler.format_results(test_dict) == test_dict
assert LookupHandler.format_results(mute_map) == test_dict
assert LookupHandler.format_results(
test_dict, get='test_key') == test_dict['test_key']
assert LookupHandler.format_results(
mute_map, get='test_key') == mute_map.test_key
assert LookupHandler.format_results(
mute_map, get='nested') == mute_map.nested.data
assert LookupHandler.format_results(
mute_map, get='nested.nested_key') == mute_map.nested.nested_key
assert LookupHandler.format_results(mute_map, get='nested.bool')
assert LookupHandler.format_results(mute_map,
transform='str') == json.dumps(
json.dumps(test_dict,
indent=0))
assert LookupHandler.format_results(mute_map,
transform='str',
indent=2) == json.dumps(
json.dumps(test_dict,
indent=2))
assert LookupHandler.format_results(mute_map,
get='nested.bool',
transform='str') == '"True"'
with self.assertRaises(TypeError):
LookupHandler.format_results(['something'], get='key')