comparison tests/test-wireproto-clientreactor.py @ 37656:e6870bca1f47

wireprotoframing: record when new stream is encountered

Without this, we choke after receiving the 2nd frame in a stream.
Not sure how we made it this far without finding this bug.

Differential Revision: https://phab.mercurial-scm.org/D3334

37655:b9502b5f2066

        with self.assertRaisesRegexp(error.ProgrammingError,
                                     'unhandled frame type'):
            sendframe(reactor, ffs(b'1 0 stream-begin text-output 0 foo'))

if __name__ == '__main__':
    import silenttestrunner
    silenttestrunner.main(__name__)

37656:e6870bca1f47

        with self.assertRaisesRegexp(error.ProgrammingError,
                                     'unhandled frame type'):
            sendframe(reactor, ffs(b'1 0 stream-begin text-output 0 foo'))

class StreamTests(unittest.TestCase):
    def testmultipleresponseframes(self):
        reactor = framing.clientreactor(buffersends=False)

        request, action, meta = reactor.callcommand(b'foo', {})

        self.assertEqual(action, 'sendframes')
        for f in meta['framegen']:
            pass

        action, meta = sendframe(
            reactor,
            ffs(b'%d 0 stream-begin 4 0 foo' % request.requestid))
        self.assertEqual(action, 'responsedata')

        action, meta = sendframe(
            reactor,
            ffs(b'%d 0 0 4 eos bar' % request.requestid))
        self.assertEqual(action, 'responsedata')

if __name__ == '__main__':
    import silenttestrunner
    silenttestrunner.main(__name__)