diff options
author | Maximilian Hils <git@maximilianhils.com> | 2016-09-22 02:03:12 -0700 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2016-09-22 02:03:12 -0700 |
commit | a3329c5a80b386a12575002d8324eb783428ba16 (patch) | |
tree | 81f8499d3cea201a07c7b75d2aae4174947e2c05 /test | |
parent | 0dbb7033ee1a6e238752381ce99439ef7d38b208 (diff) | |
download | mitmproxy-a3329c5a80b386a12575002d8324eb783428ba16.tar.gz mitmproxy-a3329c5a80b386a12575002d8324eb783428ba16.tar.bz2 mitmproxy-a3329c5a80b386a12575002d8324eb783428ba16.zip |
scripts: cut traceback properly, fix #1467
Diffstat (limited to 'test')
-rw-r--r-- | test/mitmproxy/builtins/test_script.py | 30 |
1 files changed, 27 insertions, 3 deletions
diff --git a/test/mitmproxy/builtins/test_script.py b/test/mitmproxy/builtins/test_script.py index 09e5bc92..24337760 100644 --- a/test/mitmproxy/builtins/test_script.py +++ b/test/mitmproxy/builtins/test_script.py @@ -1,10 +1,12 @@ -import time +import traceback -from mitmproxy.builtins import script +import sys +import time from mitmproxy import exceptions +from mitmproxy import options +from mitmproxy.builtins import script from mitmproxy.flow import master from mitmproxy.flow import state -from mitmproxy import options from .. import tutils, mastertest @@ -104,6 +106,10 @@ class TestScript(mastertest.MasterTest): f = tutils.tflow(resp=True) m.request(f) assert m.event_log[0][0] == "error" + assert len(m.event_log[0][1].splitlines()) == 6 + assert 'addonscripts/error.py", line 7, in request' in m.event_log[0][1] + assert 'addonscripts/error.py", line 3, in mkerr' in m.event_log[0][1] + assert m.event_log[0][1].endswith("ValueError: Error!\n") def test_duplicate_flow(self): s = state.State() @@ -136,6 +142,24 @@ class TestScript(mastertest.MasterTest): ] +class TestCutTraceback: + def raise_(self, i): + if i > 0: + self.raise_(i - 1) + raise RuntimeError() + + def test_simple(self): + try: + self.raise_(4) + except RuntimeError: + tb = sys.exc_info()[2] + tb_cut = script.cut_traceback(tb, "test_simple") + assert len(traceback.extract_tb(tb_cut)) == 5 + + tb_cut2 = script.cut_traceback(tb, "nonexistent") + assert len(traceback.extract_tb(tb_cut2)) == len(traceback.extract_tb(tb)) + + class TestScriptLoader(mastertest.MasterTest): def test_run_once(self): s = state.State() |