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

[web] Add _tflow.js generator in test_app.py

Browse Source
This commit is contained in:
Matthew Shao 2017-05-19 09:45:30 +08:00
parent 4e4be8bd68
commit 50c0703424
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/mitmproxy/tools/web/test_app.py
View File

@ -1,5 +1,6 @@
import json as _json
from unittest import mock
import os
import tornado.testing
from tornado import httpclient
@ -12,7 +13,6 @@ from mitmproxy.test import tflow
from mitmproxy.tools.web import app
from mitmproxy.tools.web import master as webmaster
def json(resp: httpclient.HTTPResponse):
return _json.loads(resp.body.decode())
@ -275,3 +275,13 @@ class TestApp(tornado.testing.AsyncHTTPTestCase):
# trigger on_close by opening a second connection.
ws_client2 = yield websocket.websocket_connect(ws_url)
ws_client2.close()
def test_generate_tflow_js(self):
tflow_json = _json.dumps(
app.flow_to_json(tflow.tflow(resp=True, err=True)), indent=4, sort_keys=True
)
web_root = os.path.join(os.getcwd(), 'web')
tflow_path = os.path.join(web_root, 'src/js/__tests__/ducks/_tflow.js')
content = """export default function(){{\n return {tflow_json}\n}}""".format(tflow_json=tflow_json)
with open(tflow_path, 'w') as f:
f.write(content)