undo-a-rino

Author: robertcoltheart

Source/Machine.VSTestAdapter/Discovery/BuiltIn/TestDiscoverer.cs

@@ -57,8 +57,8 @@ private IEnumerable<MSpecTestCase> CreateTestCase(Context context, string assemb
                     testCase.LineNumber = locationInfo.LineNumber;
                 }
 
-                if (spec.GetType().Name == "BehaviorSpecification")
-                    testCase.BehaviorFieldName = GetBehaviorFieldName(spec);
+                if (spec is BehaviorSpecification behaviorSpec)
+                    testCase.BehaviorFieldName = GetBehaviorFieldName(behaviorSpec);
 
                 if (context.Tags != null)
                     testCase.Tags = context.Tags.Select(tag => tag.Name).ToArray();
@@ -70,7 +70,7 @@ private IEnumerable<MSpecTestCase> CreateTestCase(Context context, string assemb
             }
         }
 
-        private string GetBehaviorFieldName(Specification specification)
+        private string GetBehaviorFieldName(BehaviorSpecification specification)
         {
             if (behaviorProperty?.GetValue(specification) is FieldInfo field)
                 return field.Name;

Source/Machine.VSTestAdapter/Execution/TestExecutor.cs

@@ -73,7 +73,7 @@ public void RunTestsInAssembly(string pathToAssembly, IEnumerable<VisualStudioTe
 
                     Specification specification = context.Specifications.SingleOrDefault(spec => spec.FieldInfo.Name.Equals(test.FieldName, StringComparison.Ordinal));
 
-                    if (specification?.GetType().Name == "BehaviorSpecification")
+                    if (specification is BehaviorSpecification)
                     {
                         // MSpec doesn't expose any way to run an an "It" coming from a "[Behavior]", so we have to do some trickery
                         VisualStudioTestIdentifier listenFor = specification.ToVisualStudioTestIdentifier(context);