def wait_for_data_sets_is_expanded(self, node_id, content):
print("Wait for node is expanded...")
node_top = self.get_tree_element_top_by_id(node_id)
node_padding = self.get_tree_element_padding_by_id(node_id)
child_nodes = self.find_them(UI.get_tree_nodes_locator(), parent=content)
for child_node in child_nodes:
highlight(child_node)
child_parent = self.get_parent_node(child_node)
child_top = self.get_element_top(child_parent)
if lt_pixels(child_top, node_top):
continue
child_padding = self.get_element_padding_left(child_node)
if lt_pixels(node_padding, child_padding):
print("Wait for node is expanded...Ok", node_padding, child_padding)
return
print("Wait for node is expanded...Waiting")
node = self.find_it((By.ID, node_id))
highlight(node)
arrow = self.find_it(UI.get_theia_expand_arrow_locator(), parent=node)
highlight(arrow)
is_expanded = self.is_element_expanded(arrow)
if is_expanded:
self.page_down_tree(content)
raise WebDriverException
def collapse_tree_node(self, node):
try:
parent_node = self.get_parent_node(node)
arrow = self.find_it(UI.get_theia_expand_arrow_locator(), parent=parent_node)
is_expanded = self.is_element_expanded(arrow)
if is_expanded:
self.click_me(node)
except Exception:
dump(self.get_driver(), reason=self.collapse_tree_node.__name__)
def click_node_to_expand(self, node):
parent_node = self.get_parent_node(node)
child_node_content = self.find_it(UI.get_tree_node_content_locator(), parent=node)
node_id = child_node_content.get_attribute(constants.TYPE_ID)
if constants.THEIA_TREE_EXPANDABLE_NODES in node.get_attribute(constants.TYPE_CLASS):
arrow = self.find_it(UI.get_theia_expand_arrow_locator(), parent=parent_node)
is_expanded = self.is_element_expanded(arrow)
if not is_expanded:
highlight(node)
self.click_me(node)
return node_id
