94 lines
3.1 KiB
Python
94 lines
3.1 KiB
Python
# 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
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
import logging
|
|
import typing
|
|
import unittest
|
|
|
|
import shade
|
|
|
|
from ospurge.resources.base import ServiceResource
|
|
from ospurge.tests import mock
|
|
from ospurge import utils
|
|
|
|
|
|
class TestUtils(unittest.TestCase):
|
|
def test_replace_project_info_in_config(self):
|
|
config = {
|
|
'cloud': 'foo',
|
|
'auth': {
|
|
'project_name': 'bar'
|
|
}
|
|
}
|
|
new_conf = utils.replace_project_info(
|
|
config, mock.sentinel.project)
|
|
|
|
self.assertEqual(new_conf, {
|
|
'auth': {
|
|
'project_id': mock.sentinel.project
|
|
}
|
|
})
|
|
self.assertEqual(config, {
|
|
'cloud': 'foo',
|
|
'auth': {
|
|
'project_name': 'bar'
|
|
}
|
|
})
|
|
|
|
def test_get_all_resource_classes(self):
|
|
classes = utils.get_resource_classes()
|
|
self.assertIsInstance(classes, typing.List)
|
|
for klass in classes:
|
|
self.assertTrue(issubclass(klass, ServiceResource))
|
|
|
|
def test_get_resource_classes(self):
|
|
config = "Networks"
|
|
classes = utils.get_resource_classes(config)
|
|
self.assertIsInstance(classes, typing.List)
|
|
for klass in classes:
|
|
self.assertTrue(issubclass(klass, ServiceResource))
|
|
|
|
def test_call_and_ignore_notfound(self):
|
|
def raiser():
|
|
raise shade.exc.OpenStackCloudResourceNotFound("")
|
|
|
|
self.assertIsNone(
|
|
utils.call_and_ignore_exc(
|
|
shade.exc.OpenStackCloudResourceNotFound, raiser
|
|
)
|
|
)
|
|
|
|
m = mock.Mock()
|
|
utils.call_and_ignore_exc(
|
|
shade.exc.OpenStackCloudResourceNotFound, m, 42)
|
|
self.assertEqual([mock.call(42)], m.call_args_list)
|
|
|
|
@mock.patch('logging.getLogger', autospec=True)
|
|
def test_monkeypatch_oscc_logging_warning(self, mock_getLogger):
|
|
oscc_target = 'os_client_config.cloud_config'
|
|
m_oscc_logger, m_other_logger = mock.Mock(), mock.Mock()
|
|
|
|
mock_getLogger.side_effect = \
|
|
lambda m: m_oscc_logger if m == oscc_target else m_other_logger
|
|
|
|
@utils.monkeypatch_oscc_logging_warning
|
|
def f():
|
|
logging.getLogger(oscc_target).warning("foo")
|
|
logging.getLogger(oscc_target).warning("!catalog entry not found!")
|
|
logging.getLogger("other").warning("!catalog entry not found!")
|
|
|
|
f()
|
|
|
|
self.assertEqual([mock.call.warning('foo'), ],
|
|
m_oscc_logger.mock_calls)
|
|
self.assertEqual([mock.call.warning('!catalog entry not found!')],
|
|
m_other_logger.mock_calls)
|