class TestEndEnd(unittest.TestCase):
def setUp(self):
self.s = MySettings()
self.s.read_json_file(JSON_FILE)
self.s.INPUT_FILE = DEFAULT_INPUT_FILE
self.s.OUTPUT_FILE = DEFAULT_OUTPUT_FILE
self.parser = CsvParser(self.s)
# end setUp
def tearDown(self):
pass
def test_produces_output(self):
out_docx = DocxConfig(self.s)
self.parser.write_docx(out_docx)
out_docx.save(self.s.OUTPUT_FILE)
# At least confirm the file was modified
num_seconds = 3
self.assertTrue(time.time() - os.stat(self.s.OUTPUT_FILE).st_mtime <
num_seconds)
class TestMySettings(unittest.TestCase):
def setUp(self):
self.s = MySettings()
self.s.read_json_file(JSON_FILE)
self.s.INPUT_FILE = DEFAULT_INPUT_FILE
self.s.OUTPUT_FILE = DEFAULT_OUTPUT_FILE
self.parser = CsvParser(self.s)
# end setUp
def tearDown(self):
pass
def test_all_inds_includes_expected(self):
""" Confirm mandatory index fields are in all_inds"""
s = self.s
self.assertTrue(s.id_ind in s.all_inds)
self.assertTrue(s.heading_level_ind in s.all_inds)
self.assertTrue(s.heading_num_ind in s.all_inds)
self.assertTrue(s.heading_text_ind in s.all_inds)
self.assertTrue(s.body_text_ind in s.all_inds)
class TestParser(unittest.TestCase):
def setUp(self):
self.s = MySettings()
self.s.read_json_file(JSON_FILE)
self.s.INPUT_FILE = DEFAULT_INPUT_FILE
self.s.OUTPUT_FILE = DEFAULT_OUTPUT_FILE
self.ref_dict = build_ref_dict(self.s.INPUT_FILE)
self.parser = CsvParser(self.s)
self.clean_row = ['6',
'',
'',
'',
'',
'Text after second heading reference to section {#7}, {H7} (1.1, H2) and to section {#9},{H9} (1.1.1.1, H4)',
]
self.slashed_row = ['6',
'',
'',
'',
'',
'Text after\r second heading reference\r to section {#7}, {H7} (1.1, H2) and\r to section {#9},{H9} (1.1.1.1, H4)',
]
# end setUp
def tearDown(self):
pass
def test_read_json_file(self):
json_dict = self.s.json_file_to_dict(JSON_FILE)
for key in json_dict.keys():
self.assertTrue(hasattr(self.s, key),
msg='Testing key: %s' % key)
# err.write((self.slashed_row[:]).__str__())
def test_dont_change_clean_row(self):
cleaned_row = self.parser.clean_only(self.clean_row[:])
self.assertEqual(self.clean_row, cleaned_row)
def test_changes_backslash(self):
'''At least confirm that \r is 'changed' somehow '''
clean_dict = self.parser.get_clean_dict()
self.assertTrue(len(clean_dict) > 1,
'Constructed clean_dict does not have >1 entries')
cleaned_row = self.parser.clean_n_parse_tokens(self.slashed_row, True)
# err.write("\n"+"\n".join(cleaned_row))
err.write("\nTODO: add better testing of \\r cleanup\n")
self.assertNotEqual(self.slashed_row, cleaned_row)
def test_parse_token_throws_if_key_not_found(self):
large_num = 99999
with self.assertRaises(CrossRefError):
test_token = (self.s.l_delim + self.s.heading_text_symbol + str(large_num) + self.s.r_delim)
result = self.parser.parse_token(test_token)
def test_parse_token(self):
clean_dict = self.parser.get_clean_dict()
self.assertTrue(len(clean_dict) > 1,
'Constructed clean_dict does not have >1 entries')
self.parser.clean_dict = clean_dict
for key in clean_dict.keys():
for code in (self.s.heading_number_symbol, self.s.heading_text_symbol):
test_token = (self.s.l_delim +
code +
str(key) +
self.s.r_delim)
# try:
result = self.parser.parse_token(test_token)
# if len(result.value):
# err.write ( '%s: %s\n' % (test_token, result.value) )
if len(self.ref_dict[key]) > code:
self.assertEqual(result.value, self.ref_dict[key][code])
# except CrossRefError as ex:
# err.write ( '%s: %s\n' % (test_token, ex.message) )
# break
# end for
# end for
# end test_parse_token
def test_parse_string_with_token(self):
clean_dict = self.parser.get_clean_dict()
self.assertTrue(len(clean_dict) > 1,
'Constructed clean_dict does not have >1 entries')
self.parser.clean_dict = clean_dict
text = ('Text after second heading reference to section {#7}, {H7} ' +
'(1.1, H2) and to section {#9},{H9} (1.1.1.1, H4)')
expected_subst_text = (text.replace('{#7}', '1.1').replace('{H7}', 'H2')
).replace('{#9}', '1.1.1.1').replace('{H9}', 'H4')
subst_text = ''.join(self.parser.replace_tokens(text, None))
self.assertEqual(expected_subst_text, subst_text,
("Failed to get expected subst_text in %s" %
inspect.stack()[0][3]))
# end test_parse_string_with_token
def test_parse_string_with_token_incl_image(self):
clean_dict = self.parser.get_clean_dict()
self.assertTrue(len(clean_dict) > 1,
'Constructed clean_dict does not have >1 entries')
self.parser.clean_dict = clean_dict
text = ('Text with {bad_text} and reference to section {#7}, {H7} ' +
'(1.1, H2) and to section {#9},{H9} (1.1.1.1, H4)')
expected_subst_text = (text.replace('{#7}', '1.1').replace('{H7}', 'H2')
).replace('{#9}', '1.1.1.1').replace('{H9}', 'H4')
subst_list = self.parser.replace_tokens(text, None)
subst_list_image_ignored = []
for entry in subst_list:
if isinstance(entry, CsvParser.ParsedToken):
subst_list_image_ignored.append(self.s.l_delim +
entry.value +
self.s.r_delim)
else:
subst_list_image_ignored.append(entry)
subst_text = ''.join(subst_list_image_ignored)
self.assertEqual(expected_subst_text, subst_text,
("Failed to get expected subst_text in %s" %
inspect.stack()[0][3]))