def test_apply(self, apply_items, NodeLock, ApplyResult):
repo = Repository()
n = Node("node1", {})
repo.add_node(n)
result = MagicMock()
ApplyResult.return_value = result
NodeLock.__enter__ = lambda x: x
NodeLock.__exit__ = lambda x: x
self.assertEqual(n.apply(), result)
self.assertEqual(apply_items.call_count, 1)
ApplyResult.assert_called_once()
def test_bundles(self):
repo = Repository()
repo.bundle_names = ("bundle1", "bundle2", "bundle3")
n = Node("node1", {})
repo.add_node(n)
g1 = Group("group1", {"bundles": ("bundle1", "bundle2")})
repo.add_group(g1)
g2 = Group("group2", {"bundles": ("bundle3",)})
repo.add_group(g2)
with patch("tests.unit.node_tests.Node.groups", new=(g1, g2)):
self.assertEqual(tuple(names(n.bundles)), ("bundle1", "bundle2", "bundle3"))
def test_static_members(self):
repo = Repository()
node1 = Node("node1")
node2 = Node("node2")
repo.add_node(node1)
repo.add_node(node2)
group = Group("group1", {'members': ("node2", "node1")})
repo.add_group(group)
self.assertEqual(
set(group._nodes_from_static_members),
set((node1, node2)),
)
def test_static_subgroup_members(self, *args):
repo = Repository()
group1 = Group("group1", {'subgroups': ("group2",)})
node3 = Node("node3")
node4 = Node("node4")
repo.add_group(group1)
repo.add_group(Group("group2", {'members': ("node3", "node4")}))
repo.add_node(node3)
repo.add_node(node4)
repo.add_node(Node("node5"))
self.assertEqual(
set(group1._nodes_from_subgroups),
set((node3, node4)),
)
def test_pattern_members(self, *args):
repo = Repository()
repo.add_node(Node("node1"))
repo.add_node(Node("node2"))
group = Group("all", { 'member_patterns': (r".*",) })
repo.add_group(group)
self.assertEqual(
list(names(group.nodes)),
["node1", "node2"],
)
group2 = Group("group2", { 'member_patterns': (r".*2",)} )
repo.add_group(group2)
self.assertEqual(
list(names(group2.nodes)),
["node2"],
)
