class LightsStoreTest(TestCase):
def setUp(self):
self.store = LightsStoreStub()
self.lights = Lights(store=self.store)
def test_should_add_trigger_to_store(self):
self.store.add_trigger = MagicMock()
self.lights.add_trigger(True, 10, 20)
self.store.add_trigger.assert_called_with(ANY)
def test_should_read_triggers_from_store(self):
self.store.read_triggers = MagicMock()
self.lights = Lights(store=self.store)
self.store.read_triggers.assert_called_with()
def test_should_remove_trigger_from_store(self):
self.store.remove_trigger = MagicMock()
self.lights.remove_trigger(job_id=TRIGGER.job_id)
self.store.remove_trigger.assert_called_with(TRIGGER.job_id)
def test_should_add_history_entry_to_store(self):
self.store.add_entry = MagicMock()
self.lights.set_state(True, "test")
self.store.add_entry.assert_called_with(True, "test")
def test_should_read_history_from_store(self):
self.store.read_history = MagicMock()
self.lights.get_history()
self.store.read_history.assert_called_with()
class LightsSchedulerTest(TestCase):
def setUp(self):
self.lights = Lights()
def test_starts_scheduler(self):
mock = Mock()
Lights(scheduler=mock)
mock.start.assert_called_with()
def test_add_trigger_to_scheduler(self):
scheduler = Mock(wraps=BackgroundScheduler())
lights = Lights(scheduler=scheduler)
state = True
hour = 10
minute = 20
lights.add_trigger(state, hour, minute)
scheduler.add_job.assert_called_with(func=ANY, args=[state, "trigger"], end_date=ANY, trigger="cron",
id=None, hour=hour, minute=minute, day_of_week=None)
def test_add_trigger_to_scheduler_weekday(self):
scheduler = Mock(wraps=BackgroundScheduler())
lights = Lights(scheduler=scheduler)
state = True
hour = 10
minute = 20
repeat_weekday = True
lights.add_trigger(state, hour, minute, repeat_weekday=repeat_weekday)
scheduler.add_job.assert_called_with(func=ANY, args=[state, "trigger"], end_date=None, trigger="cron",
id=None, hour=hour, minute=minute, day_of_week=Lights.weekdays)
def test_add_trigger_to_scheduler_weekend(self):
scheduler = Mock(wraps=BackgroundScheduler())
lights = Lights(scheduler=scheduler)
state = True
hour = 10
minute = 20
repeat_weekend = True
lights.add_trigger(state, hour, minute, repeat_weekend=repeat_weekend)
scheduler.add_job.assert_called_with(func=ANY, args=[state, "trigger"], end_date=None, trigger="cron",
id=None, hour=hour, minute=minute, day_of_week=Lights.weekends)
def test_add_trigger_to_scheduler_all_week(self):
scheduler = Mock(wraps=BackgroundScheduler())
lights = Lights(scheduler=scheduler)
state = True
hour = 10
minute = 20
repeat_weekday = True
repeat_weekend = True
lights.add_trigger(state, hour, minute, repeat_weekday, repeat_weekend)
scheduler.add_job.assert_called_with(func=ANY, args=[state, "trigger"], end_date=None, trigger="cron",
id=None, hour=hour, minute=minute,
day_of_week=Lights.weekdays + "," + Lights.weekends)
def test_add_trigger_and_get_job(self):
state = True
hour = 10
minute = 20
self.lights.add_trigger(state, hour, minute)
jobs = self.lights._scheduler.get_jobs()
assert_that(jobs, has_length(1))
def test_add_triggers_and_get_jobs(self):
num = 5
for i in range(num):
self.lights.add_trigger(True, 10, 20)
jobs = self.lights._scheduler.get_jobs()
assert_that(jobs, has_length(num))
triggers = self.lights.get_triggers()
assert_that(triggers, has_length(num))
def test_add_and_get_trigger(self):
state = True
hour = 10
minute = 20
now = int(time())
self.lights.add_trigger(state, hour, minute)
job_id = self.lights._scheduler.get_jobs()[0].id
triggers = self.lights.get_triggers()
assert_that(triggers, has_length(1))
trigger = triggers[0]
assert_that(trigger.job_id, is_(job_id))
assert_that(trigger.state, is_(state))
assert_that(trigger.hour, is_(hour))
assert_that(trigger.minute, is_(minute))
assert_that(trigger.next_run_time, is_(greater_than(now)))
assert_that(trigger.repeat_weekday, is_(False))
assert_that(trigger.repeat_weekend, is_(False))
def test_add_and_get_trigger_with_update_run_time(self):
next_run_time = datetime(year=2010, month=1, day=1, tzinfo=utc)
expected_next_run_time = int(next_run_time.timestamp())
self.lights.add_trigger(True, 10, 20)
job_id = self.lights._scheduler.get_jobs()[0].id
job = Mock()
job.id = job_id
job.next_run_time = next_run_time
self.lights._scheduler.get_jobs = MagicMock(return_value=[job])
triggers = self.lights.get_triggers()
assert_that(triggers, has_length(1))
trigger = triggers[0]
assert_that(trigger.next_run_time, is_(expected_next_run_time))
def test_add_no_repeat_has_end_date(self):
self.lights.add_trigger(True, 10, 20)
job = self.lights._scheduler.get_jobs()[0]
assert_that(job.trigger.end_date, is_(not_none()))
def test_remove_finished_jobs_from_trigger(self):
self.lights.add_trigger(True, 10, 20)
self.lights.add_trigger(True, 10, 20)
jobs = self.lights._scheduler.get_jobs()
removed_job_id = jobs[0].id
keep_job_id = jobs[1].id
self.lights._scheduler.remove_job(removed_job_id)
jobs = self.lights._scheduler.get_jobs()
assert_that(jobs, has_length(1))
triggers = self.lights.get_triggers()
assert_that(triggers, has_length(1))
trigger = triggers[0]
assert_that(trigger.job_id, is_(keep_job_id))
def test_delete_trigger_from_triggers(self):
self.lights.add_trigger(True, 10, 20)
jobs = self.lights._scheduler.get_jobs()
job_id = jobs[0].id
triggers = self.lights.get_triggers()
assert_that(triggers, has_length(1))
self.lights.remove_trigger(job_id)
triggers = self.lights.get_triggers()
assert_that(triggers, is_(empty()))
def test_delete_job_from_scheduler(self):
self.lights.add_trigger(True, 10, 20)
jobs = self.lights._scheduler.get_jobs()
job_id = jobs[0].id
assert_that(jobs, has_length(1))
self.lights.remove_trigger(job_id)
jobs = self.lights._scheduler.get_jobs()
assert_that(jobs, is_(empty()))
