def _assertError(self, data, msg): try: rspreader.readChunked(StringIO.StringIO(data)) self.fail('Expect error') except IOError, e: if str(e).find(msg) < 0: self.fail('Expect error "%s"; received "%s"' % (msg, e))
def testTwoBlocks(self): buf = StringIO.StringIO('5\r\nabcde\r\n1\r\n\n\r\n0\r\n') # second block is a single \n fp = rspreader.readChunked(buf) self.assertEqual('abcde\n', fp.read())
def testOneBlock(self): buf = StringIO.StringIO('5\r\nabcde\r\n0\r\n') fp = rspreader.readChunked(buf) self.assertEqual('abcde', fp.read())
def testEmpty(self): buf = StringIO.StringIO('0\r\n') fp = rspreader.readChunked(buf) self.assertEqual('', fp.read())