diff options
Diffstat (limited to 'test/http/http1/test_protocol.py')
| -rw-r--r-- | test/http/http1/test_protocol.py | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/test/http/http1/test_protocol.py b/test/http/http1/test_protocol.py index 8d05b31f..dcebbd5e 100644 --- a/test/http/http1/test_protocol.py +++ b/test/http/http1/test_protocol.py @@ -8,12 +8,9 @@ from ... import tutils, tservers def mock_protocol(data='', chunked=False): - class TCPHandlerMock(object): - pass - tcp_handler = TCPHandlerMock() - tcp_handler.rfile = cStringIO.StringIO(data) - tcp_handler.wfile = cStringIO.StringIO() - return HTTP1Protocol(tcp_handler) + rfile = cStringIO.StringIO(data) + wfile = cStringIO.StringIO() + return HTTP1Protocol(rfile=rfile, wfile=wfile) |