#!/usr/bin/env python3 """Test for the signal event notifier""" import datetime import os import unittest import signal_event_notifier as sen DUMMY_CYCLIC_RESPONSIBILITIES = "alice:+1234\nbob:+5678,charlie:+5679" TEST_CAL = "test/cal.ics" # pragma pylint: disable=missing-function-docstring,missing-class-docstring,invalid-name class TestCyclicRespLoading(unittest.TestCase): def setUp(self): os.environ[ sen.RESPONSIBILITIES_ENV_VAR_NAME] = DUMMY_CYCLIC_RESPONSIBILITIES def test_cyclic_resp_first_week(self): first = datetime.date(year=2013, month=1, day=1) resp = sen.load_responsibilities(first) self.assertEqual(resp, ['alice:+1234']) def test_cyclic_resp_second_week(self): second = datetime.date(year=2013, month=1, day=8) resp = sen.load_responsibilities(second) self.assertEqual(resp, ['bob:+5678', 'charlie:+5679']) def test_cyclic_resp_all_weeks(self): expected = [['alice:+1234'], ['bob:+5678', 'charlie:+5679']] day = datetime.date(year=2013, month=1, day=1) delta = datetime.timedelta(days=7) for week in range(52): resp = sen.load_responsibilities(day) self.assertEqual(resp, expected[(week % 2)]) day += delta if __name__ == '__main__': unittest.main()