class PixelatedMailboxesTest(unittest.TestCase): def setUp(self): self.querier = mock() self.search_engine = mock() self.account = MagicMock() self.mailboxes = Mailboxes(self.account, self.querier, self.search_engine) def test_move_to_inbox(self): mail = PixelatedMail.from_soledad(*test_helper.leap_mail(), soledad_querier=self.querier) when(self.querier).mail(1).thenReturn(mail) when(mail).save().thenReturn(None) mail.set_mailbox('TRASH') recovered_mail = self.mailboxes.move_to_inbox(1) self.assertEquals('INBOX', recovered_mail.mailbox_name) verify(mail).save()