Merge pull request #1478 from AbhitejJohn/mstestsupport

Support to run MSTest based tests
dotnet · May 15, 2017 · 6d0dfaa · 6d0dfaa
2 parents d92fa4e + 841b9a4
commit 6d0dfaa
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/features/dotnetTest.ts b/src/features/dotnetTest.ts
@@ -184,10 +184,19 @@ export function updateCodeLensForTest(bucket: vscode.CodeLens[], fileName: strin
         return;
     }
 
-    let testFeature = node.Features.find(value => (value.Name == 'XunitTestMethod' || value.Name == 'NUnitTestMethod'));
+    let testFeature = node.Features.find(value => (value.Name == 'XunitTestMethod' || value.Name == 'NUnitTestMethod' || value.Name == 'MSTestMethod'));
     if (testFeature) {
         // this test method has a test feature
-        let testFrameworkName = testFeature.Name == 'XunitTestMethod' ? 'xunit' : 'nunit';
+        let testFrameworkName = 'xunit';
+        if(testFeature.Name == 'NunitTestMethod')
+        {
+            testFrameworkName = 'nunit';
+        }
+        else if(testFeature.Name == 'MSTestMethod')
+        {
+            testFrameworkName = 'mstest';
+        }
+
         bucket.push(new vscode.CodeLens(
             toRange(node.Location),
             { title: "run test", command: 'dotnet.test.run', arguments: [testFeature.Data, fileName, testFrameworkName] }));