class SchedulerTest1(unittest.TestCase):
def setUp(self):
self.groups = Groups("groups.json")
lat = 56.045000000000002
lon = 12.718400000000001
self.s = Scheduler(lat=lat, lon=lon, groups=self.groups)
self.s.parse_conf_file("schedule.json") # Prepped schedule with 11 schedule items and 2 rules
def test1_start_at_midnight(self):
""" This is a normal situation, where the schedules are reloaded when it's 00:00/a new day
"""
no = self.s.new_day("mo")
self.assertEqual(no, 9) # 9 items expected for Monday
# Start at midnight
item = self.s.get_first()
self.assertEqual(item["time"], "06:00") # Monday, first item should be 06:00
self.assertEqual(item["scenario"], "Plant lights on")
# Start later in the day, search for exact match in time
item = self.s.get_first("08:10")
self.assertEqual(item["time"], "08:10")
self.assertEqual(item["device"], "2222-2222")
def test2_start_in_evening(self):
""" Start later in the day. The search is inexact, expecting a schedule item at a later point-in-time
"""
self.s.new_day("mo")
item = self.s.get_first("21:35")
self.assertEqual(item["time"], "21:40")
item = self.s.get_next()
self.assertEqual(item["time"], "22:00")
def test2b_start_with_errors(self):
""" Start with invalid days. 0 scheduled items expected as result
"""
no = self.s.new_day("xx") # Needs to be "mo", "tu", etc.
self.assertEqual(no, 0)
no = self.s.new_day(9) # Needs to be 1..7
self.assertEqual(no, 0)
def test3_new_day(self):
""" Normal shift of day.
"""
self.s.new_day("mo")
self.s.schedules.list_full_day()
# Get last item for Monday
item = self.s.get_first("23:00")
self.assertEqual(item["time"], "23:00")
# This should fail, there are no more items
self.assertIsNone(self.s.get_next())
# It's midnight, let's shift over to a new day
no = self.s.new_day("tu")
self.assertEqual(no, 11) # 11 items expected for Tuesday
#self.s.schedules.list_full_day()
item = self.s.get_next()
self.assertEqual(item["time"], "06:00") # Tuesday, first item should be 06:00
item = self.s.get_next()
self.assertEqual(item["time"], "06:00") # Tuesday, second item should also be 06:00
def test3b_new_day(self):
""" Normal shift of day.
"""
self.s.new_day("mo")
# Get last item for Monday
item = self.s.get_first("23:00")
self.assertEqual(item["time"], "23:00")
# This should fail, there are no more items
self.assertIsNone(self.s.get_next())
# It's midnight, let's shift over to a new day
no = self.s.new_day("tu")
self.assertEqual(no, 11) # 11 items expected for Tuesday
item = self.s.get_next()
self.assertEqual(item["time"], "06:00") # Tuesday, first item should be 06:00
item = self.s.get_next()
self.assertEqual(item["time"], "06:00") # Tuesday, second item should also be 06:00
def test4_execute_rule(self):
""" Locate and execute a rule, use day index to locate Monday
"""
self.s.new_day(1)
item = self.s.get_first("00:00")
self.assertEqual(item["time"], "06:00")
r = item["rule"]
self.assertFalse(r.execute())
def test5_get_current_weekday(self):
""" Get current weekday. Cannot be asserted
"""
dl, d = self.s.get_weekday()
print ("dl={}, d={}".format(dl, d))
def test6_list_items(self):
""" List all items for this day
"""
self.s.new_day("mo")
item = self.s.schedules.list_full_day(now="04:01") # Before first
item = self.s.schedules.list_full_day(now="09:00") # In between
item = self.s.schedules.list_full_day(now="23:01") # After last
item = self.s.schedules.list_full_day() # Cannot be validated easily
self.s.new_day("fr")
item = self.s.schedules.list_full_day(now="00:00") # Should contain 2 sunset/sunrise based times
def test7_double_assign(self):
self.weekday = weekday = all_days[2-1]
self.assertEqual(self.weekday, "tu")
self.assertEqual(weekday, "tu")
def test8a_random_time(self):
self.s.new_day("mo")
item = self.s.get_first("00:00")
print("random times")
for i in range(1, 10):
time = self.s.schedules.schedules[0].get_random_time("08:00", 5)
self.assertIsNotNone(time)
print("{}".format(time))
def test8b_random_time(self):
self.s.random_minutes = 10
self.s.new_day("mo")
item = self.s.get_first("00:00")
item = self.s.schedules.list_full_day(now="08:00")
def tearDown(self):
pass
class SchedulerTest2(unittest.TestCase):
def setUp(self):
self.groups = Groups("groups.json")
self.s = Scheduler(12.7, 56.6, self.groups) # Coords for south of Sweden
self.s.parse_conf_file("winter.json") # Prepped schedule with 9 schedule items and 2 rules
def test1_start_at_midnight(self):
""" This is a normal situation, where the schedules are reloaded when it's 00:00/a new day
"""
no = self.s.new_day("mo")
self.assertEqual(no, 7) # 7 items expected for Monday
# Start at midnight
item = self.s.get_first()
self.assertEqual(item["time"], "06:15") # Monday, first item should be 06:15
self.assertEqual(item["scenario"], "ce52eea8-8889-4dc6-b578-7d049b5a64ab") # Plant lights on"
# Start later in the day, search for exact match in time
item = self.s.get_first("08:10")
self.assertEqual(item["time"], "15:00")
self.assertEqual(item["group"], "5555-5555")
def test2_start_in_evening(self):
""" Start later in the day. The search is inexact, expecting a schedule item at a later point-in-time
"""
self.s.new_day("mo")
item = self.s.get_first("21:35")
self.assertEqual(item["time"], "21:45")
item = self.s.get_next()
self.assertEqual(item["time"], "21:59")
def test2b_start_with_errors(self):
""" Start with invalid days. 0 scheduled items expected as result
"""
no = self.s.new_day("xx") # Needs to be "mo", "tu", etc.
self.assertEqual(no, 0)
no = self.s.new_day(9) # Needs to be 1..7
self.assertEqual(no, 0)
def test3_new_day(self):
""" Normal shift of day.
"""
self.s.new_day("mo")
# Get last item for Monday
item = self.s.get_first("22:00")
self.assertEqual(item["time"], "22:00")
# This should fail, there are no more items
self.assertIsNone(self.s.get_next())
# It's midnight, let's shift over to a new day
no = self.s.new_day("tu")
self.assertEqual(no, 10) # 10 items expected for Tuesday
item = self.s.get_next()
self.assertEqual(item["time"], "06:00") # Tuesday, first item should be 06:00
item = self.s.get_next()
self.assertEqual(item["time"], "06:00") # Tuesday, second item should also be 06:00
def test3_new_day2(self):
""" Normal shift of day.
"""
self.s.new_day("we")
# Get last item for Wednesday
item = self.s.get_first("22:00")
self.assertEqual(item["time"], "22:00")
# This should fail, there are no more items
self.assertIsNone(self.s.get_next())
# It's midnight, let's shift over to a new day
no = self.s.new_day("th")
self.assertEqual(no, 7) # 10 items expected for Thursday
item = self.s.get_next()
self.assertEqual(item["time"], "06:25") # Thursday, first item should be 06:25
item = self.s.get_next()
self.assertEqual(item["time"], "15:00") # Thursday, second item should also be 15:00
def test4_list(self):
""" List full day, incl sunrise/sunset calculation"""
self.s.schedules.list_full_day()
def test5_sunrise(self):
""" Test sunrise calculation"""
now = datetime.now()
then = now.replace(year=2017, month=02, day=14)
self.assertEqual(self.s.calc_sunrise(then), "07:40")
then = now.replace(year=2017, month=07, day=22)
self.assertEqual(self.s.calc_sunrise(then), "03:51")
def tearDown(self):
pass
