X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=installers%2Fcharm%2Fgrafana%2Ftests%2F__init__.py;h=446d5ceea610b9a98651a78d2b776297ca845057;hb=5d1ec6e86da83820d316bb52d6586f9dc27106de;hp=4fd849a5245f3f35f6fc392416062360d94a99f4;hpb=b17e76b6df29ad727711175c2b6830a9db984a1d;p=osm%2Fdevops.git

diff --git a/installers/charm/grafana/tests/__init__.py b/installers/charm/grafana/tests/__init__.py
index 4fd849a5..446d5cee 100644
--- a/installers/charm/grafana/tests/__init__.py
+++ b/installers/charm/grafana/tests/__init__.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python3
-# Copyright 2021 Canonical Ltd.
+# Copyright 2020 Canonical Ltd.
 #
 # Licensed under the Apache License, Version 2.0 (the "License"); you may
 # not use this file except in compliance with the License. You may obtain
@@ -23,9 +23,18 @@
 """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 = {}