def test_EventDAO_reset_event(self, MockDB): MockDB.user_data.event.update_one.acknowledged.return_value = True dao = EventDAO(MockDB) self.assertTrue(dao.reset_event({"target": "test target", \ "name": "test event", "last_notified_time": 0, "alarm_time": Event.DEFAULT_ALARM_TIME}))
def test_EventDAO_update_last_notified_time(self, MockDB): MockDB.user_data.event.update_one.acknowledged.return_value = True dao = EventDAO(MockDB) self.assertTrue( dao.update_last_notified_time("test target", "test name", "test last_notified_time"))
def test_EventDAO_add_user_and_event(self, MockDB): MockDB.user_data.event.insert_one.acknowledged.return_value = True dao = EventDAO(MockDB) self.assertTrue( dao.add_user_and_event("test target", Event("test target", "test event")))
def __init__(self, dao=None, message_api=None): if dao == None: dao = EventDAO() if message_api == None: message_api = MessageApi() self.dao = dao self.message_api = message_api
def test_EventDAO_remove_event(self, MockDB): MockDB.user_data.event.delete_one.acknowledged.return_value = True dao = EventDAO(MockDB) self.assertTrue(dao.remove_event(Event("test target", "test event")))
def test_EventDAO_has_event(self, MockDB): MockDB.user_data.event.find.return_value = [{"name": "test event"}] dao = EventDAO(MockDB) self.assertTrue(dao.has_event(Event("test target", "test event")))