Trying alternative mock from unittest

salt-extensions · Sep 10, 2024 · b37b207 · b37b207
1 parent 739efea
commit b37b207
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/unit/modules/test_cassandra_cql.py b/tests/unit/modules/test_cassandra_cql.py
@@ -5,13 +5,16 @@
 import logging
 import ssl
 
+## from tests.support.mock import MagicMock
+## from tests.support.mock import patch
+from unittest.mock import MagicMock
+from unittest.mock import patch
+
 import pytest
 from salt.exceptions import CommandExecutionError
 
 ## import salt.modules.cassandra_cql as cassandra_cql
 import saltext.cassandra.modules.cassandra_cql as cassandra_cql
-from tests.support.mock import MagicMock
-from tests.support.mock import patch
 
 log = logging.getLogger(__name__)