fix mitmproxy/mitmproxy#705

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