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())
