def test__add_columns(self):
column_set = ReportColumnSet()
self.assertEqual(len(column_set), 0)
column_set.add_columns(sim_columns())
self.assertEqual(len(column_set), 2)
for x in column_set.report_columns:
self.assertTrue(isinstance(x, dict))
self.assertEqual(column_set.report_columns[0]['number'], 1)
self.assertEqual(column_set.report_columns[1]['number'], 2)
# Additional columns can be added, the number fields should be set
# properly.
column_set.add_columns(sim_columns())
self.assertEqual(len(column_set), 4)
self.assertEqual(column_set.report_columns[2]['number'], 3)
self.assertEqual(column_set.report_columns[3]['number'], 4)
def test__sorted(self):
column_set = ReportColumnSet()
self.assertEqual(column_set.sorted(), [])
column_set.add_columns(sim_columns())
sorted_set = column_set.sorted()
self.assertEqual(sorted_set[0]['heading'], 'ID')
self.assertEqual(sorted_set[1]['heading'], 'Name')
# Reverse the numbers
column_set.report_columns[0]['number'] = 2
column_set.report_columns[1]['number'] = 1
sorted_set = column_set.sorted()
self.assertEqual(sorted_set[0]['heading'], 'Name')
self.assertEqual(sorted_set[1]['heading'], 'ID')
