def test_return_all(self):
     io = IO(None)
     dr = DataReader(io)
     dr.lines = [b'one\r\n', b'two\r\n', b'.\r\n', b'three\r\n']
     dr.EOD = 2
     self.assertEqual(b'one\r\ntwo\r\n', dr.return_all())
     self.assertEqual(b'three\r\n', io.recv_buffer)

    def _get_message_data(self):
        max_size = self.extensions.getparam('SIZE', filter=int)
        reader = DataReader(self.io, max_size)

        err = None
        timeout = Timeout(self.data_timeout)
        timeout.start()
        try:
            data = reader.recv()
        except ConnectionLost:
            raise
        except SmtpError as e:
            data = None
            err = e
        finally:
            timeout.cancel()

        reply = Reply('250', '2.6.0 Message Accepted for Delivery')
        self._call_custom_handler('HAVE_DATA', reply, data, err)

        self.io.send_reply(reply)
        self.io.flush_send()

        self.have_mailfrom = None
        self.have_rcptto = None

 def test_return_all(self):
     io = IO(None)
     dr = DataReader(io)
     dr.lines = ["one\r\n", "two\r\n", ".\r\n", "three\r\n"]
     dr.EOD = 2
     self.assertEqual("one\r\ntwo\r\n", dr.return_all())
     self.assertEqual("three\r\n", io.recv_buffer)

 def test_recv(self):
     self.sock.recv(IsA(int)).AndReturn("\r\nthree\r\n")
     self.sock.recv(IsA(int)).AndReturn(".\r\nstuff\r\n")
     self.mox.ReplayAll()
     io = IO(self.sock)
     io.recv_buffer = "one\r\ntwo"
     dr = DataReader(io)
     self.assertEqual("one\r\ntwo\r\nthree\r\n", dr.recv())

 def test_recv(self):
     self.sock.recv(IsA(int)).AndReturn(b'\r\nthree\r\n')
     self.sock.recv(IsA(int)).AndReturn(b'.\r\nstuff\r\n')
     self.mox.ReplayAll()
     io = IO(self.sock)
     io.recv_buffer = b'one\r\ntwo'
     dr = DataReader(io)
     self.assertEqual(b'one\r\ntwo\r\nthree\r\n', dr.recv())

 def test_recv_piece(self):
     self.sock.recv(IsA(int)).AndReturn("one\r\ntwo")
     self.sock.recv(IsA(int)).AndReturn("\r\nthree\r\n.\r\nstuff\r\n")
     self.mox.ReplayAll()
     dr = DataReader(IO(self.sock))
     self.assertTrue(dr.recv_piece())
     self.assertFalse(dr.recv_piece())
     self.assertEqual(["one\r\n", "two\r\n", "three\r\n", ".\r\n", "stuff\r\n", ""], dr.lines)
     self.assertEqual(3, dr.EOD)
     self.assertEqual(5, dr.i)

 def test_recv_piece(self):
     self.sock.recv(IsA(int)).AndReturn(b'one\r\ntwo')
     self.sock.recv(IsA(int)).AndReturn(b'\r\nthree\r\n.\r\nstuff\r\n')
     self.mox.ReplayAll()
     dr = DataReader(IO(self.sock))
     self.assertTrue(dr.recv_piece())
     self.assertFalse(dr.recv_piece())
     self.assertEqual([b'one\r\n', b'two\r\n', b'three\r\n',
                       b'.\r\n', b'stuff\r\n', b''], dr.lines)
     self.assertEqual(3, dr.EOD)
     self.assertEqual(5, dr.i)

    def _get_message_data(self):
        max_size = self.extensions.getparam('SIZE', filter=int)
        reader = DataReader(self.io, max_size)

        err = None
        timeout = Timeout(self.data_timeout)
        timeout.start()
        try:
            data = reader.recv()
        except ConnectionLost:
            raise
        except SmtpError, e:
            data = None
            err = e

 def test_append_line(self):
     dr = DataReader(None)
     dr._append_line("asdf")
     dr._append_line("jkl\r\n")
     dr.i += 1
     dr._append_line("qwerty")
     self.assertEqual(["asdfjkl\r\n", "qwerty"], dr.lines)

 def test_append_line(self):
     dr = DataReader(None)
     dr._append_line(b'asdf')
     dr._append_line(b'jkl\r\n')
     dr.i += 1
     dr._append_line(b'qwerty')
     self.assertEqual([b'asdfjkl\r\n', b'qwerty'], dr.lines)

 def test_handle_finished_line_EOD(self):
     dr = DataReader(None)
     dr.lines = [".\r\n"]
     dr.handle_finished_line()
     assert_equal(0, dr.EOD)

 def test_recv_piece_already_eod(self):
     dr = DataReader(None)
     dr.EOD = 2
     self.assertFalse(dr.recv_piece())

 def test_from_recv_buffer(self):
     io = IO(None)
     io.recv_buffer = "test\r\ndata"
     dr = DataReader(io)
     dr.from_recv_buffer()
     self.assertEqual(["test\r\n", "data"], dr.lines)

 def test_add_lines(self):
     dr = DataReader(None)
     dr.add_lines(b'\r\ntwo\r\n.three\r\nfour')
     self.assertEqual([b'\r\n', b'two\r\n', b'three\r\n', b'four'], dr.lines)
     self.assertEqual(3, dr.i)
     self.assertEqual(None, dr.EOD)

 def test_handle_finished_line_initial_period(self):
     dr = DataReader(None)
     dr.lines = [b'..stuff\r\n']
     dr.handle_finished_line()
     self.assertEqual(b'.stuff\r\n', dr.lines[0])

 def test_handle_finished_line_EOD(self):
     dr = DataReader(None)
     dr.lines = [b'.\r\n']
     dr.handle_finished_line()
     self.assertEqual(0, dr.EOD)

 def test_from_recv_buffer(self):
     io = IO(None)
     io.recv_buffer = b'test\r\ndata'
     dr = DataReader(io)
     dr.from_recv_buffer()
     self.assertEqual([b'test\r\n', b'data'], dr.lines)

 def test_handle_finished_line_initial_period(self):
     dr = DataReader(None)
     dr.lines = ["..stuff\r\n"]
     dr.handle_finished_line()
     assert_equal(".stuff\r\n", dr.lines[0])

 def test_add_lines(self):
     dr = DataReader(None)
     dr.add_lines("\r\ntwo\r\n.three\r\nfour")
     self.assertEqual(["\r\n", "two\r\n", "three\r\n", "four"], dr.lines)
     self.assertEqual(3, dr.i)
     self.assertEqual(None, dr.EOD)