Use unittest.mock if it is available

tests/asyncio/future/test_async_future.py

@@ -14,7 +14,10 @@
 
 import asyncio
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import exceptions

tests/asyncio/gapic/test_method_async.py

@@ -15,7 +15,10 @@
 import datetime
 
 from grpc.experimental import aio
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import (exceptions, gapic_v1, grpc_helpers_async,

tests/asyncio/operations_v1/test_operations_async_client.py

@@ -13,7 +13,10 @@
 # limitations under the License.
 
 from grpc.experimental import aio
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import (grpc_helpers_async, operations_v1,

tests/asyncio/test_grpc_helpers_async.py

@@ -14,7 +14,10 @@
 
 import grpc
 from grpc.experimental import aio
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import exceptions

tests/asyncio/test_operation_async.py

@@ -13,7 +13,10 @@
 # limitations under the License.
 
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import exceptions

tests/asyncio/test_page_iterator_async.py

@@ -14,7 +14,10 @@
 
 import inspect
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import page_iterator_async

tests/asyncio/test_retry_async.py

@@ -15,7 +15,10 @@
 import datetime
 import re
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import exceptions

tests/unit/future/test__helpers.py

@@ -12,7 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 
 from google.api_core.future import _helpers
 

tests/unit/future/test_polling.py

@@ -16,7 +16,10 @@
 import threading
 import time
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import exceptions, retry

tests/unit/gapic/test_method.py

@@ -14,7 +14,10 @@
 
 import datetime
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 
 from google.api_core import exceptions
 from google.api_core import retry

tests/unit/test_bidi.py

@@ -17,7 +17,10 @@
 import threading
 
 import grpc
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 from six.moves import queue
 

tests/unit/test_exceptions.py

@@ -15,7 +15,10 @@
 import json
 
 import grpc
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import requests
 from six.moves import http_client
 

tests/unit/test_grpc_helpers.py

@@ -13,7 +13,10 @@
 # limitations under the License.
 
 import grpc
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import exceptions

tests/unit/test_operation.py

@@ -13,7 +13,10 @@
 # limitations under the License.
 
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 
 from google.api_core import exceptions
 from google.api_core import operation

tests/unit/test_page_iterator.py

@@ -15,7 +15,10 @@
 import math
 import types
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 import six
 

tests/unit/test_path_template.py

@@ -14,7 +14,10 @@
 
 from __future__ import unicode_literals
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 
 from google.api_core import path_template

tests/unit/test_retry.py

@@ -16,7 +16,10 @@
 import itertools
 import re
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 import pytest
 import requests.exceptions
 

tests/unit/test_timeout.py

@@ -15,7 +15,10 @@
 import datetime
 import itertools
 
-import mock
+try:
+    from unittest import mock
+except:
+    import mock
 
 from google.api_core import timeout