diff options
| author | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2015-07-17 09:37:57 +0200 |
|---|---|---|
| committer | Thomas Kriechbaumer <thomas@kriechbaumer.name> | 2015-07-22 15:30:51 +0200 |
| commit | 4617ab8a3a981f3abd8d62b561c80f9ad141e57b (patch) | |
| tree | 714bedbc37b482f4b24019f910309d48aaa5a2c1 /test/http/http2/test_protocol.py | |
| parent | 808b294865257fc3f52b33ed2a796009658b126f (diff) | |
| download | mitmproxy-4617ab8a3a981f3abd8d62b561c80f9ad141e57b.tar.gz mitmproxy-4617ab8a3a981f3abd8d62b561c80f9ad141e57b.tar.bz2 mitmproxy-4617ab8a3a981f3abd8d62b561c80f9ad141e57b.zip | |
add Request class and unify read_request interface
Diffstat (limited to 'test/http/http2/test_protocol.py')
| -rw-r--r-- | test/http/http2/test_protocol.py | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/test/http/http2/test_protocol.py b/test/http/http2/test_protocol.py index 403a2589..f41b9565 100644 --- a/test/http/http2/test_protocol.py +++ b/test/http/http2/test_protocol.py @@ -278,6 +278,7 @@ class TestReadEmptyResponse(tservers.ServerTestBase): resp = protocol.read_response() + assert resp.stream_id assert resp.httpversion == "HTTP/2" assert resp.status_code == "200" assert resp.msg == "" @@ -303,11 +304,11 @@ class TestReadRequest(tservers.ServerTestBase): c.convert_to_ssl() protocol = http2.HTTP2Protocol(c, is_server=True) - stream_id, headers, body = protocol.read_request() + resp = protocol.read_request() - assert stream_id - assert headers == {':method': 'GET', ':path': '/', ':scheme': 'https'} - assert body == b'foobar' + assert resp.stream_id + assert resp.headers == {':method': 'GET', ':path': '/', ':scheme': 'https'} + assert resp.content == b'foobar' class TestCreateResponse(): |