diff options
Diffstat (limited to 'test/test_version_check.py')
| -rw-r--r-- | test/test_version_check.py | 24 |
1 files changed, 15 insertions, 9 deletions
diff --git a/test/test_version_check.py b/test/test_version_check.py index bf6ad1f5..a16969d2 100644 --- a/test/test_version_check.py +++ b/test/test_version_check.py @@ -4,19 +4,25 @@ from netlib import version_check, version @mock.patch("sys.exit") -def test_version_check(sexit): +def test_check_mitmproxy_version(sexit): fp = cStringIO.StringIO() - version_check.version_check(version.IVERSION, fp=fp) + version_check.check_mitmproxy_version(version.IVERSION, fp=fp) + assert not fp.getvalue() assert not sexit.called b = (version.IVERSION[0] - 1, version.IVERSION[1]) - version_check.version_check(b, fp=fp) + version_check.check_mitmproxy_version(b, fp=fp) + assert fp.getvalue() assert sexit.called - sexit.reset_mock() - version_check.version_check( - version.IVERSION, - pyopenssl_min_version=(9999,), - fp=fp - ) + +@mock.patch("sys.exit") +def test_check_pyopenssl_version(sexit): + fp = cStringIO.StringIO() + version_check.check_pyopenssl_version(fp=fp) + assert not fp.getvalue() + assert not sexit.called + + version_check.check_pyopenssl_version((9999,), fp=fp) + assert fp.getvalue() assert sexit.called |