summaryrefslogtreecommitdiff
path: root/ironic/tests/unit/drivers/modules/ibmc/test_utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'ironic/tests/unit/drivers/modules/ibmc/test_utils.py')
-rw-r--r--ironic/tests/unit/drivers/modules/ibmc/test_utils.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/ironic/tests/unit/drivers/modules/ibmc/test_utils.py b/ironic/tests/unit/drivers/modules/ibmc/test_utils.py
index 87df0e6e0..f04d37e88 100644
--- a/ironic/tests/unit/drivers/modules/ibmc/test_utils.py
+++ b/ironic/tests/unit/drivers/modules/ibmc/test_utils.py
@@ -14,8 +14,8 @@
import copy
import os
+from unittest import mock
-import mock
from oslo_utils import importutils
from ironic.common import exception
@@ -144,8 +144,8 @@ class IBMCUtilsTestCase(base.IBMCTestCase):
conn = self.mock_ibmc_conn(connect_ibmc)
# Mocks
conn.system.get.side_effect = [
- ibmc_error.ConnectionError(url=self.ibmc['address'],
- error='Failed to connect to host'),
+ ibmc_error.IBMCConnectionError(url=self.ibmc['address'],
+ error='Failed to connect to host'),
mock.PropertyMock(
boot_source_override=mock.PropertyMock(
target=constants.BOOT_SOURCE_TARGET_PXE,