From 8d2a2125842391a8b6e956a3277e3e3672f33a55 Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Wed, 25 Dec 2019 22:50:40 +0800
Subject: [PATCH] fix: compatibility for Python 2.7

---
 httprunner/compat.py | 2 ++
 tests/test_cli.py    | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/httprunner/compat.py b/httprunner/compat.py
index e7e92ca3..aa2a16ec 100644
--- a/httprunner/compat.py
+++ b/httprunner/compat.py
@@ -47,6 +47,7 @@ if is_py2:
     integer_types = (int, long)
 
     FileNotFoundError = IOError
+    import StringIO as io
 
 elif is_py3:
     builtin_str = str
@@ -57,3 +58,4 @@ elif is_py3:
     integer_types = (int,)
 
     FileNotFoundError = FileNotFoundError
+    import io as io
diff --git a/tests/test_cli.py b/tests/test_cli.py
index 33914844..abf2a714 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -1,8 +1,8 @@
-import io
 import sys
 import unittest
 
 from httprunner.cli import main
+from httprunner.compat import io
 
 
 class TestCli(unittest.TestCase):