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