Update mock import.

azext_iot/tests/central/test_iot_central_unit.py

@@ -4,11 +4,11 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
-import mock
 import pytest
 import json
 import responses
 import ast
+from unittest import mock
 from datetime import datetime
 from knack.util import CLIError
 from azure.cli.core.mock import DummyCli

azext_iot/tests/product/test_command_test_case_update_unit.py

@@ -5,7 +5,7 @@
 # --------------------------------------------------------------------------------------------
 
 import unittest
-import mock
+from unittest import mock
 from knack.util import CLIError
 from azext_iot.product.test.command_test_cases import update
 

azext_iot/tests/product/test_command_test_create_unit.py

@@ -5,7 +5,7 @@
 # --------------------------------------------------------------------------------------------
 
 import unittest
-import mock
+from unittest import mock
 from knack.util import CLIError
 from azext_iot.product.test.command_tests import create, _process_models_directory as process_models
 from azext_iot.product.shared import BadgeType, AttestationType, DeviceType, ValidationType

azext_iot/tests/product/test_command_test_run_unit.py

@@ -6,9 +6,9 @@
 
 import unittest
 import pytest
-import mock
 import json
 import responses
+from unittest import mock
 from datetime import datetime
 from knack.util import CLIError
 from azext_iot.product.test.command_test_runs import show, submit

azext_iot/tests/product/test_command_test_search_unit.py

@@ -5,7 +5,7 @@
 # --------------------------------------------------------------------------------------------
 
 import unittest
-import mock
+from unittest import mock
 from knack.util import CLIError
 from azext_iot.product.test.command_tests import search
 from azext_iot.sdk.product.models import DeviceTestSearchOptions

azext_iot/tests/product/test_command_test_update_unit.py

@@ -5,7 +5,7 @@
 # --------------------------------------------------------------------------------------------
 
 import unittest
-import mock
+from unittest import mock
 from knack.util import CLIError
 from azext_iot.product.test.command_tests import update
 from azext_iot.product.shared import BadgeType, AttestationType

azext_iot/tests/product/test_device_test_tasks_unit.py

@@ -6,9 +6,10 @@
 
 import unittest
 import pytest
-import mock
 import json
 import responses
+
+from unittest import mock
 from knack.util import CLIError
 from azext_iot.sdk.product.models import DeviceTestTask, TestRun
 from azext_iot.product.test.command_test_tasks import create, delete, show

azext_iot/tests/utility/test_iot_utility_unit.py

@@ -5,10 +5,10 @@
 # --------------------------------------------------------------------------------------------
 
 import json
-import mock
 import pytest
 import os
 
+from unittest import mock
 from knack.util import CLIError
 from azure.cli.core.extension import get_extension_path
 from azext_iot.common.utility import (

azext_iot/tests/utility/test_monitor_parsers_unit.py

@@ -5,9 +5,9 @@
 # --------------------------------------------------------------------------------------------
 
 import json
-import mock
 import pytest
 
+from unittest import mock
 from uamqp.message import Message, MessageProperties
 from azext_iot.central.providers.v1 import (
     CentralDeviceProviderV1,

azext_iot/tests/utility/test_uamqp_import.py

@@ -4,7 +4,7 @@
 # Licensed under the MIT License. See License.txt in the project root for license information.
 # --------------------------------------------------------------------------------------------
 
-import mock
+from unittest import mock
 
 
 class TestUamqpImport(object):