Configure when the debug CodeLens is displayed

.gitignore

@@ -1,6 +1,8 @@
-out
-node_modules
 _jest-editor/
+.vscode/symbols.json
+coverage/
 integrations/create-react-example/node_modules
-/coverage/
-/.vscode/symbols.json
+node_modules/
+out/
+
+**/.DS_Store

CHANGELOG.md

@@ -5,12 +5,10 @@ Bug-fixes within the same version aren't needed
 
 ## Master
 
-* [your thing] - [you]
+* Adds a setting to control when the debug CodeLens appears - seanpoulter
 
 -->
 
-## Master
-
 ### 2.7.2
 
 * Fix decorators showing corrupt values when two tests of the same name in one file (e.g. in different describe blocks) exist - ThomasRooney

package.json

@@ -94,6 +94,14 @@
           "type": "boolean",
           "default": true
         },
+        "jest.debugCodeLens.showWhenTestStateIn": {
+          "description": "Show the debug CodeLens when the it/test block state is in this collection",
+          "type": "array",
+          "items": {
+            "enum": ["fail", "pass", "skip", "unknown"]
+          },
+          "default": ["fail", "unknown"]
+        },
         "jest.enableSnapshotPreviews": {
           "description": "Whether snapshot previews should show",
           "type": "boolean",

src/DebugCodeLens/DebugCodeLensProvider.ts

@@ -4,24 +4,25 @@ import { escapeRegExp } from '../helpers'
 import { basename } from 'path'
 import { DebugCodeLens } from './DebugCodeLens'
 import { TestReconciliationState, TestResultProvider } from '../TestResults'
+import { TestState, TestStateByTestReconciliationState } from './TestState'
 
 export class DebugCodeLensProvider implements vscode.CodeLensProvider {
-  private _enabled: boolean
+  private _showWhenTestStateIn: TestState[]
   onDidChange: vscode.EventEmitter<void>
   testResultProvider: TestResultProvider
 
-  constructor(testResultProvider: TestResultProvider, enabled: boolean) {
+  constructor(testResultProvider: TestResultProvider, showWhenTestStateIn: TestState[]) {
     this.testResultProvider = testResultProvider
-    this._enabled = enabled
+    this._showWhenTestStateIn = showWhenTestStateIn
     this.onDidChange = new vscode.EventEmitter()
   }
 
-  get enabled() {
-    return this._enabled
+  get showWhenTestStateIn() {
+    return this._showWhenTestStateIn
   }
 
-  set enabled(value: boolean) {
-    this._enabled = value
+  set showWhenTestStateIn(value: TestState[]) {
+    this._showWhenTestStateIn = value
     this.onDidChange.fire()
   }
 
@@ -32,15 +33,16 @@ export class DebugCodeLensProvider implements vscode.CodeLensProvider {
   provideCodeLenses(document: vscode.TextDocument, _: vscode.CancellationToken): vscode.CodeLens[] {
     const result = []
 
-    if (!this._enabled || document.isUntitled) {
+    if (this._showWhenTestStateIn.length === 0 || document.isUntitled) {
       return result
     }
 
     const filePath = document.fileName
     const testResults = this.testResultProvider.getResults(filePath)
     const fileName = basename(document.fileName)
+
     for (const test of testResults) {
-      if (test.status === TestReconciliationState.KnownSuccess || test.status === TestReconciliationState.KnownSkip) {
+      if (!this.showCodeLensAboveTest(test)) {
         continue
       }
 
@@ -53,6 +55,11 @@ export class DebugCodeLensProvider implements vscode.CodeLensProvider {
     return result
   }
 
+  showCodeLensAboveTest(test: { status: TestReconciliationState }) {
+    const state = TestStateByTestReconciliationState[test.status]
+    return this._showWhenTestStateIn.includes(state)
+  }
+
   resolveCodeLens(codeLens: vscode.CodeLens, _: vscode.CancellationToken): vscode.ProviderResult<vscode.CodeLens> {
     if (codeLens instanceof DebugCodeLens) {
       codeLens.command = {

src/DebugCodeLens/TestState.ts

@@ -0,0 +1,15 @@
+import { TestReconciliationState } from '../TestResults'
+
+export enum TestState {
+  Fail = 'fail',
+  Pass = 'pass',
+  Skip = 'skip',
+  Unknown = 'unknown',
+}
+
+export const TestStateByTestReconciliationState = {
+  [TestReconciliationState.KnownFail]: TestState.Fail,
+  [TestReconciliationState.KnownSkip]: TestState.Skip,
+  [TestReconciliationState.KnownSuccess]: TestState.Pass,
+  [TestReconciliationState.Unknown]: TestState.Unknown,
+}

src/DebugCodeLens/index.ts

@@ -1 +1,2 @@
 export { DebugCodeLensProvider } from './DebugCodeLensProvider'
+export { TestState } from './TestState'

src/IPluginSettings.ts

@@ -1,13 +1,18 @@
+import { TestState } from './DebugCodeLens'
+
 export interface IPluginSettings {
   autoEnable?: boolean
-  enableCodeLens?: boolean
+  debugCodeLens: {
+    enabled: boolean
+    showWhenTestStateIn: TestState[]
+  }
   enableInlineErrorMessages?: boolean
   enableSnapshotPreviews?: boolean
   enableSnapshotUpdateMessages?: boolean
-  pathToJest?: string
   pathToConfig?: string
+  pathToJest?: string
+  restartJestOnSnapshotUpdate?: boolean
   rootPath?: string
   runAllTestsFirst?: boolean
   showCoverageOnLoad: boolean
-  restartJestOnSnapshotUpdate?: boolean
 }

src/JestExt.ts

@@ -69,7 +69,10 @@ export class JestExt {
     this.coverageOverlay = new CoverageOverlay(this.coverageMapProvider, pluginSettings.showCoverageOnLoad)
 
     this.testResultProvider = new TestResultProvider()
-    this.debugCodeLensProvider = new DebugCodeLensProvider(this.testResultProvider, pluginSettings.enableCodeLens)
+    this.debugCodeLensProvider = new DebugCodeLensProvider(
+      this.testResultProvider,
+      pluginSettings.debugCodeLens.enabled ? pluginSettings.debugCodeLens.showWhenTestStateIn : []
+    )
     this.debugConfigurationProvider = new DebugConfigurationProvider()
 
     this.jestProcessManager = new JestProcessManager({
@@ -244,7 +247,10 @@ export class JestExt {
     this.jestSettings = new Settings(this.workspace)
 
     this.coverageOverlay.enabled = updatedSettings.showCoverageOnLoad
-    this.debugCodeLensProvider.enabled = updatedSettings.enableCodeLens
+
+    this.debugCodeLensProvider.showWhenTestStateIn = updatedSettings.debugCodeLens.enabled
+      ? updatedSettings.debugCodeLens.showWhenTestStateIn
+      : []
 
     this.stopProcess()
 

src/extension.ts

@@ -9,6 +9,7 @@ import { IPluginSettings } from './IPluginSettings'
 import { registerStatusBar } from './statusBar'
 import { registerSnapshotCodeLens, registerSnapshotPreview } from './SnapshotCodeLens'
 import { registerCoverageCodeLens } from './Coverage'
+import { TestState } from './DebugCodeLens'
 
 let extensionInstance: JestExt
 
@@ -76,19 +77,22 @@ export function deactivate() {
   extensionInstance.deactivate()
 }
 
-function getExtensionSettings(): IPluginSettings {
+export function getExtensionSettings(): IPluginSettings {
   const config = vscode.workspace.getConfiguration('jest')
   return {
     autoEnable: config.get<boolean>('autoEnable'),
-    pathToConfig: config.get<string>('pathToConfig'),
-    pathToJest: config.get<string>('pathToJest'),
-    enableCodeLens: config.get<boolean>('enableCodeLens'),
+    debugCodeLens: {
+      enabled: config.get<boolean>('enableCodeLens'),
+      showWhenTestStateIn: config.get<TestState[]>('debugCodeLens.showWhenTestStateIn'),
+    },
     enableInlineErrorMessages: config.get<boolean>('enableInlineErrorMessages'),
     enableSnapshotPreviews: config.get<boolean>('enableSnapshotPreviews'),
     enableSnapshotUpdateMessages: config.get<boolean>('enableSnapshotUpdateMessages'),
+    pathToConfig: config.get<string>('pathToConfig'),
+    pathToJest: config.get<string>('pathToJest'),
+    restartJestOnSnapshotUpdate: config.get<boolean>('restartJestOnSnapshotUpdate'),
     rootPath: path.join(vscode.workspace.rootPath, config.get<string>('rootPath')),
     runAllTestsFirst: config.get<boolean>('runAllTestsFirst'),
     showCoverageOnLoad: config.get<boolean>('showCoverageOnLoad'),
-    restartJestOnSnapshotUpdate: config.get<boolean>('restartJestOnSnapshotUpdate'),
   }
 }