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_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_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_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")))