feat: include coverage % in toggle coverage code lems

Codeneos · Jan 24, 2024 · 7c8b5a2 · 7c8b5a2
1 parent 1615815
commit 7c8b5a2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/packages/vscode-extension/src/codeLensProviders/testCoverageLensProvider.ts b/packages/vscode-extension/src/codeLensProviders/testCoverageLensProvider.ts
@@ -70,9 +70,16 @@ export class TestCoverageLensProvider implements vscode.CodeLensProvider<TestCov
 class TestCoverageCodeLens extends vscode.CodeLens {
     constructor(range: vscode.Range, public readonly className: string, coverageDetails: ApexTestCoverage) {
         super(range, {
-            title: 'Test Coverage',
+            title: `Test Coverage (${TestCoverageCodeLens.getCoveragePercentage(coverageDetails)}%)`,
             command: VlocodeCommand.apexToggleCoverage,
             arguments: [ coverageDetails ]
         });
     }
+
+    static getCoveragePercentage(coverageDetails: ApexTestCoverage) {
+        return Math.round(
+            coverageDetails.coveredLines.length /
+            (coverageDetails.coveredLines.length + coverageDetails.uncoveredLines.length) * 100
+        );
+    }
 }
diff --git a/packages/vscode-extension/src/commands/apex/toggleTestCoverage.ts b/packages/vscode-extension/src/commands/apex/toggleTestCoverage.ts
@@ -89,7 +89,7 @@ export default class ToggleApexTestCoverage extends CommandBase {
         }
 
         const coverageDetails = await this.getCoverage(apexClassName);
-        if (!coverageDetails) {
+        if (!coverageDetails || !coverageDetails.coveredLines.length && !coverageDetails.uncoveredLines.length) {
             void vscode.window.showWarningMessage(`No test coverage collected for ${apexClassName}.`);
             return;
         }