rougail/tests/test_others.py

162 lines
6.2 KiB
Python

from pytest import raises
from lxml.etree import DTD
from os.path import isfile
from copy import copy
from rougail import RougailConvert, RougailConfig
from rougail.error import DictConsistencyError
def parse_dtd_subelement(obj_name, content, default_required=False):
if not content:
return
if content.type == 'pcdata':
if obj_name in ['value', 'param']:
type_ = 'any'
else:
type_ = 'str'
yield {'name': 'text', 'type': type_, 'required': False}
elif content.name:
yield {'name': content.name, 'type': None, 'required': default_required}
if content.left:
if content.left.type == 'pcdata':
# choice
yield {'name': 'text', 'type': 'any', 'required': False}
else:
yield {'name': content.left.name, 'type': None, 'required': default_required}
if content.right:
if content.occur == 'once':
if content.right and content.right.name:
yield {'name': content.right.name, 'type': None, 'required': default_required}
elif content.occur == 'mult':
yield from parse_dtd_subelement(obj_name, content.right, default_required=False)
elif content.occur == 'plus':
yield from parse_dtd_subelement(obj_name, content.right, default_required=False)
else:
raise Exception('pffff')
def parse_dtd(elt_name, elts, space=0):
elt = elts.get(elt_name)
if not elt:
return ''
schema = ''
if space:
if elt_name not in ['variable', 'family']:
schema += " " * space + "type: seq\n"
schema += " " * space + "sequence:\n"
space = space + 2
schema += " " * space + "- type: map\n"
space = space + 2
else:
schema += " " * space + "type: map\n"
schema += " " * space + "mapping:\n"
if elt_name == 'variables':
# type
subspace = space + 2
schema += " " * subspace + "variable:\n"
schema += parse_dtd('variable', elts, subspace + 2)
# variables
if subspace < 50:
schema += " " * subspace + "family:\n"
schema += parse_dtd('family', elts, subspace + 2)
subspace = subspace + 4
schema += " " * subspace + "variables:\n"
schema += parse_dtd(elt_name, elts, subspace + 2)
else:
schema += " " * space + "type: map\n"
schema += " " * space + "mapping:\n"
space = space + 2
if elt_name != 'variables':
if 'subelts' in elt:
for subelt in elt['subelts']:
schema += " " * space + subelt['name'] + ":\n"
subspace = space + 2
schema += " " * subspace + 'required: ' + {False: 'false', True: 'true'}[subelt['required']] + "\n"
if subelt['type']:
schema += " " * subspace + 'type: ' + subelt['type'] + "\n"
if subelt['name'] != 'family' or subspace < 26:
schema += parse_dtd(subelt['name'], elts, subspace)
for dtd_attr in elt['elt'].iterattributes():
schema += " " * space + dtd_attr.name + ':\n'
enum = list(dtd_attr.itervalues())
type_ = 'str'
if set(enum) == {'True', 'False'} or set(enum) == {'True', 'False', 'nil'}:
# it's a boolean
type_ = 'bool'
else:
type_ = 'str'
subspace = space + 2
schema += " " * subspace + "type: " + type_ + '\n'
if type_ != 'bool' and enum:
enumspace = subspace + 2
schema += " " * subspace + 'enum:\n'
for value in enum:
schema += " " * enumspace + '- "' + value + '"\n'
return schema
def test_valid_dtd_schema_equal():
with open(RougailConfig['dtdfilename'], 'r') as dtdfd:
dtd = DTD(dtdfd)
elts = {}
for dtd_elt in dtd.iterelements():
subelts = tuple(parse_dtd_subelement(dtd_elt.name, dtd_elt.content))
ret = {'elt': dtd_elt}
if subelts:
ret['subelts'] = subelts
elts[dtd_elt.name] = ret
schema = parse_dtd('rougail', elts)
if not isfile(RougailConfig['yamlschema_filename']):
with open(RougailConfig['yamlschema_filename'], 'w') as ymldf:
ymldf.write(schema)
with open(RougailConfig['yamlschema_filename'], 'r') as ymldf:
ori_schema = ymldf.read()
assert schema == ori_schema, 'DTD and YAML schema are different'
def test_no_dtd():
cfg = RougailConfig.copy()
cfg['dtdfilename'] = 'notexists.dtd'
with raises(IOError):
eolobj = RougailConvert(cfg)
def test_mode_invalid_default():
# default variable mode is not in modes_level
RougailConfig['dictionaries_dir'] = ['tests/personalize_mode/dictionary']
RougailConfig['modes_level'] = ('level1', 'level2')
eolobj = RougailConvert()
with raises(DictConsistencyError) as err:
eolobj.reflexion()
assert err.value.errno == 72
def test_mode_invalid_default_family():
# default family mode is not in modes_level
RougailConfig['dictionaries_dir'] = ['tests/personalize_mode/dictionary']
RougailConfig['modes_level'] = ('level1', 'level2')
RougailConfig['default_variable_mode'] = 'level1'
eolobj = RougailConvert()
with raises(DictConsistencyError) as err:
eolobj.reflexion()
assert err.value.errno == 73
def test_personalize_mode():
RougailConfig['dictionaries_dir'] = ['tests/personalize_mode/dictionary']
RougailConfig['modes_level'] = ('level1', 'level2')
RougailConfig['default_variable_mode'] = 'level1'
RougailConfig['default_family_mode'] = 'level1'
eolobj = RougailConvert()
eolobj.save(None)
def test_personalize_mode_unknown():
# a variable has an unknown mode
RougailConfig['dictionaries_dir'] = ['tests/personalize_mode/dictionary']
RougailConfig['modes_level'] = ('level1',)
RougailConfig['default_variable_mode'] = 'level1'
RougailConfig['default_family_mode'] = 'level1'
eolobj = RougailConvert()
with raises(DictConsistencyError) as err:
eolobj.reflexion()
assert err.value.errno == 71