Fixing highlight in test fixture issue

JustoSenka · Apr 12, 2019 · 9b61663 · 9b61663
1 parent b963b36
commit 9b61663
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/RobotEditor/Windows/Base/BaseHierarchyWindow.cs b/RobotEditor/Windows/Base/BaseHierarchyWindow.cs
@@ -128,7 +128,9 @@ protected virtual void OnCommandRunning(Guid guid)
             if (recording == null)
                 return;
 
-            var recordingNode = m_Nodes.FirstOrDefault(node => node.Recording == recording);
+            // For Hierarchy, recordings are in root, for Test Fixtures, recordings are one level deeper. Supports both
+            var recNodesToSearch = m_Nodes.First()?.Recording == null ? m_Nodes.SelectMany(n => n.Children) : m_Nodes;
+            var recordingNode = recNodesToSearch.FirstOrDefault(node => node.Recording == recording);
             if (recordingNode == null)
                 return;