diff options
author | Matthew Shao <me@matshao.com> | 2017-05-19 09:45:30 +0800 |
---|---|---|
committer | Matthew Shao <me@matshao.com> | 2017-05-19 09:45:30 +0800 |
commit | 50c07034249a07f25172230b689c4c6da1b63c2d (patch) | |
tree | eb555b7b4e6dd4bfc64448d1663019fc67efee1f /test | |
parent | 4e4be8bd6854f8e5364049fa30e51e8ab0127e9b (diff) | |
download | mitmproxy-50c07034249a07f25172230b689c4c6da1b63c2d.tar.gz mitmproxy-50c07034249a07f25172230b689c4c6da1b63c2d.tar.bz2 mitmproxy-50c07034249a07f25172230b689c4c6da1b63c2d.zip |
[web] Add _tflow.js generator in test_app.py
Diffstat (limited to 'test')
-rw-r--r-- | test/mitmproxy/tools/web/test_app.py | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/test/mitmproxy/tools/web/test_app.py b/test/mitmproxy/tools/web/test_app.py index 2b6181d3..dc5fe733 100644 --- a/test/mitmproxy/tools/web/test_app.py +++ b/test/mitmproxy/tools/web/test_app.py @@ -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) |