class PixelatedMailboxTest(unittest.TestCase):
def setUp(self):
self.tag_service = mock()
self.querier = mock()
self.mailbox = PixelatedMailbox('INBOX', self.querier, tag_service=self.tag_service)
def test_remove_message_from_mailbox(self):
mail = PixelatedMail.from_soledad(*test_helper.leap_mail(), soledad_querier=self.querier)
when(self.querier).mail(1).thenReturn(mail)
self.mailbox.remove(1)
verify(self.querier).remove_mail(mail)
