29 lines
760 B
Python
29 lines
760 B
Python
import cStringIO
|
|
import mock
|
|
from netlib import version_check, version
|
|
|
|
|
|
@mock.patch("sys.exit")
|
|
def test_check_mitmproxy_version(sexit):
|
|
fp = cStringIO.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 = 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
|