From d66e16c46f649cbf47ef1bab393c8903a68aec74 Mon Sep 17 00:00:00 2001
From: IvenBach <ivenbach@gmail.com>
Date: Thu, 9 May 2019 18:28:32 -0700
Subject: [PATCH] Add filtering for TestExplorer

---
 .../UI/UnitTesting/TestExplorerControl.xaml   | 12 +++++
 .../UI/UnitTesting/TestExplorerViewModel.cs   | 49 +++++++++++++++++++
 .../UnitTesting/TestExplorer.Designer.cs      | 18 +++++++
 .../UnitTesting/TestExplorer.resx             |  3 ++
 4 files changed, 82 insertions(+)

diff --git a/Rubberduck.Core/UI/UnitTesting/TestExplorerControl.xaml b/Rubberduck.Core/UI/UnitTesting/TestExplorerControl.xaml
index 21535950d0..20580a85d7 100644
--- a/Rubberduck.Core/UI/UnitTesting/TestExplorerControl.xaml
+++ b/Rubberduck.Core/UI/UnitTesting/TestExplorerControl.xaml
@@ -216,6 +216,18 @@
 
                 <Separator />
 
+                <Label Content="{Resx Key=TestExplorer_Filter, ResxName=Rubberduck.Resources.UnitTesting.TestExplorer}" />
+                <TextBox MinWidth="50"
+                         Text="{Binding TestNameFilter, UpdateSourceTrigger=PropertyChanged}"/>
+
+                <Label Content="{Resx Key=TestExplorer_Outcome, ResxName=Rubberduck.Resources.UnitTesting.TestExplorer}" />
+
+                <ComboBox Width="100" 
+                          ItemsSource="{Binding OutcomeFilters, UpdateSourceTrigger=PropertyChanged}"
+                          SelectedItem="{Binding SelectedOutcomeFilter, Mode=TwoWay, UpdateSourceTrigger=PropertyChanged}" />
+
+                <Separator />
+
                 <Button Name="CollapseAll" Command="{Binding CollapseAllCommand}" Margin="2"
                         ToolTip="{Resx ResxName=Rubberduck.Resources.RubberduckUI, Key=InspectionResults_CollapseAll}">
                     <Image Source="{StaticResource CollapseAllImage}" Style="{StaticResource ToolbarImageOpacity}" />
diff --git a/Rubberduck.Core/UI/UnitTesting/TestExplorerViewModel.cs b/Rubberduck.Core/UI/UnitTesting/TestExplorerViewModel.cs
index 7ff0b6178a..dad262a888 100644
--- a/Rubberduck.Core/UI/UnitTesting/TestExplorerViewModel.cs
+++ b/Rubberduck.Core/UI/UnitTesting/TestExplorerViewModel.cs
@@ -18,6 +18,7 @@
 using Rubberduck.VBEditor.ComManagement;
 using Rubberduck.VBEditor.Utility;
 using DataFormats = System.Windows.DataFormats;
+using Rubberduck.Resources.UnitTesting;
 
 namespace Rubberduck.UI.UnitTesting
 {
@@ -67,6 +68,10 @@ internal sealed class TestExplorerViewModel : ViewModelBase, INavigateSelection,
 
             OnPropertyChanged(nameof(Tests));
             TestGrouping = TestExplorerGrouping.Outcome;
+
+            OutcomeFilters = new System.Collections.ObjectModel.ObservableCollection<string>(
+                new[] { _allResultsFilter }
+                    .Concat(Enum.GetNames(typeof(TestOutcome)).Select(s => s.ToString())));
         }
 
         public TestExplorerModel Model { get; }
@@ -143,6 +148,50 @@ public bool ExpandedState
             }
         }
 
+        public System.Collections.ObjectModel.ObservableCollection<string> OutcomeFilters { get; }
+
+        private string _testNameFilter = string.Empty;
+        public string TestNameFilter
+        {
+            get => _testNameFilter;
+            set
+            {
+                if (_testNameFilter != value)
+                {
+                    _testNameFilter = value;
+                    OnPropertyChanged();
+                    Tests.Filter = item => FilterResults(item);
+                }
+            }
+        }
+
+        private static readonly string _allResultsFilter = TestExplorer.ResourceManager.GetString("TestExplorer_AllResults", CultureInfo.CurrentUICulture);
+        private string _selectedOutcomeFilter = _allResultsFilter;
+        public string SelectedOutcomeFilter
+        {
+            get => _selectedOutcomeFilter;
+            set
+            {
+                if (_selectedOutcomeFilter != value)
+                {
+                    _selectedOutcomeFilter = value.Replace(" ", string.Empty);
+                    OnPropertyChanged();
+                    Tests.Filter = item => FilterResults(item);
+                }
+            }
+        }
+
+        private bool FilterResults(object unitTest)
+        {
+            OnPropertyChanged(nameof(Tests));
+
+            var testMethodViewModel = unitTest as TestMethodViewModel;
+            var memberName = testMethodViewModel.QualifiedName.MemberName;
+
+            return memberName.ToUpper().Contains(TestNameFilter.ToUpper())
+                && (SelectedOutcomeFilter.Equals(_allResultsFilter) || testMethodViewModel.Result.Outcome.ToString().Equals(_selectedOutcomeFilter));
+        }
+
         private void HandleTestCompletion(object sender, TestCompletedEventArgs e)
         {
             if (TestGrouping != TestExplorerGrouping.Outcome)
diff --git a/Rubberduck.Resources/UnitTesting/TestExplorer.Designer.cs b/Rubberduck.Resources/UnitTesting/TestExplorer.Designer.cs
index 8c1305830a..8cfc92a37b 100644
--- a/Rubberduck.Resources/UnitTesting/TestExplorer.Designer.cs
+++ b/Rubberduck.Resources/UnitTesting/TestExplorer.Designer.cs
@@ -105,6 +105,15 @@ public class TestExplorer {
             }
         }
         
+        /// <summary>
+        ///   Looks up a localized string similar to All.
+        /// </summary>
+        public static string TestExplorer_AllResults {
+            get {
+                return ResourceManager.GetString("TestExplorer_AllResults", resourceCulture);
+            }
+        }
+        
         /// <summary>
         ///   Looks up a localized string similar to Category.
         /// </summary>
@@ -159,6 +168,15 @@ public class TestExplorer {
             }
         }
         
+        /// <summary>
+        ///   Looks up a localized string similar to Test Name.
+        /// </summary>
+        public static string TestExplorer_Filter {
+            get {
+                return ResourceManager.GetString("TestExplorer_Filter", resourceCulture);
+            }
+        }
+        
         /// <summary>
         ///   Looks up a localized string similar to Navigate to selected.
         /// </summary>
diff --git a/Rubberduck.Resources/UnitTesting/TestExplorer.resx b/Rubberduck.Resources/UnitTesting/TestExplorer.resx
index 003dd02902..980bf84ae1 100644
--- a/Rubberduck.Resources/UnitTesting/TestExplorer.resx
+++ b/Rubberduck.Resources/UnitTesting/TestExplorer.resx
@@ -343,4 +343,7 @@
     <value>Spectacular Failure</value>
     <comment>Easter egg status.</comment>
   </data>
+  <data name="TestExplorer_Filter" xml:space="preserve">
+    <value>Test Name</value>
+  </data>
 </root>
\ No newline at end of file