feat/watch command

src/codelens/TestWatchRunnerCodeLens.ts

@@ -0,0 +1,16 @@
+import { CodeLens, Range, WorkspaceFolder } from "vscode";
+
+export default class TestWatchRunnerCodeLens extends CodeLens {
+  constructor(
+    rootPath: WorkspaceFolder,
+    fileName: string,
+    testName: string,
+    range: Range
+  ) {
+    super(range, {
+      arguments: [rootPath, fileName, testName],
+      command: "testify.run.watch",
+      title: "Watch Test"
+    });
+  }
+}

src/commands/watchTestCommand.ts

@@ -0,0 +1,17 @@
+import { relative } from "path";
+import { WorkspaceFolder } from "vscode";
+
+import { getTestRunner } from "../runners/TestRunnerFactory";
+
+async function runWatch(
+  rootPath: WorkspaceFolder,
+  fileName: string,
+  testName: string
+) {
+  const relativeFilename = relative(rootPath.uri.fsPath, fileName);
+  const testRunner = await getTestRunner(rootPath);
+
+  testRunner.runWatch(rootPath, relativeFilename, testName);
+}
+
+export default runWatch;

src/extension.ts

@@ -1,6 +1,7 @@
 import { commands, ExtensionContext, languages } from "vscode";
 import debugTestCommand from "./commands/debugTestCommand";
 import runTestCommand from "./commands/runTestCommand";
+import watchTestCommand from "./commands/watchTestCommand";
 import FILE_SELECTOR from "./constants/fileSelector";
 import TestRunnerCodeLensProvider from "./providers/TestRunnerCodeLensProvider";
 
@@ -13,5 +14,6 @@ export function activate(context: ExtensionContext) {
   );
 
   commands.registerCommand("testify.run.test", runTestCommand);
+  commands.registerCommand("testify.run.watch", watchTestCommand);
   commands.registerCommand("testify.debug.test", debugTestCommand);
 }

src/interfaces/ITestRunnerInterface.ts

