def test_covid_generate(self):
# self.create_test_data()
report = reports.CovidReport()
if os.path.exists(report.target_file_path):
os.remove(report.target_file_path)
self.assertFalse(os.path.exists(report.target_file_path))
report.generate_spread_sheet()
self.assertTrue(os.path.exists(report.target_file_path))
sheet = report.get_worksheet(title=report.__sheets__[0]["title"])
strings_array = sheet.str_table
string_table = sheet.table
# The SharedStringTable stores strings for the xlswriter, but it hast to be "sorted" before strings can be
# retirieved from it.
# --- In this case the strtings_array was already sorted when the workbook gets closed at the end of the
# Generate_spread_sheet function
# strings_array._sort_string_data()
self.assert_project_data(strings_array, string_table, 2, self.prj_1)
self.assert_project_data(strings_array, string_table, 3, self.prj_2)
def test_covid_generate_by_fiscal_year(self):
# self.create_test_data()
report = reports.CovidReport(fiscal_year=self.fy_current.pk)
if os.path.exists(report.target_file_path):
os.remove(report.target_file_path)
self.assertFalse(os.path.exists(report.target_file_path))
report.generate_spread_sheet()
self.assertTrue(os.path.exists(report.target_file_path))
sheet = report.get_worksheet(title=report.__sheets__[0]["title"])
strings_array = sheet.str_table
string_table = sheet.table
self.assertEqual(len(string_table), 3)
def test_covid_generate_by_section(self):
# self.create_test_data()
report = reports.CovidReport(sections=str(self.sec_2.pk))
if os.path.exists(report.target_file_path):
os.remove(report.target_file_path)
self.assertFalse(os.path.exists(report.target_file_path))
report.generate_spread_sheet()
self.assertTrue(os.path.exists(report.target_file_path))
sheet = report.get_worksheet(title=report.__sheets__[0]["title"])
strings_array = sheet.str_table
string_table = sheet.table
self.assertEqual(len(string_table), 3)
self.assert_project_data(strings_array, string_table, 2, self.prj_2)
def setUp(self):
super().setUp()
self.report = reports.CovidReport()
self.create_test_data()