installers/charm/mysqld-exporter/tests/__init__.py

   1 #!/usr/bin/env python3
   2 # Copyright 2021 Canonical Ltd.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License"); you may
   5 # not use this file except in compliance with the License. You may obtain
   6 # a copy of the License at
   7 #
   8 #         http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  12 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  13 # License for the specific language governing permissions and limitations
  14 # under the License.
  15 #
  16 # For those usages not covered by the Apache License, Version 2.0 please
  17 # contact: legal@canonical.com
  18 #
  19 # To get in touch with the maintainers, please contact:
  20 # osm-charmers@lists.launchpad.net
  21 ##
  22
  23 """Init mocking for unit tests."""
  24
  25 import sys
  26
  27 import mock
  28
  29
  30 class OCIImageResourceErrorMock(Exception):
  31     pass
  32
  33
  34 sys.path.append("src")
  35
  36 oci_image = mock.MagicMock()
  37 oci_image.OCIImageResourceError = OCIImageResourceErrorMock
  38 sys.modules["oci_image"] = oci_image
  39 sys.modules["oci_image"].OCIImageResource().fetch.return_value = {}