mitmproxy/test/netlib/test_version_check.py

26 lines
681 B
Python
Raw Normal View History

2016-10-17 03:38:31 +00:00
import io
import mock
2016-05-28 12:36:43 +00:00
from netlib import version_check
2015-08-03 16:06:31 +00:00
@mock.patch("sys.exit")
def test_check_pyopenssl_version(sexit):
2016-10-17 03:38:31 +00:00
fp = io.StringIO()
2015-08-03 16:06:31 +00:00
version_check.check_pyopenssl_version(fp=fp)
assert not fp.getvalue()
assert not sexit.called
version_check.check_pyopenssl_version((9999,), fp=fp)
2015-09-10 10:32:38 +00:00
assert "outdated" in fp.getvalue()
assert sexit.called
2015-09-10 10:32:38 +00:00
@mock.patch("sys.exit")
@mock.patch("OpenSSL.__version__")
def test_unparseable_pyopenssl_version(version, sexit):
version.split.return_value = ["foo", "bar"]
2016-10-17 03:38:31 +00:00
fp = io.StringIO()
2015-09-10 10:32:38 +00:00
version_check.check_pyopenssl_version(fp=fp)
assert "Cannot parse" in fp.getvalue()
assert not sexit.called