mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-23 00:01:36 +00:00
85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
import socket
|
|
from io import BytesIO
|
|
from netlib.exceptions import HttpSyntaxException
|
|
|
|
from netlib.http import http1
|
|
from netlib.tcp import TCPClient
|
|
from netlib.tutils import treq, raises
|
|
import tutils
|
|
import tservers
|
|
|
|
|
|
class TestHTTPResponse:
|
|
def test_read_from_stringio(self):
|
|
s = (
|
|
b"HTTP/1.1 200 OK\r\n"
|
|
b"Content-Length: 7\r\n"
|
|
b"\r\n"
|
|
b"content\r\n"
|
|
b"HTTP/1.1 204 OK\r\n"
|
|
b"\r\n"
|
|
)
|
|
rfile = BytesIO(s)
|
|
r = http1.read_response(rfile, treq())
|
|
assert r.status_code == 200
|
|
assert r.content == b"content"
|
|
assert http1.read_response(rfile, treq()).status_code == 204
|
|
|
|
rfile = BytesIO(s)
|
|
# HEAD must not have content by spec. We should leave it on the pipe.
|
|
r = http1.read_response(rfile, treq(method=b"HEAD"))
|
|
assert r.status_code == 200
|
|
assert r.content == b""
|
|
|
|
with raises(HttpSyntaxException):
|
|
http1.read_response(rfile, treq())
|
|
|
|
|
|
class TestHTTPFlow(object):
|
|
def test_repr(self):
|
|
f = tutils.tflow(resp=True, err=True)
|
|
assert repr(f)
|
|
|
|
|
|
class TestInvalidRequests(tservers.HTTPProxTest):
|
|
ssl = True
|
|
|
|
def test_double_connect(self):
|
|
p = self.pathoc()
|
|
r = p.request("connect:'%s:%s'" % ("127.0.0.1", self.server2.port))
|
|
assert r.status_code == 400
|
|
assert "Invalid HTTP request form" in r.content
|
|
|
|
def test_relative_request(self):
|
|
p = self.pathoc_raw()
|
|
p.connect()
|
|
r = p.request("get:/p/200")
|
|
assert r.status_code == 400
|
|
assert "Invalid HTTP request form" in r.content
|
|
|
|
|
|
class TestExpectHeader(tservers.HTTPProxTest):
|
|
def test_simple(self):
|
|
client = TCPClient(("127.0.0.1", self.proxy.port))
|
|
client.connect()
|
|
|
|
# call pathod server, wait a second to complete the request
|
|
client.wfile.write(
|
|
b"POST http://localhost:%d/p/200 HTTP/1.1\r\n"
|
|
b"Expect: 100-continue\r\n"
|
|
b"Content-Length: 16\r\n"
|
|
b"\r\n" % self.server.port
|
|
)
|
|
client.wfile.flush()
|
|
|
|
assert client.rfile.readline() == "HTTP/1.1 100 Continue\r\n"
|
|
assert client.rfile.readline() == "\r\n"
|
|
|
|
client.wfile.write(b"0123456789abcdef\r\n")
|
|
client.wfile.flush()
|
|
|
|
resp = http1.read_response(client.rfile, treq())
|
|
assert resp.status_code == 200
|
|
|
|
client.finish()
|