예제 #1
0
    def test_mark_as_read(self):
        MessageSender(self.dmd).sendToBrowser('title',
                                              'This is a browser message')
        MessageSender(self.dmd).sendToUser('title',
                                            'This is a user message')
        brow = BrowserMessageBox(self.dmd)
        user = UserMessageBox(self.dmd)

        self.assertEqual(len(brow.get_messages()), 1)
        brow.get_messages()[0].mark_as_read()
        # For browser messages, mark_as_read deletes
        self.assertEqual(len(brow.get_messages()), 0)
        self.assertEqual(len(brow.get_unread()), 0)

        self.assertEqual(len(user.get_messages()), 1)
        user.get_messages()[0].mark_as_read()
        self.assertEqual(len(user.get_messages()), 1)
        self.assertEqual(len(user.get_unread()), 0)