Fixes #303 Dont run test if current file is not test file

microsoft · Jan 13, 2017 · a46f313 · a46f313
1 parent 99c0398
commit a46f313
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/goTest.ts b/src/goTest.ts
@@ -51,6 +51,10 @@ export function testAtCursor(goConfig: vscode.WorkspaceConfiguration) {
 		vscode.window.showInformationMessage('No editor is active.');
 		return;
 	}
+	if (!editor.document.fileName.endsWith('_test.go')) {
+		vscode.window.showInformationMessage('No tests found. Current file is not a test file.');
+		return;
+	}
 	getTestFunctions(editor.document).then(testFunctions => {
 		let testFunction: vscode.SymbolInformation;
 		// Find any test function containing the cursor.
@@ -105,6 +109,10 @@ export function testCurrentFile(goConfig: vscode.WorkspaceConfiguration): Thenab
 		vscode.window.showInformationMessage('No editor is active.');
 		return;
 	}
+	if (!editor.document.fileName.endsWith('_test.go')) {
+		vscode.window.showInformationMessage('No tests found. Current file is not a test file.');
+		return;
+	}
 	return getTestFunctions(editor.document).then(testFunctions => {
 		return goTest({
 			goConfig: goConfig,