def add_ccnodeid_default_resource(node_mgr: NodeManager) -> None:
"""
In order for this to function properly, ccnodeid must be defined
"""
def get_node_id(n: Node) -> Optional[str]:
return n.delayed_node_id.node_id
node_mgr.add_default_resource({}, "ccnodeid", get_node_id)
def preprocess_node_mgr(self, config: Dict, node_mgr: NodeManager) -> None:
"""
We add a default resource to map group_id to node.placement_group
"""
super().preprocess_node_mgr(config, node_mgr)
def group_id(node: Node) -> str:
return node.placement_group if node.placement_group else "_none_"
node_mgr.add_default_resource({},
"group_id",
group_id,
allow_none=False)
def ungrouped(node: Node) -> str:
return str(not bool(node.placement_group)).lower()
node_mgr.add_default_resource({}, "ungrouped", ungrouped)
def test_node_software_configuration_alias(node_mgr: NodeManager) -> None:
node_mgr.add_default_resource({}, "int_alias",
"node.software_configuration.custom_int")
b = node_mgr.get_buckets()[0]
assert b.resources["int_alias"] == b.software_configuration["custom_int"]
def test_node_resources_alias(node_mgr: NodeManager) -> None:
node_mgr.add_default_resource({}, "memgb_alias", "node.resources.memgb")
b = node_mgr.get_buckets()[0]
assert b.resources["memgb_alias"] == b.resources["memgb"]