From bb259e04236caa8ceeeca8a37e101ae88cffe5b4 Mon Sep 17 00:00:00 2001
From: Sergei Zyubin <sergei.zyubin@goflink.com>
Date: Tue, 3 Jan 2023 12:26:40 +0100
Subject: [PATCH] fix: replace mock module with unittest.mock (#668)

* fix: replace mock module with unittest.mock

* fix: replace mock module in tests
---
 terraform_compliance/common/error_handling.py       | 2 +-
 tests/terraform_compliance/steps/test_main_steps.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/terraform_compliance/common/error_handling.py b/terraform_compliance/common/error_handling.py
index 6068cfe2..b72112bb 100644
--- a/terraform_compliance/common/error_handling.py
+++ b/terraform_compliance/common/error_handling.py
@@ -4,7 +4,7 @@
 from terraform_compliance.main import Step
 import colorful
 from ast import literal_eval
-from mock import MagicMock
+from unittest.mock import MagicMock
 import os
 
 
diff --git a/tests/terraform_compliance/steps/test_main_steps.py b/tests/terraform_compliance/steps/test_main_steps.py
index 0d79baf0..cbc95b7b 100644
--- a/tests/terraform_compliance/steps/test_main_steps.py
+++ b/tests/terraform_compliance/steps/test_main_steps.py
@@ -16,7 +16,7 @@
 )
 from terraform_compliance.common.exceptions import TerraformComplianceNotImplemented, Failure
 from tests.mocks import MockedStep, MockedWorld
-from mock import patch
+from unittest.mock import patch
 
 
 class TestStepCases(TestCase):