mirror of https://github.com/ansible/ansible.git
Teaching objects to load themselves, making the JSON/YAML parsing ambidexterous.
parent
c75aeca435
commit
56b6cb5328
@ -0,0 +1 @@
|
|||||||
|
|
@ -0,0 +1,85 @@
|
|||||||
|
# TODO: header
|
||||||
|
|
||||||
|
import unittest
|
||||||
|
from ansible.parsing import load
|
||||||
|
from ansible.errors import AnsibleParserError
|
||||||
|
|
||||||
|
import json
|
||||||
|
|
||||||
|
class MockFile(file):
|
||||||
|
|
||||||
|
def __init__(self, ds, method='json'):
|
||||||
|
self.ds = ds
|
||||||
|
self.method = method
|
||||||
|
|
||||||
|
def read(self):
|
||||||
|
if method == 'json':
|
||||||
|
return json.dumps(ds)
|
||||||
|
elif method == 'yaml':
|
||||||
|
return yaml.dumps(ds)
|
||||||
|
elif method == 'fail':
|
||||||
|
return """
|
||||||
|
AAARGGGGH
|
||||||
|
THIS WON'T PARSE !!!
|
||||||
|
NOOOOOOOOOOOOOOOOOO
|
||||||
|
"""
|
||||||
|
else:
|
||||||
|
raise Exception("untestable serializer")
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
class TestGeneralParsing(unittest.TestCase):
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def setUp(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def tearDown(self):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def parse_json_from_string(self):
|
||||||
|
input = """
|
||||||
|
{
|
||||||
|
"asdf" : "1234",
|
||||||
|
"jkl" : 5678
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
output = load(input)
|
||||||
|
assert output['asdf'] == '1234'
|
||||||
|
assert output['jkl'] == 5678
|
||||||
|
|
||||||
|
def parse_json_from_file(self):
|
||||||
|
output = load(MockFile(dict(a=1,b=2,c=3)),'json')
|
||||||
|
assert ouput == dict(a=1,b=2,c=3)
|
||||||
|
|
||||||
|
def parse_yaml_from_dict(self):
|
||||||
|
input = """
|
||||||
|
asdf: '1234'
|
||||||
|
jkl: 5678
|
||||||
|
"""
|
||||||
|
output = load(input)
|
||||||
|
assert output['asdf'] == '1234'
|
||||||
|
assert output['jkl'] == 5678
|
||||||
|
|
||||||
|
def parse_yaml_from_file(self):
|
||||||
|
output = load(MockFile(dict(a=1,b=2,c=3),'yaml'))
|
||||||
|
assert output == dict(a=1,b=2,c=3)
|
||||||
|
|
||||||
|
def parse_fail(self):
|
||||||
|
input = """
|
||||||
|
TEXT
|
||||||
|
***
|
||||||
|
NOT VALID
|
||||||
|
"""
|
||||||
|
self.failUnlessRaises(load(input), AnsibleParserError)
|
||||||
|
|
||||||
|
def parse_fail_from_file(self):
|
||||||
|
self.failUnlessRaises(load(MockFile(None,'fail')), AnsibleParserError)
|
||||||
|
|
||||||
|
def parse_fail_invalid_type(self):
|
||||||
|
self.failUnlessRaises(3000, AnsibleParsingError)
|
||||||
|
self.failUnlessRaises(dict(a=1,b=2,c=3), AnsibleParserError)
|
||||||
|
|
@ -1 +1,18 @@
|
|||||||
# TODO: header
|
# TODO: header
|
||||||
|
|
||||||
|
from ansible.errors import AnsibleError, AnsibleInternalError
|
||||||
|
|
||||||
|
def load(self, data):
|
||||||
|
|
||||||
|
if instanceof(data, file):
|
||||||
|
fd = open(f)
|
||||||
|
data = fd.read()
|
||||||
|
fd.close()
|
||||||
|
|
||||||
|
if instanceof(data, basestring):
|
||||||
|
try:
|
||||||
|
return json.loads(data)
|
||||||
|
except:
|
||||||
|
return safe_load(data)
|
||||||
|
|
||||||
|
raise AnsibleInternalError("expected file or string, got %s" % type(data))
|
||||||
|
Loading…
Reference in New Issue