Can now filter by test usages in find results

citizenmatt · Jan 30, 2013 · 066fd56 · 066fd56
1 parent faf2f71
commit 066fd56
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/Source/Machine.Specifications.ReSharperRunner.6.0/MSpecUnitTestProvider.cs b/Source/Machine.Specifications.ReSharperRunner.6.0/MSpecUnitTestProvider.cs
@@ -119,6 +119,12 @@ public bool IsElementOfKind(IUnitTestElement element, UnitTestElementKind elemen
           return element is ContextSpecificationElement || element is BehaviorSpecificationElement;
         case UnitTestElementKind.TestContainer:
           return element is ContextElement || element is BehaviorElement;
+        case UnitTestElementKind.TestStuff:
+          return element is ContextSpecificationElement || element is BehaviorSpecificationElement ||
+                 element is ContextElement || element is BehaviorElement;
+        case UnitTestElementKind.Unknown:
+          return !(element is ContextSpecificationElement || element is BehaviorSpecificationElement ||
+                   element is ContextElement || element is BehaviorElement);
       }
 
       return false;
@@ -132,6 +138,10 @@ public bool IsElementOfKind(IDeclaredElement declaredElement, UnitTestElementKin
           return declaredElement.IsSpecification();
         case UnitTestElementKind.TestContainer:
           return declaredElement.IsContext() || declaredElement.IsBehavior();
+        case UnitTestElementKind.TestStuff:
+          return declaredElement.IsSpecification() || declaredElement.IsContext() || declaredElement.IsBehavior();
+        case UnitTestElementKind.Unknown:
+          return !(declaredElement.IsSpecification() || declaredElement.IsContext() || declaredElement.IsBehavior());
       }
 
       return false;