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_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
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
