39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
from io import StringIO
|
|
import mock
|
|
from netlib import version_check, version
|
|
|
|
|
|
@mock.patch("sys.exit")
|
|
def test_check_mitmproxy_version(sexit):
|
|
fp = StringIO()
|
|
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.check_mitmproxy_version(b, fp=fp)
|
|
assert fp.getvalue()
|
|
assert sexit.called
|
|
|
|
|
|
@mock.patch("sys.exit")
|
|
def test_check_pyopenssl_version(sexit):
|
|
fp = 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 "outdated" in fp.getvalue()
|
|
assert sexit.called
|
|
|
|
|
|
@mock.patch("sys.exit")
|
|
@mock.patch("OpenSSL.__version__")
|
|
def test_unparseable_pyopenssl_version(version, sexit):
|
|
version.split.return_value = ["foo", "bar"]
|
|
fp = StringIO()
|
|
version_check.check_pyopenssl_version(fp=fp)
|
|
assert "Cannot parse" in fp.getvalue()
|
|
assert not sexit.called
|