def test_generated_items(self, get_all_attrs_from_file): def my_item_generator(node, bundle, item): generated_items = {'files': {}} if item.ITEM_TYPE_NAME == 'user': file_path = "/home/{}/.screenrc".format(item.name) generated_items['files'][file_path] = { 'content': "foo", } return generated_items get_all_attrs_from_file.return_value = { 'item_generators': [ 'test.generator', ], 'users': { 'jdoe': {}, }, } repo = Repository() repo.bundle_names = ["generatingbundle"] repo.libs = MagicMock() repo.libs.test.generator = my_item_generator node = Node("node1", {'bundles': ["generatingbundle"]}) repo.add_node(node) self.assertEqual(len(node.bundles[0]._generated_items), 1) generated_item = node.bundles[0]._generated_items[0] self.assertEqual(generated_item.id, "file:/home/jdoe/.screenrc")