@@ -8,6 +8,7 @@ export interface ITestRunnerInterface {
   terminalProvider: TerminalProvider;
   configurationProvider: ConfigurationProvider;
 
+  runWatch(rootPath: WorkspaceFolder, fileName: string, testName: string): void;
   runTest(rootPath: WorkspaceFolder, fileName: string, testName: string): void;
   debugTest(
     rootPath: WorkspaceFolder,

src/providers/TestRunnerCodeLensProvider.ts

@@ -2,6 +2,7 @@ import { CodeLens, CodeLensProvider, TextDocument, workspace } from "vscode";
 
 import TestRunnerDebugCodeLens from "../codelens/TestDebugRunnerCodeLens";
 import TestRunnerCodeLens from "../codelens/TestRunnerCodeLens";
+import TestWatchRunnerCodeLens from '../codelens/TestWatchRunnerCodeLens';
 import { codeParser } from "../parser/codeParser";
 
 function getRootPath({ uri }) {
@@ -15,6 +16,13 @@ function getRootPath({ uri }) {
 }
 
 function getCodeLens(rootPath, fileName, testName, startPosition) {
+  const watchRunnerCodeLens = new TestWatchRunnerCodeLens(
+    rootPath,
+    fileName,
+    testName,
+    startPosition
+  );
+
   const testRunnerCodeLens = new TestRunnerCodeLens(
     rootPath,
     fileName,
@@ -29,7 +37,7 @@ function getCodeLens(rootPath, fileName, testName, startPosition) {
     startPosition
   );
 
-  return [testRunnerCodeLens, debugRunnerCodeLens];
+  return [testRunnerCodeLens, debugRunnerCodeLens, watchRunnerCodeLens];
 }
 
 export default class TestRunnerCodeLensProvider implements CodeLensProvider {

src/runners/AvaTestRunner.ts

@@ -24,14 +24,36 @@ export class AvaTestRunner implements ITestRunnerInterface {
     }
   }
 
+  public runWatch(
+    rootPath: WorkspaceFolder,
+    fileName: string,
+    testName: string
+  ) {
+    const additionalArguments = this.configurationProvider.additionalArguments;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
+
+    const command = `${this.path} ${this.transformFileName(
+      fileName
+    )} -m "${testName}" --watch ${additionalArguments}`;
+
+    const terminal = this.terminalProvider.get(
+      { env: environmentVariables },
+      rootPath
+    );
+
+    terminal.sendText(command, true);
+    terminal.show(true);
+  }
+
   public runTest(
     rootPath: WorkspaceFolder,
     fileName: string,
     testName: string
   ) {
     const additionalArguments = this.configurationProvider.additionalArguments;
-    const environmentVariables = this.configurationProvider
-      .environmentVariables;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
 
     const command = `${this.path} ${this.transformFileName(
       fileName
@@ -52,8 +74,8 @@ export class AvaTestRunner implements ITestRunnerInterface {
     testName: string
   ) {
     const additionalArguments = this.configurationProvider.additionalArguments;
-    const environmentVariables = this.configurationProvider
-      .environmentVariables;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
     const skipFiles = this.configurationProvider.skipFiles;
 
     debug.startDebugging(rootPath, {

src/runners/JestTestRunner.ts

@@ -24,14 +24,36 @@ export class JestTestRunner implements ITestRunnerInterface {
     }
   }
 
+  public runWatch(
+    rootPath: WorkspaceFolder,
+    fileName: string,
+    testName: string
+  ) {
+    const additionalArguments = this.configurationProvider.additionalArguments;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
+
+    const command = `${this.path} ${this.transformFileName(
+      fileName
+    )} --testNamePattern="${testName}" --watch ${additionalArguments}`;
+
+    const terminal = this.terminalProvider.get(
+      { env: environmentVariables },
+      rootPath
+    );
+
+    terminal.sendText(command, true);
+    terminal.show(true);
+  }
+
   public runTest(
     rootPath: WorkspaceFolder,
     fileName: string,
     testName: string
   ) {
     const additionalArguments = this.configurationProvider.additionalArguments;
-    const environmentVariables = this.configurationProvider
-      .environmentVariables;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
 
     const command = `${this.path} ${this.transformFileName(
       fileName
@@ -52,8 +74,8 @@ export class JestTestRunner implements ITestRunnerInterface {
     testName: string
   ) {
     const additionalArguments = this.configurationProvider.additionalArguments;
-    const environmentVariables = this.configurationProvider
-      .environmentVariables;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
     const skipFiles = this.configurationProvider.skipFiles;
 
     debug.startDebugging(rootPath, {

src/runners/MochaTestRunner.ts

@@ -24,18 +24,36 @@ export class MochaTestRunner implements ITestRunnerInterface {
     }
   }
 
+  public runWatch(
+    rootPath: WorkspaceFolder,
+    fileName: string,
+    testName: string
+  ) {
+    const additionalArguments = this.configurationProvider.additionalArguments;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
+
+    const command = `${this.path} ${fileName} --fgrep="${testName}" --watch ${additionalArguments}`;
+
+    const terminal = this.terminalProvider.get(
+      { env: environmentVariables },
+      rootPath
+    );
+
+    terminal.sendText(command, true);
+    terminal.show(true);
+  }
+
   public runTest(
     rootPath: WorkspaceFolder,
     fileName: string,
     testName: string
   ) {
     const additionalArguments = this.configurationProvider.additionalArguments;
-    const environmentVariables = this.configurationProvider
-      .environmentVariables;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
 
-    const command = `${
-      this.path
-    } ${fileName} --fgrep="${testName}" ${additionalArguments}`;
+    const command = `${this.path} ${fileName} --fgrep="${testName}" ${additionalArguments}`;
 
     const terminal = this.terminalProvider.get(
       { env: environmentVariables },
@@ -52,8 +70,8 @@ export class MochaTestRunner implements ITestRunnerInterface {
     testName: string
   ) {
     const additionalArguments = this.configurationProvider.additionalArguments;
-    const environmentVariables = this.configurationProvider
-      .environmentVariables;
+    const environmentVariables =
+      this.configurationProvider.environmentVariables;
     const skipFiles = this.configurationProvider.skipFiles;
 
     debug.startDebugging(rootPath, {

src/runners/PlaywrightTestRunner.ts

@@ -23,6 +23,21 @@ export class PlaywrightTestRunner implements ITestRunnerInterface {
     }
   }
 
+  /**
+   * Playwright test runner executes default test function since it doesn't support watch yet.
+   * https://github.com/microsoft/playwright/issues/7035
+   * @param rootPath
+   * @param fileName
+   * @param testName
+   */
+  public runWatch(
+    rootPath: WorkspaceFolder,
+    fileName: string,
+    testName: string
+  ) {
+    this.runTest(rootPath, fileName, testName);
+  }
+
   public runTest(
     rootPath: WorkspaceFolder,
     fileName: string,
@@ -63,18 +78,18 @@ export class PlaywrightTestRunner implements ITestRunnerInterface {
         "-g",
         testName,
         ...additionalArguments.split(" "),
-        this.transformFileName(fileName),
+        this.transformFileName(fileName)
       ],
       console: "integratedTerminal",
       env: {
         ...{ PLAYWRIGHT_CHROMIUM_DEBUG_PORT: 9222, PWDEBUG: true },
-        ...environmentVariables,
+        ...environmentVariables
       },
       name: "Debug Test",
       program: join(rootPath.uri.fsPath, this.path),
       request: "launch",
       skipFiles,
-      type: "node",
+      type: "node"
     });
   }