Merge "Increase unit test coverage for dsl attribute_store and context_manager."
This commit is contained in:
Jenkins
Gerrit Code Review
commit
52b7f55670
133
murano/tests/unit/dsl/test_attribute_store.py
Normal file
133
murano/tests/unit/dsl/test_attribute_store.py
Normal file
@ -0,0 +1,133 @@
|
||||
# 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
|
||||
|
||||
from murano.dsl import attribute_store
|
||||
from murano.dsl import dsl
|
||||
from murano.dsl import dsl_types
|
||||
from murano.tests.unit.dsl.foundation import test_case
|
||||
|
||||
|
||||
class TestAttributeStore(test_case.DslTestCase):
|
||||
def setUp(self):
|
||||
super(TestAttributeStore, self).setUp()
|
||||
self.attribute_store = attribute_store.AttributeStore()
|
||||
self.fake_object = mock.MagicMock(object_id=mock.sentinel.oid)
|
||||
self.tagged_obj = dsl.MuranoObjectInterface(self.fake_object)
|
||||
self.owner_type = dsl_types.MuranoTypeReference(self.fake_object)
|
||||
self.owner_type.type.name = mock.sentinel.typename
|
||||
self.name = 'foobar'
|
||||
|
||||
def test_get_attribute_key(self):
|
||||
oid, typename = mock.sentinel.oid, mock.sentinel.typename
|
||||
|
||||
key = self.attribute_store._get_attribute_key(
|
||||
self.tagged_obj, self.owner_type, self.name)
|
||||
expected_key = (oid, (typename, 'foobar'))
|
||||
|
||||
self.assertEqual(expected_key, key)
|
||||
|
||||
@mock.patch.object(attribute_store.AttributeStore, '_get_attribute_key',
|
||||
return_value=(mock.sentinel.key1, mock.sentinel.key2))
|
||||
def test_get(self, mock_get_attr_key):
|
||||
key1, key2 = mock.sentinel.key1, mock.sentinel.key2
|
||||
get_val = mock.sentinel.get_val
|
||||
|
||||
self.attribute_store._attributes = mock.MagicMock()
|
||||
self.attribute_store._attributes[key1].get.return_value = get_val
|
||||
val = self.attribute_store.get(
|
||||
self.tagged_obj, self.owner_type, self.name)
|
||||
|
||||
mock_get_attr_key.assert_called_with(
|
||||
self.tagged_obj, self.owner_type, self.name)
|
||||
self.attribute_store._attributes[key1].get.assert_called_with(key2)
|
||||
self.assertEqual(get_val, val)
|
||||
|
||||
@mock.patch.object(attribute_store.AttributeStore, '_get_attribute_key',
|
||||
return_value=(mock.sentinel.key1, mock.sentinel.key2))
|
||||
def test_set_object_if(self, mock_get_attr_key):
|
||||
val = dsl.MuranoObjectInterface(self.fake_object)
|
||||
self.attribute_store._attributes = mock.MagicMock()
|
||||
self.attribute_store.set(
|
||||
self.tagged_obj, self.owner_type, self.name, val)
|
||||
|
||||
@mock.patch.object(attribute_store.AttributeStore, '_get_attribute_key',
|
||||
return_value=(mock.sentinel.key1, mock.sentinel.key2))
|
||||
def test_set_object(self, mock_get_attr_key):
|
||||
key1, key2 = mock.sentinel.key1, mock.sentinel.key2
|
||||
|
||||
val = dsl_types.MuranoObject()
|
||||
val.object_id = mock.sentinel.oid
|
||||
self.attribute_store.set(
|
||||
self.tagged_obj, self.owner_type, self.name, val)
|
||||
self.assertEqual(self.attribute_store._attributes[key1][key2],
|
||||
mock.sentinel.oid)
|
||||
|
||||
@mock.patch.object(attribute_store.AttributeStore, '_get_attribute_key',
|
||||
return_value=(mock.sentinel.key1, mock.sentinel.key2))
|
||||
def test_set_none(self, mock_get_attr_key):
|
||||
key1, key2 = mock.sentinel.key1, mock.sentinel.key2
|
||||
|
||||
val = None
|
||||
self.attribute_store._attributes = mock.MagicMock()
|
||||
self.attribute_store.set(
|
||||
self.tagged_obj, self.owner_type, self.name, val)
|
||||
|
||||
self.attribute_store._attributes[key1].pop.assert_called_with(
|
||||
key2, None)
|
||||
|
||||
def test_serialize(self):
|
||||
known_objects = ['obj1', 'obj3']
|
||||
self.attribute_store._attributes = {
|
||||
'obj1': {
|
||||
('foo', 'obj11'): 11,
|
||||
('bar', 'obj12'): 12
|
||||
},
|
||||
'obj2': {
|
||||
('baz', 'obj21'): 21
|
||||
},
|
||||
'obj3': {
|
||||
('foobar', 'obj31'): 31
|
||||
}
|
||||
}
|
||||
val = self.attribute_store.serialize(known_objects)
|
||||
expected = [
|
||||
['obj1', 'foo', 'obj11', 11],
|
||||
['obj1', 'bar', 'obj12', 12],
|
||||
['obj3', 'foobar', 'obj31', 31]]
|
||||
|
||||
self.assertEqual(sorted(expected), sorted(val))
|
||||
|
||||
def test_load(self):
|
||||
data = [
|
||||
['a', 'b', 'c', 'd'],
|
||||
['a', 'f', 'g', None],
|
||||
['b', 'i', 'j', 'k']
|
||||
]
|
||||
|
||||
self.attribute_store.load(data)
|
||||
expected = {'a': {('b', 'c'): 'd'},
|
||||
'b': {('i', 'j'): 'k'}}
|
||||
self.assertEqual(expected, self.attribute_store._attributes)
|
||||
|
||||
def test_forget_object_if(self):
|
||||
obj = dsl.MuranoObjectInterface(mock.MagicMock(object_id='bar'))
|
||||
self.attribute_store._attributes = {'foo': 42, 'bar': 43}
|
||||
self.attribute_store.forget_object(obj)
|
||||
self.assertEqual({'foo': 42}, self.attribute_store._attributes)
|
||||
|
||||
def test_forget_object(self):
|
||||
obj = dsl_types.MuranoObject()
|
||||
obj.object_id = 'foo'
|
||||
self.attribute_store._attributes = {'foo': 42, 'bar': 43}
|
||||
self.attribute_store.forget_object(obj)
|
||||
self.assertEqual({'bar': 43}, self.attribute_store._attributes)
|
||||
44
murano/tests/unit/dsl/test_context_manager.py
Normal file
44
murano/tests/unit/dsl/test_context_manager.py
Normal file
@ -0,0 +1,44 @@
|
||||
# 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
|
||||
|
||||
from murano.dsl import context_manager
|
||||
from murano.dsl import yaql_integration
|
||||
from murano.tests.unit.dsl.foundation import test_case
|
||||
|
||||
|
||||
class TestContextManager(test_case.DslTestCase):
|
||||
def setUp(self):
|
||||
super(TestContextManager, self).setUp()
|
||||
self.context_manager = context_manager.ContextManager()
|
||||
|
||||
@mock.patch.object(yaql_integration, 'create_context', return_value='foo')
|
||||
def test_create_root_context(self, mock_create_context):
|
||||
val = self.context_manager.create_root_context('myrunver')
|
||||
|
||||
self.assertEqual('foo', val)
|
||||
mock_create_context.assert_called_with('myrunver')
|
||||
|
||||
def test_create_package_context(self):
|
||||
package = mock.MagicMock(context='mycontext')
|
||||
self.assertEqual('mycontext',
|
||||
self.context_manager.create_package_context(package))
|
||||
|
||||
def test_create_type_context(self):
|
||||
murano_type = mock.MagicMock(context='mycontext')
|
||||
self.assertEqual('mycontext',
|
||||
self.context_manager.create_type_context(murano_type))
|
||||
|
||||
def test_create_object_context(self):
|
||||
obj = 'obj'
|
||||
self.assertIs(None, self.context_manager.create_object_context(obj))
|
||||
Loading…
Reference in New Issue
Block a user