def setUp(self):
self.model = UnreadModel()
def setUp(self):
self.model = UnreadModel()
class TestUnreadModel(unittest.TestCase):
def setUp(self):
self.model = UnreadModel()
def test_unread(self):
self.model.collection.remove()
unread = self.model.get_unread_values()
eq_(1, unread['id'])
# Check if there is a log and exception field
eq_(True, unread.has_key('logs'))
eq_(True, unread.has_key('exceptions'))
def test_mark_logs_as_read(self):
self.model.get_unread_values(
) # It will create the record if it doesn't exist
self.model.collection.update({"id": 1}, {"$inc": {"logs": 1}})
self.model.mark_logs_as_read()
result = self.model.get_unread_values()
eq_(result['logs'], 0)
def test_mark_exceptions_as_read(self):
self.model.get_unread_values(
) # It will create the record if it doesn't exist
self.model.collection.update({"id": 1}, {"$inc": {"exceptions": 1}})
self.model.mark_exceptions_as_read()
result = self.model.get_unread_values()
eq_(result['exceptions'], 0)
class TestUnreadModel(unittest.TestCase):
def setUp(self):
self.model = UnreadModel()
def test_unread(self):
self.model.collection.remove()
unread = self.model.get_unread_values()
eq_(1, unread["id"])
# Check if there is a log and exception field
eq_(True, unread.has_key("logs"))
eq_(True, unread.has_key("exceptions"))
def test_mark_logs_as_read(self):
self.model.get_unread_values() # It will create the record if it doesn't exist
self.model.collection.update({"id": 1}, {"$inc": {"logs": 1}})
self.model.mark_logs_as_read()
result = self.model.get_unread_values()
eq_(result["logs"], 0)
def test_mark_exceptions_as_read(self):
self.model.get_unread_values() # It will create the record if it doesn't exist
self.model.collection.update({"id": 1}, {"$inc": {"exceptions": 1}})
self.model.mark_exceptions_as_read()
result = self.model.get_unread_values()
eq_(result["exceptions"], 0)
