From e7600f4301dd618b2cdd2ae08ddea31eb6be0f54 Mon Sep 17 00:00:00 2001
From: Carlos Martinez <carlosmart626@gmail.com>
Date: Thu, 26 Oct 2017 12:22:12 -0500
Subject: [PATCH] Fix import Mock

---
 graphene_django/tests/test_auth.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/graphene_django/tests/test_auth.py b/graphene_django/tests/test_auth.py
index a875bf3..e2acb7c 100644
--- a/graphene_django/tests/test_auth.py
+++ b/graphene_django/tests/test_auth.py
@@ -1,7 +1,7 @@
 import datetime
+import sys
 
 import pytest
-from unittest.mock import Mock
 from django.db import models
 from django.utils.functional import SimpleLazyObject
 from py.test import raises
@@ -22,6 +22,11 @@ from ..rest_framework.mutation import SerializerMutation
 
 pytestmark = pytest.mark.django_db
 
+if sys.version_info < (3, 0):
+    from unittest.mock import Mock
+else:
+    from mock import Mock
+
 
 class MockUserContext(object):