def test_noop(self): mockCommand = Mock(return_value=sentinel.tag) self.client._imap._command = mockCommand mockConsume = Mock(return_value=sentinel.out) self.client._consume_until_tagged_response = mockConsume result = self.client.noop() mockCommand.assert_called_with('NOOP') mockConsume.assert_called_with(sentinel.tag, 'NOOP') self.assertEquals(result, sentinel.out)
def test_idle_done(self): self.client._idle_tag = sentinel.tag mockSend = Mock() self.client._imap.send = mockSend mockConsume = Mock(return_value=sentinel.out) self.client._consume_until_tagged_response = mockConsume result = self.client.idle_done() mockSend.assert_called_with('DONE\r\n') mockConsume.assert_called_with(sentinel.tag, 'IDLE') self.assertEquals(result, sentinel.out)