
* Single universal ClientManager class was dropped in favor of of individual in-context methods to create OS clients without ClientManager restrictions. * Environment class was renamed to ExecutionSession to avoid common confusion with io.murano.Environment * execution_session_local module was introduced to simplify keep of per-execution session (per-deployment) data. This is similar to thread-locals with the difference that there can be many threads in single session. * All OS-clients related code was migrated to keystone client sessions and API v3 (except for GLARE and Mistral that doesn't support sessions). This increases performance and solves authentication problems that could be caused by token expiration even with trusts enabled. * [DEFAULT]/home_region setting was introduced instead of [murano]/region_for_services to configure what region should be used by the clients by default (where Murano API resides). All client factories respect this setting. Change-Id: If02c7e5d7d39574d0621e0e8dc27d1f501a31984
134 lines
4.9 KiB
Python
134 lines
4.9 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 mock
|
|
import six
|
|
from yaql import contexts
|
|
from yaql import specs
|
|
|
|
from murano.dsl import constants
|
|
from murano.dsl import executor
|
|
from murano.dsl import murano_class
|
|
from murano.engine import execution_session
|
|
from murano.engine import mock_context_manager
|
|
from murano.engine.system import test_fixture
|
|
from murano.tests.unit import base
|
|
from murano.tests.unit.dsl.foundation import object_model as om
|
|
from murano.tests.unit.dsl.foundation import runner
|
|
from murano.tests.unit.dsl.foundation import test_case
|
|
|
|
|
|
FIXTURE_CLASS = 'io.murano.system.Agent'
|
|
FIXTURE_FUNC = 'call'
|
|
|
|
|
|
def _get_fd(set_to_extract):
|
|
return list(set_to_extract)[0]
|
|
|
|
|
|
class TestMockContextManager(mock_context_manager.MockContextManager):
|
|
def __init__(self, functions):
|
|
super(TestMockContextManager, self).__init__()
|
|
self.__functions = functions
|
|
|
|
def create_root_context(self, runtime_version):
|
|
root_context = super(TestMockContextManager, self).create_root_context(
|
|
runtime_version)
|
|
context = root_context.create_child_context()
|
|
for name, func in six.iteritems(self.__functions):
|
|
context.register_function(func, name)
|
|
return context
|
|
|
|
|
|
class MockRunner(runner.Runner):
|
|
def __init__(self, model, package_loader, functions):
|
|
if isinstance(model, six.string_types):
|
|
model = om.Object(model)
|
|
model = om.build_model(model)
|
|
if 'Objects' not in model:
|
|
model = {'Objects': model}
|
|
self.executor = executor.MuranoDslExecutor(
|
|
package_loader, TestMockContextManager(functions),
|
|
execution_session.ExecutionSession())
|
|
self._root = self.executor.load(model).object
|
|
|
|
|
|
class TestMockManager(base.MuranoTestCase):
|
|
|
|
def test_create_class_context(self):
|
|
mock_manager = mock_context_manager.MockContextManager()
|
|
mock_murano_class = mock.MagicMock(spec=murano_class.MuranoClass)
|
|
mock_murano_class.name = FIXTURE_CLASS
|
|
original_function = mock.MagicMock(spec=specs.FunctionDefinition)
|
|
original_function.is_method = True
|
|
original_function.name = FIXTURE_FUNC
|
|
original_context = contexts.Context()
|
|
p = mock.patch("inspect.getargspec", new=mock.MagicMock())
|
|
p.start()
|
|
original_context.register_function(original_function)
|
|
mock_murano_class.context = original_context
|
|
p.stop()
|
|
|
|
mock_function = mock.MagicMock(spec=specs.FunctionDefinition)
|
|
mock_function.is_method = True
|
|
mock_function.name = FIXTURE_FUNC
|
|
|
|
mock_manager.class_mock_ctx[FIXTURE_CLASS] = [mock_function]
|
|
|
|
result_context = mock_manager.create_class_context(mock_murano_class)
|
|
all_functions = result_context.collect_functions(FIXTURE_FUNC)
|
|
|
|
# Mock function should go first, but result context should contain both
|
|
self.assertIs(mock_function, _get_fd(all_functions[0]))
|
|
self.assertIs(original_function, _get_fd(all_functions[1]))
|
|
|
|
def test_create_root_context(self):
|
|
mock_manager = mock_context_manager.MockContextManager()
|
|
ctx_to_check = mock_manager.create_root_context(
|
|
constants.RUNTIME_VERSION_1_1)
|
|
inject_count = ctx_to_check.collect_functions('inject')
|
|
with_original_count = ctx_to_check.collect_functions('withOriginal')
|
|
|
|
self.assertEqual(2, len(inject_count[0]))
|
|
self.assertEqual(1, len(with_original_count[0]))
|
|
|
|
|
|
class TestMockYaqlFunctions(test_case.DslTestCase):
|
|
|
|
def setUp(self):
|
|
super(TestMockYaqlFunctions, self).setUp()
|
|
self.package_loader.load_package('io.murano', None).register_class(
|
|
test_fixture.TestFixture)
|
|
self.runner = MockRunner(om.Object('TestMocks'),
|
|
self.package_loader, self._functions)
|
|
|
|
def test_inject_method_with_str(self):
|
|
self.runner.testInjectMethodWithString()
|
|
|
|
def test_inject_object_with_str(self):
|
|
self.runner.testInjectObjectWithString()
|
|
|
|
def test_inject_method_with_yaql_expr(self):
|
|
self.runner.testInjectMethodWithYaqlExpr()
|
|
|
|
def test_inject_method_with_yaql_expr2(self):
|
|
self.runner.testInjectMethodWithYaqlExpr2()
|
|
|
|
def test_inject_object_with_yaql_expr(self):
|
|
self.runner.testInjectObjectWithYaqlExpr()
|
|
|
|
def test_with_original(self):
|
|
self.runner.testWithoriginal()
|
|
|
|
def test_original_method(self):
|
|
self.runner.testOriginalMethod()
|