Moving exporter charms to use opslib

[osm/devops.git] / installers / charm / kafka-exporter / tests / __init__.py

diff --git a/installers/charm/kafka-exporter/tests/__init__.py b/installers/charm/kafka-exporter/tests/__init__.py
index 4fd849a..90dc417 100644 (file)
--- a/installers/charm/kafka-exporter/tests/__init__.py
+++ b/installers/charm/kafka-exporter/tests/__init__.py
@@ -23,9 +23,17 @@
 """Init mocking for unit tests."""
 
 import sys
+
 import mock
 
+
+class OCIImageResourceErrorMock(Exception):
+    pass
+
+
 sys.path.append("src")
 
 oci_image = mock.MagicMock()
+oci_image.OCIImageResourceError = OCIImageResourceErrorMock
 sys.modules["oci_image"] = oci_image
+sys.modules["oci_image"].OCIImageResource().fetch.return_value = {}