def test_empty_table_zero_alarms(self): """ Check that an empty table returns a 0 length list of items """ adh = AlarmDb(self.db_name) adh.delete_all_alarms() number_of_alarms = adh.get_number_of_alarms() all_alarms = adh.get_all_alarms() self.assertEqual(number_of_alarms, 0) self.assertEqual(len(all_alarms), 0)
def test_get_all_alarms(self): """ Adds 5 alarms to the db, then checks all are retrieved. Also test the get_number_of_alarms method. """ adh = AlarmDb(self.db_name) self.only_five_entries(adh) number_of_alarms = adh.get_number_of_alarms() all_alarms = adh.get_all_alarms() self.assertEqual(number_of_alarms, 5) self.assertEqual(number_of_alarms, len(all_alarms)) hour = 13 minute = 35 for alarm in all_alarms: self.assertEqual(hour, alarm.hour) self.assertEqual(minute, alarm.minute) hour += 1 minute += 1