Import mock correctly on py3

This commit sponsored by Andrew McNicol. Thank you!
2014-09-16 14:46:02 -05:00 · 2014-09-16 14:46:02 -05:00 · fa3f46d714
commit fa3f46d714
parent b6774d339a
1 changed files with 4 additions and 1 deletions
--- a/mediagoblin/tests/test_util.py
+++ b/mediagoblin/tests/test_util.py
@ -14,7 +14,10 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.

-import mock
+try:
+    import mock
+except ImportError:
+    import unittest.mock as mock
 import email
 import pytest
 import smtplib