Fixing bug #1529 "Using '(De)select all tests' option will expand all nodes" by simply remembering the state of the element when it is walked over and restore it in the end.

Author: fzuellich

src/robotide/ui/tree.py

@@ -639,14 +639,21 @@ def _for_all_tests(self, item, func):
         if not self.HasAGWFlag(customtreectrl.TR_HIDE_ROOT) or item != self.GetRootItem():
             if isinstance(item.GetData(), ResourceRootHandler or ResourceFileHandler):
                 return
-            self.Expand(item)
+
+            is_item_expanded = self.IsExpanded(item)
+            if is_item_expanded is False:
+                self.Expand(item)
             if self._is_test_node(item):
                 func(item)
             if not self.IsExpanded(item):
                 return
+
         for child in item.GetChildren():
             self._for_all_tests(child, func)
 
+        if is_item_expanded is False:
+            self.Collapse(item)
+
     def _for_all_drawn_tests(self, item, func):
         if self._is_test_node(item):
             func(item)