def test_get_message(self): from tinymail.account import Folder folder = Folder(Mock(), 'fol1') msg1, msg2 = Mock(), Mock() folder._messages = {1: msg1, 2: msg2} self.assertEqual(folder.get_message(1), msg1) self.assertEqual(folder.get_message(2), msg2)
def test_list_messages(self): from tinymail.account import Folder folder = Folder(Mock(), 'fol1') msg1, msg2 = Mock(), Mock() folder._messages = {1: msg1, 2: msg2} messages = list(folder.list_messages()) self.assertEqual(messages, [msg1, msg2])