You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
131 lines
4.2 KiB
131 lines
4.2 KiB
7 months ago
|
import unittest
|
||
|
|
from unittest import TestCase
|
||
|
|
from metadata_model import *
|
||
|
|
|
||
|
|
class TestInnerNamespace(TestCase):
|
||
|
|
def test_combine_children_by_name(self):
|
||
|
|
#
|
||
|
|
# Set up
|
||
|
|
#
|
||
|
|
kind = Kind("some_root_kind", parent=None)
|
||
|
|
ins_outer = InnerNamespace("static", parent=kind)
|
||
|
|
kind._namespaces = [ins_outer]
|
||
|
|
|
||
|
|
ins1 = InnerNamespace("ins1", parent=ins_outer)
|
||
|
|
ins1a = InnerNamespace("ins1", parent=ins_outer) # same name deliberately
|
||
|
|
entry1 = Entry(name="entry1", type="int32", kind="static",
|
||
|
|
parent=ins1, hal_version="3.2")
|
||
|
|
entry2 = Entry(name="entry2", type="int32", kind="static",
|
||
|
|
parent=ins1a, hal_version="3.2")
|
||
|
|
entry3 = Entry(name="entry3", type="int32", kind="static",
|
||
|
|
parent=ins_outer, hal_version="3.2")
|
||
|
|
|
||
|
|
ins_outer._namespaces = [ins1, ins1a]
|
||
|
|
ins_outer._entries = [entry3]
|
||
|
|
|
||
|
|
ins1._entries = [entry1]
|
||
|
|
ins1a._entries = [entry2]
|
||
|
|
|
||
|
|
#
|
||
|
|
# Test
|
||
|
|
#
|
||
|
|
combined_children_namespace = ins_outer.combine_children_by_name()
|
||
|
|
|
||
|
|
self.assertIsInstance(combined_children_namespace, InnerNamespace)
|
||
|
|
combined_ins = [i for i in combined_children_namespace.namespaces]
|
||
|
|
combined_ent = [i for i in combined_children_namespace.entries]
|
||
|
|
|
||
|
|
self.assertEquals(kind, combined_children_namespace.parent)
|
||
|
|
self.assertEquals(1, len(combined_ins))
|
||
|
|
self.assertEquals(1, len(combined_ent))
|
||
|
|
|
||
|
|
self.assertEquals("ins1", combined_ins[0].name)
|
||
|
|
self.assertEquals("entry3", combined_ent[0].name)
|
||
|
|
|
||
|
|
new_ins = combined_ins[0]
|
||
|
|
self.assertIn(entry1, new_ins.entries)
|
||
|
|
self.assertIn(entry2, new_ins.entries)
|
||
|
|
|
||
|
|
|
||
|
|
class TestKind(TestCase):
|
||
|
|
def test_combine_kinds_into_single_node(self):
|
||
|
|
#
|
||
|
|
# Set up
|
||
|
|
#
|
||
|
|
section = Section("some_section", parent=None)
|
||
|
|
kind_static = Kind("static", parent=section)
|
||
|
|
kind_dynamic = Kind("dynamic", parent=section)
|
||
|
|
section._kinds = [kind_static, kind_dynamic]
|
||
|
|
|
||
|
|
ins1 = InnerNamespace("ins1", parent=kind_static)
|
||
|
|
ins2 = InnerNamespace("ins2", parent=kind_dynamic)
|
||
|
|
entry1 = Entry(name="entry1", type="int32", kind="static",
|
||
|
|
parent=kind_static, hal_version="3.2")
|
||
|
|
entry2 = Entry(name="entry2", type="int32", kind="static",
|
||
|
|
parent=kind_dynamic, hal_version="3.2")
|
||
|
|
|
||
|
|
kind_static._namespaces = [ins1]
|
||
|
|
kind_static._entries = [entry1]
|
||
|
|
|
||
|
|
kind_dynamic._namespaces = [ins2]
|
||
|
|
kind_dynamic._entries = [entry2]
|
||
|
|
|
||
|
|
#
|
||
|
|
# Test
|
||
|
|
#
|
||
|
|
combined_kind = section.combine_kinds_into_single_node()
|
||
|
|
|
||
|
|
self.assertEquals(section, combined_kind.parent)
|
||
|
|
|
||
|
|
self.assertIn(ins1, combined_kind.namespaces)
|
||
|
|
self.assertIn(ins2, combined_kind.namespaces)
|
||
|
|
|
||
|
|
self.assertIn(entry1, combined_kind.entries)
|
||
|
|
self.assertIn(entry2, combined_kind.entries)
|
||
|
|
|
||
|
|
def test_combine_children_by_name(self):
|
||
|
|
#
|
||
|
|
# Set up
|
||
|
|
#
|
||
|
|
section = Section("some_section", parent=None)
|
||
|
|
kind_static = Kind("static", parent=section)
|
||
|
|
section._kinds = [kind_static]
|
||
|
|
|
||
|
|
ins1 = InnerNamespace("ins1", parent=kind_static)
|
||
|
|
ins1a = InnerNamespace("ins1", parent=kind_static) # same name deliberately
|
||
|
|
entry1 = Entry(name="entry1", type="int32", kind="static",
|
||
|
|
parent=ins1, hal_version="3.2")
|
||
|
|
entry2 = Entry(name="entry2", type="int32", kind="static",
|
||
|
|
parent=ins1a, hal_version="3.2")
|
||
|
|
entry3 = Entry(name="entry3", type="int32", kind="static",
|
||
|
|
parent=kind_static, hal_version="3.2")
|
||
|
|
|
||
|
|
kind_static._namespaces = [ins1, ins1a]
|
||
|
|
kind_static._entries = [entry3]
|
||
|
|
|
||
|
|
ins1._entries = [entry1]
|
||
|
|
ins1a._entries = [entry2]
|
||
|
|
|
||
|
|
#
|
||
|
|
# Test
|
||
|
|
#
|
||
|
|
combined_children_kind = kind_static.combine_children_by_name()
|
||
|
|
|
||
|
|
self.assertIsInstance(combined_children_kind, Kind)
|
||
|
|
combined_ins = [i for i in combined_children_kind.namespaces]
|
||
|
|
combined_ent = [i for i in combined_children_kind.entries]
|
||
|
|
|
||
|
|
self.assertEquals(section, combined_children_kind.parent)
|
||
|
|
self.assertEquals(1, len(combined_ins))
|
||
|
|
self.assertEquals(1, len(combined_ent))
|
||
|
|
|
||
|
|
self.assertEquals("ins1", combined_ins[0].name)
|
||
|
|
self.assertEquals("entry3", combined_ent[0].name)
|
||
|
|
|
||
|
|
new_ins = combined_ins[0]
|
||
|
|
self.assertIn(entry1, new_ins.entries)
|
||
|
|
self.assertIn(entry2, new_ins.entries)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|