wip

1 files changed, 4 insertions, 4 deletions

diff --git a/test/test_version_check.py b/test/test_version_check.py
index 9a127814..ec2396fe 100644
--- a/test/test_version_check.py
+++ b/test/test_version_check.py
@@ -1,11 +1,11 @@
-import cStringIO
+from io import StringIO
 import mock
 from netlib import version_check, version
 
 
 @mock.patch("sys.exit")
 def test_check_mitmproxy_version(sexit):
-    fp = cStringIO.StringIO()
+    fp = StringIO()
     version_check.check_mitmproxy_version(version.IVERSION, fp=fp)
     assert not fp.getvalue()
     assert not sexit.called
@@ -18,7 +18,7 @@ def test_check_mitmproxy_version(sexit):
 
 @mock.patch("sys.exit")
 def test_check_pyopenssl_version(sexit):
-    fp = cStringIO.StringIO()
+    fp = StringIO()
     version_check.check_pyopenssl_version(fp=fp)
     assert not fp.getvalue()
     assert not sexit.called
@@ -32,7 +32,7 @@ def test_check_pyopenssl_version(sexit):
 @mock.patch("OpenSSL.__version__")
 def test_unparseable_pyopenssl_version(version, sexit):
     version.split.return_value = ["foo", "bar"]
-    fp = cStringIO.StringIO()
+    fp = StringIO()
     version_check.check_pyopenssl_version(fp=fp)
     assert "Cannot parse" in fp.getvalue()
     assert not sexit.called