def test_filter_only(self):
exp = (["intel", "scsi"], ["intel", "virtio"])
tree_yaml_path = os.path.join(BASEDIR, "tests/.data/mux-selftest.yaml")
tree_yaml_url = f"/:{tree_yaml_path}"
act = yaml_to_mux.create_from_yaml([tree_yaml_url])
act = mux.apply_filters(act, ("/hw/cpu/intel", "/distro/fedora", "/hw"))
act = tuple(mux.MuxTree(act))
self.assertEqual(act, exp)
def test_filter_only(self):
exp = (['intel', 'scsi'], ['intel', 'virtio'])
tree_yaml_path = os.path.join(BASEDIR, 'tests/.data/mux-selftest.yaml')
tree_yaml_url = f'/:{tree_yaml_path}'
act = yaml_to_mux.create_from_yaml([tree_yaml_url])
act = mux.apply_filters(act,
('/hw/cpu/intel', '/distro/fedora', '/hw'))
act = tuple(mux.MuxTree(act))
self.assertEqual(act, exp)
def setUp(self):
yaml_path = os.path.join(BASEDIR, "tests/.data/mux-selftest-params.yaml")
yaml_url = f"/:{yaml_path}"
yamls = yaml_to_mux.create_from_yaml([yaml_url])
self.yamls = iter(mux.MuxTree(yamls))
self.params1 = parameters.AvocadoParams(next(self.yamls), ["/ch0/*", "/ch1/*"])
next(self.yamls) # Skip 2nd
next(self.yamls) # and 3rd
self.params2 = parameters.AvocadoParams(next(self.yamls), ["/ch1/*", "/ch0/*"])
def test_advanced_yaml(self):
tree2_yaml_path = os.path.join(
BASEDIR, "tests/.data/mux-selftest-advanced.yaml"
)
tree2_yaml_url = f"/:{tree2_yaml_path}"
tree2 = yaml_to_mux.create_from_yaml([tree2_yaml_url])
exp = [
"intel",
"amd",
"arm",
"scsi",
"virtio",
"fedora",
"6",
"7",
"gentoo",
"\u0161mint",
"prod",
"new_node",
"on",
"dict",
]
act = tree2.get_leaves()
oldroot = tree2.children[0]
self.assertEqual(exp, act)
self.assertEqual(tree2.children[0].children[0].path, "/\u0161virt/hw")
self.assertEqual({"enterprise": True}, oldroot.children[1].children[1].value)
self.assertEqual({"new_init": "systemd"}, oldroot.children[1].children[0].value)
self.assertEqual({"is_cool": True}, oldroot.children[1].children[2].value)
self.assertEqual(
{"new_value": "\u0161omething"},
oldroot.children[3].children[0].children[0].value,
)
# Convert values, but not keys
self.assertEqual({"on": True, "true": "true"}, oldroot.children[4].value)
# Dicts as values
self.assertEqual(
{
"explicit": {"foo\u0161": "\u0161bar", "bar": "baz"},
"in_list": [{"foo": "bar", "bar": "baz"}],
},
oldroot.children[5].value,
)
# multiplex root (always True)
self.assertEqual(tree2.multiplex, None)
# multiplex /virt/
self.assertEqual(tree2.children[0].multiplex, None)
# multiplex /virt/hw
self.assertEqual(tree2.children[0].children[0].multiplex, None)
# multiplex /virt/distro
self.assertEqual(tree2.children[0].children[1].multiplex, True)
# multiplex /virt/env
self.assertEqual(tree2.children[0].children[2].multiplex, True)
# multiplex /virt/absolutely
self.assertEqual(tree2.children[0].children[3].multiplex, None)
# multiplex /virt/distro/fedora
self.assertEqual(tree2.children[0].children[1].children[0].multiplex, None)
def test_multiple_loaders(self):
"""
Verifies that `create_from_yaml` does not affects the main yaml.Loader
"""
yaml_path = os.path.join(BASEDIR, 'tests/.data/mux-selftest.yaml')
yaml_url = f'/:{yaml_path}'
treenode = yaml_to_mux.create_from_yaml([yaml_url])
self.assertEqual(type(treenode), mux.MuxTreeNode)
self.assertEqual(type(treenode.children[0]), mux.MuxTreeNode)
# equivalent to yaml.load("...", Loader=yaml.SafeLoader)
plain = yaml.safe_load("foo: bar")
self.assertIn(type(plain), [dict, collections.OrderedDict])
def test_advanced_yaml(self):
tree2_yaml_path = os.path.join(
BASEDIR, 'tests/.data/mux-selftest-advanced.yaml')
tree2_yaml_url = f'/:{tree2_yaml_path}'
tree2 = yaml_to_mux.create_from_yaml([tree2_yaml_url])
exp = [
'intel', 'amd', 'arm', 'scsi', 'virtio', 'fedora', '6', '7',
'gentoo', '\u0161mint', 'prod', 'new_node', 'on', 'dict'
]
act = tree2.get_leaves()
oldroot = tree2.children[0]
self.assertEqual(exp, act)
self.assertEqual(tree2.children[0].children[0].path, "/\u0161virt/hw")
self.assertEqual({'enterprise': True},
oldroot.children[1].children[1].value)
self.assertEqual({'new_init': 'systemd'},
oldroot.children[1].children[0].value)
self.assertEqual({'is_cool': True},
oldroot.children[1].children[2].value)
self.assertEqual({'new_value': '\u0161omething'},
oldroot.children[3].children[0].children[0].value)
# Convert values, but not keys
self.assertEqual({
'on': True,
"true": "true"
}, oldroot.children[4].value)
# Dicts as values
self.assertEqual(
{
"explicit": {
"foo\u0161": "\u0161bar",
"bar": "baz"
},
"in_list": [{
"foo": "bar",
"bar": "baz"
}]
}, oldroot.children[5].value)
# multiplex root (always True)
self.assertEqual(tree2.multiplex, None)
# multiplex /virt/
self.assertEqual(tree2.children[0].multiplex, None)
# multiplex /virt/hw
self.assertEqual(tree2.children[0].children[0].multiplex, None)
# multiplex /virt/distro
self.assertEqual(tree2.children[0].children[1].multiplex, True)
# multiplex /virt/env
self.assertEqual(tree2.children[0].children[2].multiplex, True)
# multiplex /virt/absolutely
self.assertEqual(tree2.children[0].children[3].multiplex, None)
# multiplex /virt/distro/fedora
self.assertEqual(tree2.children[0].children[1].children[0].multiplex,
None)
def setUp(self):
yaml_path = os.path.join(BASEDIR,
'tests/.data/mux-selftest-params.yaml')
yaml_url = '/:%s' % yaml_path
yamls = yaml_to_mux.create_from_yaml([yaml_url])
self.yamls = iter(mux.MuxTree(yamls))
self.params1 = parameters.AvocadoParams(next(self.yamls),
['/ch0/*', '/ch1/*'])
next(self.yamls) # Skip 2nd
next(self.yamls) # and 3rd
self.params2 = parameters.AvocadoParams(next(self.yamls),
['/ch1/*', '/ch0/*'])
def test_filter_out(self):
tree_yaml_path = os.path.join(BASEDIR, 'tests/.data/mux-selftest.yaml')
tree_yaml_url = f'/:{tree_yaml_path}'
act = yaml_to_mux.create_from_yaml([tree_yaml_url])
act = mux.apply_filters(act, None,
('/hw/cpu/intel', '/distro/fedora', '/distro'))
act = tuple(mux.MuxTree(act))
self.assertEqual(len(act), 4)
self.assertEqual(len(act[0]), 3)
str_act = str(act)
self.assertIn('amd', str_act)
self.assertIn('prod', str_act)
self.assertNotIn('intel', str_act)
self.assertNotIn('fedora', str_act)
def test_filter_out(self):
tree_yaml_path = os.path.join(BASEDIR, "tests/.data/mux-selftest.yaml")
tree_yaml_url = f"/:{tree_yaml_path}"
act = yaml_to_mux.create_from_yaml([tree_yaml_url])
act = mux.apply_filters(
act, None, ("/hw/cpu/intel", "/distro/fedora", "/distro")
)
act = tuple(mux.MuxTree(act))
self.assertEqual(len(act), 4)
self.assertEqual(len(act[0]), 3)
str_act = str(act)
self.assertIn("amd", str_act)
self.assertIn("prod", str_act)
self.assertNotIn("intel", str_act)
self.assertNotIn("fedora", str_act)
def test_create_variants(self):
tree_yaml_path = os.path.join(BASEDIR, 'tests/.data/mux-selftest.yaml')
tree_yaml_url = f'/:{tree_yaml_path}'
from_file = yaml_to_mux.create_from_yaml([tree_yaml_url])
from_file = mux.MuxTree(from_file)
self.assertEqual(self.mux_full, tuple(from_file))
def setUp(self):
tree_yaml_path = os.path.join(BASEDIR, 'tests/.data/mux-selftest.yaml')
tree_yaml_url = f'/:{tree_yaml_path}'
self.mux_tree = yaml_to_mux.create_from_yaml([tree_yaml_url])
self.mux_full = tuple(mux.MuxTree(self.mux_tree))
class TestMuxTree(unittest.TestCase):
# Share tree with all tests
tree_yaml_path = os.path.join(BASEDIR, 'tests/.data/mux-selftest.yaml')
tree_yaml_url = f'/:{tree_yaml_path}'
tree = yaml_to_mux.create_from_yaml([tree_yaml_url])
def test_node_order(self):
self.assertIsInstance(self.tree, mux.MuxTreeNode)
self.assertEqual('hw', self.tree.children[0])
self.assertEqual({'cpu_CFLAGS': '-march=core2'},
self.tree.children[0].children[0].children[0].value)
disk = self.tree.children[0].children[1]
self.assertEqual('scsi', disk.children[0])
self.assertEqual({
'disk_type': 'scsi',
'corruptlist': ['againlist']
}, disk.children[0].value)
self.assertEqual('virtio', disk.children[1])
self.assertEqual({}, disk.children[1].value)
self.assertEqual('distro', self.tree.children[1])
self.assertEqual('env', self.tree.children[2])
self.assertEqual({'opt_CFLAGS': '-O2'},
self.tree.children[2].children[0].value)
def test_eq(self):
# Copy
tree2 = copy.deepcopy(self.tree)
self.assertEqual(self.tree, tree2)
# Additional node
child = mux.MuxTreeNode("20", {'name': 'Heisenbug'})
tree2.children[1].children[1].add_child(child)
self.assertNotEqual(self.tree, tree2)
# Should match again
child.detach()
self.assertEqual(self.tree, tree2)
# Missing node
tree2.children[1].children[1].detach()
self.assertNotEqual(self.tree, tree2)
self.assertEqual(self.tree.children[0], tree2.children[0])
# Different value
tree2.children[0].children[0].children[0].value = {'something': 'else'}
self.assertNotEqual(self.tree.children[0], tree2.children[0])
tree3 = mux.MuxTreeNode()
self.assertNotEqual(tree3, tree2)
# Merge
tree3.merge(tree2)
self.assertEqual(tree3, tree2)
# Add_child existing
tree3.add_child(tree2.children[0])
self.assertEqual(tree3, tree2)
def test_links(self):
""" Verify child->parent links """
for leaf in self.tree:
self.assertEqual(leaf.root, self.tree)
def test_basic_functions(self):
# repr
self.assertEqual("MuxTreeNode(name='hw')", repr(self.tree.children[0]))
# str
self.assertEqual("/distro/\u0161mint: init=systemv",
astring.to_text(self.tree.children[1].children[1]))
# len
self.assertEqual(8, len(self.tree)) # number of leaves
# __iter__
self.assertEqual(8, sum((1 for _ in self.tree))) # number of leaves
# .root
self.assertEqual(
id(self.tree),
id(self.tree.children[0].children[0].children[0].root))
# .parents
self.assertEqual(['hw', ''], self.tree.children[0].children[0].parents)
# environment / (root)
self.assertEqual({}, self.tree.environment)
# environment /hw (nodes first)
self.assertEqual({'corruptlist': ['upper_node_list']},
self.tree.children[0].environment)
cpu = self.tree.children[0].children[0]
# environment /hw/cpu (mixed env)
self.assertEqual(
{
'corruptlist': ['upper_node_list'],
'joinlist': ['first_item']
}, cpu.environment)
# environment /hw/cpu/amd (list extension)
vals = {
'corruptlist': ['upper_node_list'],
'cpu_CFLAGS': '-march=athlon64',
'joinlist': ['first_item', 'second', 'third']
}
self.assertEqual(vals, cpu.children[1].environment)
# environment /hw/cpu/arm (deep env)
vals = {
'corruptlist': ['upper_node_list'],
'joinlist': ['first_item'],
'cpu_CFLAGS': '-mabi=apcs-gnu '
'-march=armv8-a -mtune=arm8'
}
self.assertEqual(vals, cpu.children[2].environment)
# environment /hw/disk (list -> string)
vals = {'corruptlist': 'nonlist', 'disk_type': 'virtio'}
disk = self.tree.children[0].children[1]
self.assertEqual(vals, disk.environment)
# environment /hw/disk/scsi (string -> list)
vals = {'corruptlist': ['againlist'], 'disk_type': 'scsi'}
self.assertEqual(vals, disk.children[0].environment)
# environment /env
vals = {'opt_CFLAGS': '-Os'}
self.assertEqual(vals, self.tree.children[2].environment)
# leaves order
leaves = [
'intel', 'amd', 'arm', 'scsi', 'virtio', 'fedora', '\u0161mint',
'prod'
]
self.assertEqual(leaves, self.tree.get_leaves())
tree_view = tree.tree_view(self.tree, 0, False)
# ascii treeview contains only ascii chars
tree_view.decode('ascii')
# ascii treeview contain all leaves
for leaf in leaves:
# In ascii mode we replace non-ascii character using
# xmlcharrefreplace, make sure this is performed
leaf = leaf.encode('ascii', errors='xmlcharrefreplace')
self.assertIn(leaf, tree_view,
f"Leaf {leaf} not in ascii:\n{tree_view}")
def test_filters(self):
tree2 = copy.deepcopy(self.tree)
exp = ['intel', 'amd', 'arm', 'fedora', '\u0161mint', 'prod']
act = mux.apply_filters(tree2, filter_only=['/hw/cpu',
'']).get_leaves()
self.assertEqual(exp, act)
tree2 = copy.deepcopy(self.tree)
exp = ['scsi', 'virtio', 'fedora', '\u0161mint', 'prod']
act = mux.apply_filters(tree2, filter_out=['/hw/cpu', '']).get_leaves()
self.assertEqual(exp, act)
def test_merge_trees(self):
tree2 = copy.deepcopy(self.tree)
tree3 = mux.MuxTreeNode()
tree3.add_child(mux.MuxTreeNode('hw', {'another_value': 'bbb'}))
tree3.children[0].add_child(mux.MuxTreeNode('nic'))
tree3.children[0].children[0].add_child(mux.MuxTreeNode('default'))
tree3.children[0].children[0].add_child(
mux.MuxTreeNode('virtio', {'nic': 'virtio'}))
tree3.children[0].add_child(
mux.MuxTreeNode('cpu', {'test_value': ['z']}))
tree2.merge(tree3)
exp = [
'intel', 'amd', 'arm', 'scsi', 'virtio', 'default', 'virtio',
'fedora', '\u0161mint', 'prod'
]
self.assertEqual(exp, tree2.get_leaves())
self.assertEqual(
{
'corruptlist': ['upper_node_list'],
'another_value': 'bbb'
}, tree2.children[0].value)
self.assertEqual({
'joinlist': ['first_item'],
'test_value': ['z']
}, tree2.children[0].children[0].value)
self.assertFalse(tree2.children[0].children[2].children[0].value)
self.assertEqual({'nic': 'virtio'},
tree2.children[0].children[2].children[1].value)
def test_advanced_yaml(self):
tree2_yaml_path = os.path.join(
BASEDIR, 'tests/.data/mux-selftest-advanced.yaml')
tree2_yaml_url = f'/:{tree2_yaml_path}'
tree2 = yaml_to_mux.create_from_yaml([tree2_yaml_url])
exp = [
'intel', 'amd', 'arm', 'scsi', 'virtio', 'fedora', '6', '7',
'gentoo', '\u0161mint', 'prod', 'new_node', 'on', 'dict'
]
act = tree2.get_leaves()
oldroot = tree2.children[0]
self.assertEqual(exp, act)
self.assertEqual(tree2.children[0].children[0].path, "/\u0161virt/hw")
self.assertEqual({'enterprise': True},
oldroot.children[1].children[1].value)
self.assertEqual({'new_init': 'systemd'},
oldroot.children[1].children[0].value)
self.assertEqual({'is_cool': True},
oldroot.children[1].children[2].value)
self.assertEqual({'new_value': '\u0161omething'},
oldroot.children[3].children[0].children[0].value)
# Convert values, but not keys
self.assertEqual({
'on': True,
"true": "true"
}, oldroot.children[4].value)
# Dicts as values
self.assertEqual(
{
"explicit": {
"foo\u0161": "\u0161bar",
"bar": "baz"
},
"in_list": [{
"foo": "bar",
"bar": "baz"
}]
}, oldroot.children[5].value)
# multiplex root (always True)
self.assertEqual(tree2.multiplex, None)
# multiplex /virt/
self.assertEqual(tree2.children[0].multiplex, None)
# multiplex /virt/hw
self.assertEqual(tree2.children[0].children[0].multiplex, None)
# multiplex /virt/distro
self.assertEqual(tree2.children[0].children[1].multiplex, True)
# multiplex /virt/env
self.assertEqual(tree2.children[0].children[2].multiplex, True)
# multiplex /virt/absolutely
self.assertEqual(tree2.children[0].children[3].multiplex, None)
# multiplex /virt/distro/fedora
self.assertEqual(tree2.children[0].children[1].children[0].multiplex,
None)
def test_get_node(self):
self.assertRaises(ValueError, self.tree.get_node, '/non-existing-node')
def test_fingerprint_order(self):
"""
Checks whether different order changes the fingerprint
"""
children1 = (tree.TreeNode("child1"), tree.TreeNode("child2"))
tree1 = tree.TreeNode("root", children=children1)
children2 = (tree.TreeNode("child2"), tree.TreeNode("child1"))
tree2 = tree.TreeNode("root", children=children2)
mux1 = mux.MuxPlugin()
mux2 = mux.MuxPlugin()
mux1.initialize_mux(tree1, "")
mux2.initialize_mux(tree2, "")
variant1 = next(iter(mux1))
variant2 = next(iter(mux2))
self.assertNotEqual(variant1, variant2)
# test variant __str__()
str(variant1)
variant_list = []
for item in variant1:
variant_list.append(f"'{item}': '{variant1[item]}'")
expected_items = [
"'paths': ''", "'variant': '[TreeNode(name='child1'), "
"TreeNode(name='child2')]'", "'variant_id': 'child1-child2-f47e'"
]
for item in expected_items:
self.assertIn(item, variant_list)
variant_list.remove(item)
self.assertFalse(variant_list)
class TestMuxTree(unittest.TestCase):
# Share tree with all tests
tree_yaml_path = os.path.join(BASEDIR, "tests/.data/mux-selftest.yaml")
tree_yaml_url = f"/:{tree_yaml_path}"
tree = yaml_to_mux.create_from_yaml([tree_yaml_url])
def test_node_order(self):
self.assertIsInstance(self.tree, mux.MuxTreeNode)
self.assertEqual("hw", self.tree.children[0])
self.assertEqual(
{"cpu_CFLAGS": "-march=core2"},
self.tree.children[0].children[0].children[0].value,
)
disk = self.tree.children[0].children[1]
self.assertEqual("scsi", disk.children[0])
self.assertEqual(
{"disk_type": "scsi", "corruptlist": ["againlist"]}, disk.children[0].value
)
self.assertEqual("virtio", disk.children[1])
self.assertEqual({}, disk.children[1].value)
self.assertEqual("distro", self.tree.children[1])
self.assertEqual("env", self.tree.children[2])
self.assertEqual({"opt_CFLAGS": "-O2"}, self.tree.children[2].children[0].value)
def test_eq(self):
# Copy
tree2 = copy.deepcopy(self.tree)
self.assertEqual(self.tree, tree2)
# Additional node
child = mux.MuxTreeNode("20", {"name": "Heisenbug"})
tree2.children[1].children[1].add_child(child)
self.assertNotEqual(self.tree, tree2)
# Should match again
child.detach()
self.assertEqual(self.tree, tree2)
# Missing node
tree2.children[1].children[1].detach()
self.assertNotEqual(self.tree, tree2)
self.assertEqual(self.tree.children[0], tree2.children[0])
# Different value
tree2.children[0].children[0].children[0].value = {"something": "else"}
self.assertNotEqual(self.tree.children[0], tree2.children[0])
tree3 = mux.MuxTreeNode()
self.assertNotEqual(tree3, tree2)
# Merge
tree3.merge(tree2)
self.assertEqual(tree3, tree2)
# Add_child existing
tree3.add_child(tree2.children[0])
self.assertEqual(tree3, tree2)
def test_links(self):
"""Verify child->parent links"""
for leaf in self.tree:
self.assertEqual(leaf.root, self.tree)
def test_basic_functions(self):
# repr
self.assertEqual("MuxTreeNode(name='hw')", repr(self.tree.children[0]))
# str
self.assertEqual(
"/distro/\u0161mint: init=systemv",
astring.to_text(self.tree.children[1].children[1]),
)
# len
self.assertEqual(8, len(self.tree)) # number of leaves
# __iter__
self.assertEqual(8, sum((1 for _ in self.tree))) # number of leaves
# .root
self.assertEqual(
id(self.tree), id(self.tree.children[0].children[0].children[0].root)
)
# .parents
self.assertEqual(["hw", ""], self.tree.children[0].children[0].parents)
# environment / (root)
self.assertEqual({}, self.tree.environment)
# environment /hw (nodes first)
self.assertEqual(
{"corruptlist": ["upper_node_list"]}, self.tree.children[0].environment
)
cpu = self.tree.children[0].children[0]
# environment /hw/cpu (mixed env)
self.assertEqual(
{"corruptlist": ["upper_node_list"], "joinlist": ["first_item"]},
cpu.environment,
)
# environment /hw/cpu/amd (list extension)
vals = {
"corruptlist": ["upper_node_list"],
"cpu_CFLAGS": "-march=athlon64",
"joinlist": ["first_item", "second", "third"],
}
self.assertEqual(vals, cpu.children[1].environment)
# environment /hw/cpu/arm (deep env)
vals = {
"corruptlist": ["upper_node_list"],
"joinlist": ["first_item"],
"cpu_CFLAGS": "-mabi=apcs-gnu " "-march=armv8-a -mtune=arm8",
}
self.assertEqual(vals, cpu.children[2].environment)
# environment /hw/disk (list -> string)
vals = {"corruptlist": "nonlist", "disk_type": "virtio"}
disk = self.tree.children[0].children[1]
self.assertEqual(vals, disk.environment)
# environment /hw/disk/scsi (string -> list)
vals = {"corruptlist": ["againlist"], "disk_type": "scsi"}
self.assertEqual(vals, disk.children[0].environment)
# environment /env
vals = {"opt_CFLAGS": "-Os"}
self.assertEqual(vals, self.tree.children[2].environment)
# leaves order
leaves = [
"intel",
"amd",
"arm",
"scsi",
"virtio",
"fedora",
"\u0161mint",
"prod",
]
self.assertEqual(leaves, self.tree.get_leaves())
tree_view = tree.tree_view(self.tree, 0, False)
# ascii treeview contains only ascii chars
tree_view.decode("ascii")
# ascii treeview contain all leaves
for leaf in leaves:
# In ascii mode we replace non-ascii character using
# xmlcharrefreplace, make sure this is performed
leaf = leaf.encode("ascii", errors="xmlcharrefreplace")
self.assertIn(leaf, tree_view, f"Leaf {leaf} not in ascii:\n{tree_view}")
def test_filters(self):
tree2 = copy.deepcopy(self.tree)
exp = ["intel", "amd", "arm", "fedora", "\u0161mint", "prod"]
act = mux.apply_filters(tree2, filter_only=["/hw/cpu", ""]).get_leaves()
self.assertEqual(exp, act)
tree2 = copy.deepcopy(self.tree)
exp = ["scsi", "virtio", "fedora", "\u0161mint", "prod"]
act = mux.apply_filters(tree2, filter_out=["/hw/cpu", ""]).get_leaves()
self.assertEqual(exp, act)
def test_merge_trees(self):
tree2 = copy.deepcopy(self.tree)
tree3 = mux.MuxTreeNode()
tree3.add_child(mux.MuxTreeNode("hw", {"another_value": "bbb"}))
tree3.children[0].add_child(mux.MuxTreeNode("nic"))
tree3.children[0].children[0].add_child(mux.MuxTreeNode("default"))
tree3.children[0].children[0].add_child(
mux.MuxTreeNode("virtio", {"nic": "virtio"})
)
tree3.children[0].add_child(mux.MuxTreeNode("cpu", {"test_value": ["z"]}))
tree2.merge(tree3)
exp = [
"intel",
"amd",
"arm",
"scsi",
"virtio",
"default",
"virtio",
"fedora",
"\u0161mint",
"prod",
]
self.assertEqual(exp, tree2.get_leaves())
self.assertEqual(
{"corruptlist": ["upper_node_list"], "another_value": "bbb"},
tree2.children[0].value,
)
self.assertEqual(
{"joinlist": ["first_item"], "test_value": ["z"]},
tree2.children[0].children[0].value,
)
self.assertFalse(tree2.children[0].children[2].children[0].value)
self.assertEqual(
{"nic": "virtio"}, tree2.children[0].children[2].children[1].value
)
def test_advanced_yaml(self):
tree2_yaml_path = os.path.join(
BASEDIR, "tests/.data/mux-selftest-advanced.yaml"
)
tree2_yaml_url = f"/:{tree2_yaml_path}"
tree2 = yaml_to_mux.create_from_yaml([tree2_yaml_url])
exp = [
"intel",
"amd",
"arm",
"scsi",
"virtio",
"fedora",
"6",
"7",
"gentoo",
"\u0161mint",
"prod",
"new_node",
"on",
"dict",
]
act = tree2.get_leaves()
oldroot = tree2.children[0]
self.assertEqual(exp, act)
self.assertEqual(tree2.children[0].children[0].path, "/\u0161virt/hw")
self.assertEqual({"enterprise": True}, oldroot.children[1].children[1].value)
self.assertEqual({"new_init": "systemd"}, oldroot.children[1].children[0].value)
self.assertEqual({"is_cool": True}, oldroot.children[1].children[2].value)
self.assertEqual(
{"new_value": "\u0161omething"},
oldroot.children[3].children[0].children[0].value,
)
# Convert values, but not keys
self.assertEqual({"on": True, "true": "true"}, oldroot.children[4].value)
# Dicts as values
self.assertEqual(
{
"explicit": {"foo\u0161": "\u0161bar", "bar": "baz"},
"in_list": [{"foo": "bar", "bar": "baz"}],
},
oldroot.children[5].value,
)
# multiplex root (always True)
self.assertEqual(tree2.multiplex, None)
# multiplex /virt/
self.assertEqual(tree2.children[0].multiplex, None)
# multiplex /virt/hw
self.assertEqual(tree2.children[0].children[0].multiplex, None)
# multiplex /virt/distro
self.assertEqual(tree2.children[0].children[1].multiplex, True)
# multiplex /virt/env
self.assertEqual(tree2.children[0].children[2].multiplex, True)
# multiplex /virt/absolutely
self.assertEqual(tree2.children[0].children[3].multiplex, None)
# multiplex /virt/distro/fedora
self.assertEqual(tree2.children[0].children[1].children[0].multiplex, None)
def test_get_node(self):
self.assertRaises(ValueError, self.tree.get_node, "/non-existing-node")
def test_fingerprint_order(self):
"""
Checks whether different order changes the fingerprint
"""
children1 = (tree.TreeNode("child1"), tree.TreeNode("child2"))
tree1 = tree.TreeNode("root", children=children1)
children2 = (tree.TreeNode("child2"), tree.TreeNode("child1"))
tree2 = tree.TreeNode("root", children=children2)
mux1 = mux.MuxPlugin()
mux2 = mux.MuxPlugin()
mux1.initialize_mux(tree1, "")
mux2.initialize_mux(tree2, "")
variant1 = next(iter(mux1))
variant2 = next(iter(mux2))
self.assertNotEqual(variant1, variant2)
# test variant __str__()
str(variant1)
variant_list = []
for item in variant1:
variant_list.append(f"'{item}': '{variant1[item]}'")
expected_items = [
"'paths': ''",
"'variant': '[TreeNode(name='child1'), TreeNode(name='child2')]'",
"'variant_id': 'child1-child2-f47e'",
]
for item in expected_items:
self.assertIn(item, variant_list)
variant_list.remove(item)
self.assertFalse(variant_list)
