From 7c8b5a2a6a0b395338bff3da7c0dfe7405361bf8 Mon Sep 17 00:00:00 2001
From: Peter van Gulik <peter@curlybracket.nl>
Date: Wed, 24 Jan 2024 12:13:08 +0100
Subject: [PATCH] feat: include coverage % in toggle coverage code lems

---
 .../src/codeLensProviders/testCoverageLensProvider.ts    | 9 ++++++++-
 .../src/commands/apex/toggleTestCoverage.ts              | 2 +-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/packages/vscode-extension/src/codeLensProviders/testCoverageLensProvider.ts b/packages/vscode-extension/src/codeLensProviders/testCoverageLensProvider.ts
index 4f787a2e..b5f96e1d 100644
--- 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
+        );
+    }
 }
\ No newline at end of file
diff --git a/packages/vscode-extension/src/commands/apex/toggleTestCoverage.ts b/packages/vscode-extension/src/commands/apex/toggleTestCoverage.ts
index 6ffc8618..8b52b406 100644
--- 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;
         }