Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2025-01-30 23:09:44 +00:00
Code Issues Projects Releases Wiki Activity

Tests for flow_export.python_code

Browse Source
This commit is contained in:
Shadab Zafar 2016-02-08 21:56:43 +05:30
parent 61a4743b62
commit ddeb3731a8
1 changed files with 46 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
test/test_export_flow.py
View File

@ -31,3 +31,49 @@ def test_curl_command():
result = """curl -H 'header:qvalue' -X PATCH 'http://address/path?query=param' --data-binary 'content'"""
assert flow_export.curl_command(flow) == result
def test_python_code():
flow = tutils.tflow(req=req_get)
result = ("""import requests\n\n"""
"""url = 'http://address/path'\n\n"""
"""headers = {\n"""
""" 'header': 'qvalue',\n"""
"""}\n\n"""
"""response = requests.request(\n"""
""" method='GET',\n"""
""" url=url,\n"""
""" headers=headers,\n"""
""")\n\n"""
"""print(response.text)""")
assert flow_export.python_code(flow) == result
flow = tutils.tflow(req=req_post)
result = ("""import requests\n\n"""
"""url = 'http://address/path'\n\n"""
"""data = '''content'''\n\n"""
"""response = requests.request(\n"""
""" method='POST',\n"""
""" url=url,\n"""
""" data=data,\n)\n\n"""
"""print(response.text)""")
assert flow_export.python_code(flow) == result
flow = tutils.tflow(req=req_patch)
result = ("""import requests\n\n"""
"""url = 'http://address/path'\n\n"""
"""headers = {\n"""
""" 'header': 'qvalue',\n"""
"""}\n\n"""
"""params = {\n"""
""" 'query': 'param',\n"""
"""}\n\n"""
"""data = '''content'''\n\n"""
"""response = requests.request(\n"""
""" method='PATCH',\n"""
""" url=url,\n"""
""" headers=headers,\n"""
""" params=params,\n"""
""" data=data,\n"""
""")\n\n"""
"""print(response.text)""")
assert flow_export.python_code(flow) == result