def test_disconnect_failure(self): self.sock.sendall(b'QUIT\r\n') self.sock.recv(IsA(int)).AndRaise(socket_error(None, None)) self.sock.close() self.mox.ReplayAll() client = LmtpRelayClient('addr', self.queue, socket_creator=self._socket_creator, ehlo_as='there') client._connect() client._disconnect()
def test_disconnect(self): self.sock.sendall(b'QUIT\r\n') self.sock.recv(IsA(int)).AndReturn(b'221 Goodbye\r\n') self.sock.close() self.mox.ReplayAll() client = LmtpRelayClient('addr', self.queue, socket_creator=self._socket_creator, ehlo_as='there') client._connect() client._disconnect()