Move version check to netlib, unit test it.

1 files changed, 22 insertions, 0 deletions

diff --git a/test/test_version_check.py b/test/test_version_check.py
new file mode 100644
index 00000000..bf6ad1f5
--- /dev/null
+++ b/test/test_version_check.py
@@ -0,0 +1,22 @@
+import cStringIO
+import mock
+from netlib import version_check, version
+
+
+@mock.patch("sys.exit")
+def test_version_check(sexit):
+    fp = cStringIO.StringIO()
+    version_check.version_check(version.IVERSION, fp=fp)
+    assert not sexit.called
+
+    b = (version.IVERSION[0] - 1, version.IVERSION[1])
+    version_check.version_check(b, fp=fp)
+    assert sexit.called
+
+    sexit.reset_mock()
+    version_check.version_check(
+        version.IVERSION,
+        pyopenssl_min_version=(9999,),
+        fp=fp
+    )
+    assert sexit.called