Fallback on unittest.mock

Author: singingwolfboy

tests/test_core.py

@@ -1,13 +1,17 @@
 # -*- coding: utf-8 -*-
 from __future__ import unicode_literals
-import mock
 import requests
 import requests_oauthlib
 import oauthlib
 import os.path
 from io import StringIO
 import unittest
 
+try:
+    import mock
+except ImportError:
+    from unittest import mock
+
 
 @mock.patch('oauthlib.oauth1.rfc5849.generate_timestamp')
 @mock.patch('oauthlib.oauth1.rfc5849.generate_nonce')

tests/test_oauth1_session.py

@@ -1,5 +1,4 @@
 from __future__ import unicode_literals, print_function
-import mock
 import unittest
 import sys
 import requests
@@ -9,6 +8,10 @@
 from oauthlib.oauth1 import SIGNATURE_RSA, SIGNATURE_PLAINTEXT
 from requests_oauthlib import OAuth1Session
 
+try:
+    import mock
+except ImportError:
+    from unittest import mock
 
 try:
     import cryptography

tests/test_oauth2_session.py

@@ -1,10 +1,13 @@
 from __future__ import unicode_literals
 import json
-import mock
 import time
 from base64 import b64encode
 from copy import deepcopy
 from unittest import TestCase
+try:
+    import mock
+except ImportError:
+    from unittest import mock
 
 from oauthlib.common import urlencode
 from oauthlib.oauth2 import TokenExpiredError, OAuth2Error