103 |
103 |
104 with self.assertRaisesRegexp(error.ProgrammingError, |
104 with self.assertRaisesRegexp(error.ProgrammingError, |
105 'unhandled frame type'): |
105 'unhandled frame type'): |
106 sendframe(reactor, ffs(b'1 0 stream-begin text-output 0 foo')) |
106 sendframe(reactor, ffs(b'1 0 stream-begin text-output 0 foo')) |
107 |
107 |
|
108 class StreamTests(unittest.TestCase): |
|
109 def testmultipleresponseframes(self): |
|
110 reactor = framing.clientreactor(buffersends=False) |
|
111 |
|
112 request, action, meta = reactor.callcommand(b'foo', {}) |
|
113 |
|
114 self.assertEqual(action, 'sendframes') |
|
115 for f in meta['framegen']: |
|
116 pass |
|
117 |
|
118 action, meta = sendframe( |
|
119 reactor, |
|
120 ffs(b'%d 0 stream-begin 4 0 foo' % request.requestid)) |
|
121 self.assertEqual(action, 'responsedata') |
|
122 |
|
123 action, meta = sendframe( |
|
124 reactor, |
|
125 ffs(b'%d 0 0 4 eos bar' % request.requestid)) |
|
126 self.assertEqual(action, 'responsedata') |
|
127 |
108 if __name__ == '__main__': |
128 if __name__ == '__main__': |
109 import silenttestrunner |
129 import silenttestrunner |
110 silenttestrunner.main(__name__) |
130 silenttestrunner.main(__name__) |