diff options
Diffstat (limited to 'test/test_protocol_http.py')
| -rw-r--r-- | test/test_protocol_http.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/test/test_protocol_http.py b/test/test_protocol_http.py index c6a9159c..940d6c7a 100644 --- a/test/test_protocol_http.py +++ b/test/test_protocol_http.py @@ -4,6 +4,7 @@ from cStringIO import StringIO from mock import MagicMock from libmproxy.protocol.http import * +import netlib.http from netlib import odict from netlib.http import http1 from netlib.http.semantics import CONTENT_MISSING @@ -27,19 +28,19 @@ class TestHTTPResponse: "\r\n" protocol = mock_protocol(s) - r = HTTPResponse.from_protocol(protocol, "GET") + r = HTTPResponse.from_protocol(protocol, netlib.http.EmptyRequest(method="GET")) assert r.status_code == 200 assert r.content == "content" - assert HTTPResponse.from_protocol(protocol, "GET").status_code == 204 + assert HTTPResponse.from_protocol(protocol, netlib.http.EmptyRequest(method="GET")).status_code == 204 protocol = mock_protocol(s) # HEAD must not have content by spec. We should leave it on the pipe. - r = HTTPResponse.from_protocol(protocol, "HEAD") + r = HTTPResponse.from_protocol(protocol, netlib.http.EmptyRequest(method="HEAD")) assert r.status_code == 200 assert r.content == "" tutils.raises( "Invalid server response: 'content", - HTTPResponse.from_protocol, protocol, "GET" + HTTPResponse.from_protocol, protocol, netlib.http.EmptyRequest(method="GET") ) |