def cq_production_recreate_all(recreate_all=False):
from core.data.production import get_root_nodes
if recreate_all:
for root_node in get_root_nodes():
production_drop_nested_query_and_dependencies(root_node, recreate_all)
for root_node in get_root_nodes():
production_build_nested_query_and_dependencies(root_node, recreate_all)
return True
def cq_multitree_recreate_all(recreate_all=False):
from core.data.multitree import get_root_nodes
if recreate_all:
for root_node in get_root_nodes():
multitree_drop_nested_query_and_dependencies(root_node, recreate_all)
for root_node in get_root_nodes():
multitree_build_nested_query_and_dependencies(root_node, recreate_all)
return True
def test_get_sensors_tree(self):
from core.data.multitree import get_root_nodes, get_sensors_tree
root_nodes = get_root_nodes()
sensors_tree = get_sensors_tree(root_nodes[0])
self.assertIsNotNone(sensors_tree)
self.assertListEqual(sensors_tree, ['cluster_hardware'])
def test_get_root_nodes(self):
from core.data.multitree import get_root_nodes
root_nodes = get_root_nodes()
self.assertEqual(len(root_nodes), 1)
self.assertEqual(root_nodes[0].get("id"), "datacenter")
self.assertListEqual(root_nodes[0].get("children"), ["cluster"])
self.assertEqual(root_nodes[0].get("root", False), True)
def test_get_tree(self):
from core.data.multitree import get_root_nodes, get_tree
root_nodes = get_root_nodes()
tree = get_tree(root_nodes[0])
self.assertIsNotNone(tree)
self.assertEqual(tree["level"], 0)
self.assertEqual(tree["node"], {'id': 'datacenter', 'name': 'Datacenter', 'children': ['cluster'], 'root': True})
self.assertEqual(len(tree["children"]), 1)
self.assertListEqual([child["node"]["id"] for child in tree["children"]], ["cluster"])
def multitree_root_nodes():
return jsonify(get_root_nodes())