mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 03:14:22 +00:00
126 lines
4.1 KiB
Python
126 lines
4.1 KiB
Python
import re
|
|
|
|
import netlib.tutils
|
|
from netlib.http import Headers
|
|
from mitmproxy.flow import export # heh
|
|
from . import tutils
|
|
|
|
|
|
def clean_blanks(s):
|
|
return re.sub(r"^(\s+)$", "", s, flags=re.MULTILINE)
|
|
|
|
|
|
def python_equals(testdata, text):
|
|
"""
|
|
Compare two bits of Python code, disregarding non-significant differences
|
|
like whitespace on blank lines and trailing space.
|
|
"""
|
|
d = open(tutils.test_data.path(testdata)).read()
|
|
assert clean_blanks(text).rstrip() == clean_blanks(d).rstrip()
|
|
|
|
|
|
def req_get():
|
|
return netlib.tutils.treq(method=b'GET', content=b'', path=b"/path?a=foo&a=bar&b=baz")
|
|
|
|
|
|
def req_post():
|
|
return netlib.tutils.treq(method=b'POST', headers=())
|
|
|
|
|
|
def req_patch():
|
|
return netlib.tutils.treq(method=b'PATCH', path=b"/path?query=param")
|
|
|
|
|
|
class TestExportCurlCommand():
|
|
def test_get(self):
|
|
flow = tutils.tflow(req=req_get())
|
|
result = """curl -H 'header:qvalue' -H 'content-length:7' 'http://address/path?a=foo&a=bar&b=baz'"""
|
|
assert export.curl_command(flow) == result
|
|
|
|
def test_post(self):
|
|
flow = tutils.tflow(req=req_post())
|
|
result = """curl -X POST 'http://address/path' --data-binary 'content'"""
|
|
assert export.curl_command(flow) == result
|
|
|
|
def test_patch(self):
|
|
flow = tutils.tflow(req=req_patch())
|
|
result = """curl -H 'header:qvalue' -H 'content-length:7' -X PATCH 'http://address/path?query=param' --data-binary 'content'"""
|
|
assert export.curl_command(flow) == result
|
|
|
|
|
|
class TestExportPythonCode():
|
|
def test_get(self):
|
|
flow = tutils.tflow(req=req_get())
|
|
python_equals("data/test_flow_export/python_get.py", export.python_code(flow))
|
|
|
|
def test_post(self):
|
|
flow = tutils.tflow(req=req_post())
|
|
python_equals("data/test_flow_export/python_post.py", export.python_code(flow))
|
|
|
|
def test_post_json(self):
|
|
p = req_post()
|
|
p.content = b'{"name": "example", "email": "example@example.com"}'
|
|
p.headers = Headers(content_type="application/json")
|
|
flow = tutils.tflow(req=p)
|
|
python_equals("data/test_flow_export/python_post_json.py", export.python_code(flow))
|
|
|
|
def test_patch(self):
|
|
flow = tutils.tflow(req=req_patch())
|
|
python_equals("data/test_flow_export/python_patch.py", export.python_code(flow))
|
|
|
|
|
|
class TestExportLocustCode():
|
|
def test_get(self):
|
|
flow = tutils.tflow(req=req_get())
|
|
python_equals("data/test_flow_export/locust_get.py", export.locust_code(flow))
|
|
|
|
def test_post(self):
|
|
p = req_post()
|
|
p.content = b'content'
|
|
p.headers = ''
|
|
flow = tutils.tflow(req=p)
|
|
python_equals("data/test_flow_export/locust_post.py", export.locust_code(flow))
|
|
|
|
def test_patch(self):
|
|
flow = tutils.tflow(req=req_patch())
|
|
python_equals("data/test_flow_export/locust_patch.py", export.locust_code(flow))
|
|
|
|
|
|
class TestExportLocustTask():
|
|
def test_get(self):
|
|
flow = tutils.tflow(req=req_get())
|
|
python_equals("data/test_flow_export/locust_task_get.py", export.locust_task(flow))
|
|
|
|
def test_post(self):
|
|
flow = tutils.tflow(req=req_post())
|
|
python_equals("data/test_flow_export/locust_task_post.py", export.locust_task(flow))
|
|
|
|
def test_patch(self):
|
|
flow = tutils.tflow(req=req_patch())
|
|
python_equals("data/test_flow_export/locust_task_patch.py", export.locust_task(flow))
|
|
|
|
|
|
class TestIsJson():
|
|
def test_empty(self):
|
|
assert export.is_json(None, None) is False
|
|
|
|
def test_json_type(self):
|
|
headers = Headers(content_type="application/json")
|
|
assert export.is_json(headers, b"foobar") is False
|
|
|
|
def test_valid(self):
|
|
headers = Headers(content_type="application/foobar")
|
|
j = export.is_json(headers, b'{"name": "example", "email": "example@example.com"}')
|
|
assert j is False
|
|
|
|
def test_valid2(self):
|
|
headers = Headers(content_type="application/json")
|
|
j = export.is_json(headers, b'{"name": "example", "email": "example@example.com"}')
|
|
assert isinstance(j, dict)
|
|
|
|
|
|
class TestURL():
|
|
def test_url(self):
|
|
flow = tutils.tflow()
|
|
assert export.url(flow) == "http://address:22/path"
|