def test_bulid_list(): tester = event_model.Event(settings.HOST, settings.TEST_DB) chat_id = 66613871638461 tester.add_event(chat_id, 'Sara Cohen', 'Birthday', datetime.datetime(1996, 11, 29), False, ) tester.add_event(chat_id, 'Tzippy Levi', 'Birthday', datetime.datetime(1996, 2, 1), False,) tester.add_event(chat_id, 'Rachel Vagshal', 'Anniversary', datetime.datetime(1996, 2, 1), False, ) chat_id = 55555871638461 tester.add_event(chat_id, 'Sara Cohen', 'Anniversary', datetime.datetime(1997, 1, 1), False, ) tester.add_event(chat_id, 'Malka Levi', 'Anniversary', datetime.datetime(1998, 1, 1), False, ) tester.add_event(chat_id, 'Shira Vagshal', 'Birthday', datetime.datetime(1996, 1, 5), False, ) assert tester.count_events() == 6
def test_off(): tester = event_model.Event(settings.HOST, settings.TEST_DB) tester.events.drop()
def test_upcoming_events(): tester = event_model.Event(settings.HOST, settings.TEST_DB) assert len(tester.get_upcoming_events(datetime.datetime(1990, 1, 5), datetime.datetime(1990,1,6))) == 0 assert len(tester.get_upcoming_events(datetime.datetime(1990, 1, 5), datetime.datetime(1999,1,6))) == 5 assert len(tester.get_upcoming_events(datetime.datetime(1996, 3, 5), datetime.datetime(1999,1,6))) == 3
def test_all_events(): tester = event_model.Event(settings.HOST, settings.TEST_DB) assert len(tester.get_all_events()) == 5
def test_event_by_name(): tester = event_model.Event(settings.HOST, settings.TEST_DB) assert len(tester.get_events_by_name("Rut")) == 0 assert len(tester.get_events_by_name("Malka Levi")) == 1 assert len(tester.get_events_by_name("Sara Cohen")) == 2
def test_event_by_date(): tester = event_model.Event(settings.HOST, settings.TEST_DB) assert len(tester.get_events_by_date(datetime.datetime(1996, 1, 5))) == 0 assert len(tester.get_events_by_date(datetime.datetime(1997, 1, 1))) == 1 assert len(tester.get_events_by_date(datetime.datetime(1996, 2, 1))) == 2
def test_delete_event(): tester = event_model.Event(settings.HOST, settings.TEST_DB) tester.delete_event('Shira Vagshal', 'Birthday', datetime.datetime(1996, 1, 5)) assert tester.count_events() == 5