diff options
author | Maximilian Hils <git@maximilianhils.com> | 2014-03-10 05:11:51 +0100 |
---|---|---|
committer | Maximilian Hils <git@maximilianhils.com> | 2014-03-10 05:11:51 +0100 |
commit | fe58c1c6eb16fdc14bd24843cb896b3d8a4eefc8 (patch) | |
tree | 0587a4a1a7351d313454a72ce237694c6523ff96 /test/tutils.py | |
parent | 78750a8b4da217a2b3f3eac23bea92b6c428fc35 (diff) | |
download | mitmproxy-fe58c1c6eb16fdc14bd24843cb896b3d8a4eefc8.tar.gz mitmproxy-fe58c1c6eb16fdc14bd24843cb896b3d8a4eefc8.tar.bz2 mitmproxy-fe58c1c6eb16fdc14bd24843cb896b3d8a4eefc8.zip |
add advanced proxying options, add SSL-terminating capability to mitmproxy
Diffstat (limited to 'test/tutils.py')
-rw-r--r-- | test/tutils.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/test/tutils.py b/test/tutils.py index 3f6592b0..c527a64a 100644 --- a/test/tutils.py +++ b/test/tutils.py @@ -1,4 +1,4 @@ -import os, shutil, tempfile +import os, shutil, tempfile, argparse from contextlib import contextmanager from libmproxy import flow, utils, controller from libmproxy.protocol import http @@ -136,6 +136,15 @@ def tmpdir(*args, **kwargs): shutil.rmtree(temp_workdir) +class MockParser(argparse.ArgumentParser): + """ + argparse.ArgumentParser sys.exits() by default. + Make it more testable by throwing an exception instead. + """ + def error(self, message): + raise Exception(message) + + def raises(exc, obj, *args, **kwargs): """ Assert that a callable raises a specified exception. |