2812f9f664
The mock third party library was needed for mock support in py2 runtimes. Since we now only support py36 and later, we can use the standard lib unittest.mock module instead. Change-Id: I491ca9f482a00b7ca3fa44aa8c26ef73559c178f
79 lines
3.0 KiB
Python
79 lines
3.0 KiB
Python
# Copyright 2014-2018 Canonical Limited.
|
|
#
|
|
# This file is part of charms.reactive.
|
|
#
|
|
# charms.reactive is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License version 3 as
|
|
# published by the Free Software Foundation.
|
|
#
|
|
# charm-helpers is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with charm-helpers. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import os
|
|
import unittest
|
|
|
|
from unittest import mock
|
|
|
|
import charms_openstack.bus as bus
|
|
|
|
|
|
class TestBus(unittest.TestCase):
|
|
|
|
@mock.patch.object(bus.os, 'walk')
|
|
@mock.patch.object(bus, '_register_handlers_from_file')
|
|
@mock.patch('charmhelpers.core.hookenv.charm_dir')
|
|
def test_discover(self, charm_dir, _register_handlers_from_file, walk):
|
|
os.walk.return_value = [(
|
|
'/x/unit-aodh-1/charm/lib/charm/openstack',
|
|
['__pycache__'],
|
|
['__init__.py', 'aodh.py'])]
|
|
|
|
charm_dir.return_value = '/x/unit-aodh-1/charm'
|
|
bus.discover()
|
|
expect_calls = [
|
|
mock.call(
|
|
'/x/unit-aodh-1/charm/lib/charm',
|
|
'/x/unit-aodh-1/charm/lib/charm/openstack/__init__.py'),
|
|
mock.call(
|
|
'/x/unit-aodh-1/charm/lib/charm',
|
|
'/x/unit-aodh-1/charm/lib/charm/openstack/aodh.py')]
|
|
_register_handlers_from_file.assert_has_calls(expect_calls)
|
|
|
|
@mock.patch.object(bus.os, 'walk')
|
|
@mock.patch.object(bus, '_register_handlers_from_file')
|
|
@mock.patch('charmhelpers.core.hookenv.charm_dir')
|
|
def test_discover_search_path(self, charm_dir,
|
|
_register_handlers_from_file, walk):
|
|
os.walk.return_value = [(
|
|
'/x/unit-aodh-1/charm/lib/charms',
|
|
['__pycache__'],
|
|
['__init__.py', 'aodh.py'])]
|
|
|
|
bus.discover(search_path='/x/unit-aodh-1/charm/lib/charms')
|
|
expect_calls = [
|
|
mock.call(
|
|
'/x/unit-aodh-1/charm/lib',
|
|
'/x/unit-aodh-1/charm/lib/charms/__init__.py'),
|
|
mock.call(
|
|
'/x/unit-aodh-1/charm/lib',
|
|
'/x/unit-aodh-1/charm/lib/charms/aodh.py')]
|
|
_register_handlers_from_file.assert_has_calls(expect_calls)
|
|
|
|
@mock.patch.object(bus.importlib, 'import_module')
|
|
def test_load_module(self, import_module):
|
|
import_module.side_effect = lambda x: x
|
|
bus._load_module(
|
|
'/x/charm/lib/charm',
|
|
'/x/charm/lib/charm/openstack/aodh.py'),
|
|
import_module.assert_called_once_with('charm.openstack.aodh')
|
|
|
|
@mock.patch.object(bus, '_load_module')
|
|
def test_register_handlers_from_file(self, _load_module):
|
|
bus._register_handlers_from_file('reactive', 'reactive/foo.py')
|
|
_load_module.assert_called_once_with('reactive', 'reactive/foo.py')
|