Blob Blame History Raw
diff --git a/Makefile b/Makefile
index 77169d0..6662ca6 100644
--- a/Makefile
+++ b/Makefile
@@ -37,7 +37,7 @@ include release.mk
 
 
 bin/py.test: | bin/pip
-	bin/pip install pytest mock
+	bin/pip install pytest 'mock;python_version<"3.3"'
 	ln -sfr .venv/$@ $@
 
 bin/restview: setup.py | bin/pip
diff --git a/setup.py b/setup.py
index 2e3a16c..4b1c9d6 100755
--- a/setup.py
+++ b/setup.py
@@ -60,7 +60,7 @@ setup(
     install_requires=['docutils', 'readme_renderer', 'pygments'],
     extras_require={'syntax': []},
     test_suite='restview.tests.test_suite',
-    tests_require=['mock'],
+    tests_require=['mock;python_version<"3.3"'],
     zip_safe=False,
     entry_points="""
     [console_scripts]
diff --git a/src/restview/tests.py b/src/restview/tests.py
index 34a4c81..fe405f2 100644
--- a/src/restview/tests.py
+++ b/src/restview/tests.py
@@ -10,7 +10,11 @@ try:
 except ImportError:
     from io import StringIO
 
-from mock import Mock, patch
+try:
+    from unittest.mock import Mock, patch
+except ImportError:
+    from mock import Mock, patch
+
 import docutils.utils
 
 from restview.restviewhttp import (MyRequestHandler, RestViewer,
diff --git a/tox.ini b/tox.ini
index 32d92f3..d331517 100644
--- a/tox.ini
+++ b/tox.ini
@@ -4,7 +4,7 @@ envlist =
 
 [testenv]
 deps =
-    mock
+    mock;python_version<"3.3"
     pytest
 commands =
     py.test {posargs}