[test] Prepare tests for python3 compatibility

1 files changed, 7 insertions, 2 deletions

diff --git a/testing/test_soledad/u1db_tests/__init__.py b/testing/test_soledad/u1db_tests/__init__.py
index ba776864..36db548d 100644
--- a/testing/test_soledad/u1db_tests/__init__.py
+++ b/testing/test_soledad/u1db_tests/__init__.py
@@ -23,11 +23,16 @@ import socket
 import tempfile
 import threading
 import json
+import sys
 
+from six import StringIO
 from wsgiref import simple_server
 
-from pysqlcipher import dbapi2
-from StringIO import StringIO
+
+if sys.version_info[0] < 3:
+    from pysqlcipher import dbapi2
+else:
+    from pysqlcipher3 import dbapi2
 
 import testscenarios
 from twisted.trial import unittest