def test_parse_flatfiledefinitions_complex():
"""Tests the parse_flatfiledefinitions function by supplying
testdata to the function and asserting that the correct number of
ADDML sections are returned per addmldata. Also asserts that the
correct number of addml data files are returned and that the
sections have correct name attributes and that various elements
contain right data for each addml data file respectively.
"""
addml = 'tests/data/addml_complex.xml'
i = 0
for addmls in s.parse_flatfiledefinitions(addml):
i = i + 1
assert f.flatfiledefinition_count(addmls) == 1
assert a.sections_count(addmls, 'flatFileType') == 1
assert a.sections_count(addmls, 'recordType') == 1
assert a.sections_count(addmls, 'fieldTypes') == 1
for ffdef in f.iter_flatfiledefinitions(addmls):
assert a.parse_name(ffdef) == 'testdef' + six.text_type(i)
if a.parse_name(ffdef) == 'testdef1':
assert f.flatfile_count(addmls) == 3
assert f.parse_charset(addmls) == 'UTF-8'
assert a.sections_count(addmls, 'fieldDefinition') == 3
elif a.parse_name(ffdef) == 'testdef2':
assert f.flatfile_count(addmls) == 2
assert f.parse_charset(addmls) == 'ISO-8859-15'
assert a.sections_count(addmls, 'fieldDefinition') == 2
elif a.parse_name(ffdef) == 'testdef3':
assert f.flatfile_count(addmls) == 1
assert f.parse_charset(addmls) == 'ASCII'
assert a.sections_count(addmls, 'fieldDefinition') == 3
assert i == 3
def test_flatfiledefinition_count():
"""Test flatfiledefinition_count by asserting that the number of
counted sections matches the testdata.
"""
file1 = f.definition_elems('flatFile', 'file1', reference='def1')
file2 = f.definition_elems('flatFile', 'file2', reference='def2')
file3 = f.definition_elems('flatFile', 'file3', reference='def3')
def1 = f.definition_elems('flatFileDefinition', 'def1', reference='type1')
def2 = f.definition_elems('flatFileDefinition', 'def2', reference='type2')
xml = a.addml(child_elements=[file1, file2, file3, def1, def2])
assert f.flatfiledefinition_count(xml) == 2
def parse_flatfiledefinitions(path):
"""Parses ADDML data and splits the data into new ADDML data
files for each flatFileDefinition in the original data file.
Returns the ADDML data for each created file.
"""
root = readfile(path).getroot()
addmldata = root
count = flatfiledefinition_count(root)
for flatfiledef in iter_flatfiledefinitions(root):
if count > 1:
addmldata = create_new_addml(root, flatfiledef)
yield addmldata
def test_create_new_addml_simple():
"""Tests the create_new_addml function by supplying testdata to the
function and asserting that the correct number of ADDML sections are
returned.
"""
addml = 'tests/data/addml_simple.xml'
root = h.readfile(addml)
for ffdef in f.iter_flatfiledefinitions(root):
testdef = ffdef
addml_new = s.create_new_addml(root, testdef)
assert f.flatfile_count(addml_new) == 1
assert f.flatfiledefinition_count(addml_new) == 1
assert a.sections_count(addml_new, 'flatFileType') == 1
assert a.sections_count(addml_new, 'recordType') == 1
assert a.sections_count(addml_new, 'fieldTypes') == 1
def test_parse_flatfiledefinitions_simple():
"""Tests the parse_flatfiledefinitions function by supplying
testdata to the function and asserting that the correct number of
ADDML sections are returned per addmldata. Also asserts that the
correct number of addml data files are returned.
"""
addml = 'tests/data/addml_simple.xml'
i = 0
for addmls in s.parse_flatfiledefinitions(addml):
i = i + 1
assert f.flatfile_count(addmls) == 1
assert f.flatfiledefinition_count(addmls) == 1
assert a.sections_count(addmls, 'flatFileType') == 1
assert a.sections_count(addmls, 'recordType') == 1
assert a.sections_count(addmls, 'fieldTypes') == 1
assert i == 1
def test_create_new_addml_complex():
"""Tests the create_new_addml function by supplying testdata to the
function and asserting that the correct number of ADDML sections are
returned and that various elements contain the correct data.
"""
addml = 'tests/data/addml_complex.xml'
root = h.readfile(addml)
for ffdef in f.iter_flatfiledefinitions(root):
testdef = a.find_section_by_name(root, 'flatFileDefinition',
'testdef2')
addml_new = s.create_new_addml(root, testdef)
assert f.flatfile_count(addml_new) == 2
assert f.flatfiledefinition_count(addml_new) == 1
assert a.sections_count(addml_new, 'flatFileType') == 1
assert a.sections_count(addml_new, 'recordType') == 1
assert a.sections_count(addml_new, 'fieldTypes') == 1
assert f.flatfile_count(addml_new) == 2
assert f.parse_charset(addml_new) == 'ISO-8859-15'
assert a.sections_count(addml_new, 'fieldDefinition') == 2
def test_parse_flatfiledefinitions_medium():
"""Tests the parse_flatfiledefinitions function by supplying
testdata to the function and asserting that the correct number of
ADDML sections are returned per addmldata. Also asserts that the
correct number of addml data files are returned and that the
sections have correct name attributes.
"""
addml = 'tests/data/addml_medium.xml'
i = 0
for addmls in s.parse_flatfiledefinitions(addml):
i = i + 1
assert f.flatfile_count(addmls) == 1
assert f.flatfiledefinition_count(addmls) == 1
assert a.sections_count(addmls, 'flatFileType') == 1
assert a.sections_count(addmls, 'recordType') == 1
assert a.sections_count(addmls, 'fieldTypes') == 1
for flatfile in f.iter_flatfiles(addmls):
assert \
a.parse_name(flatfile) == 'csvfile' + six.text_type(i) + '.csv'
assert i == 3