feat(acton): add coverage support for tests (#240)

i582 · web-flow · commit 81f3c71c5748 · 2026-01-03T15:29:29.000+04:00
diff --git a/editors/code/src/acton/ActonCommand.ts b/editors/code/src/acton/ActonCommand.ts
@@ -41,13 +41,22 @@ export class TestCommand extends ActonCommand {
         public target: string = "",
         public functionName: string = "",
         public clearCache: boolean = false,
+        public coverage: boolean = false,
+        public coverageFormat: string = "lcov",
+        public coverageFile: string = "",
     ) {
         super("test")
     }
 
     public override getArguments(): string[] {
         const args: string[] = ["--reporter", "console"]
         if (this.clearCache) args.push("--clear-cache")
+        if (this.coverage) {
+            args.push("--coverage", "--coverage-format", this.coverageFormat)
+            if (this.coverageFile.trim() !== "") {
+                args.push("--coverage-file", this.coverageFile)
+            }
+        }
 
         switch (this.mode) {
             case TestMode.FUNCTION: {
diff --git a/editors/code/src/acton/ActonTestController.ts b/editors/code/src/acton/ActonTestController.ts
@@ -1,16 +1,20 @@
 //  SPDX-License-Identifier: MIT
 //  Copyright © 2026 TON Core
 
+import * as fs from "node:fs/promises"
 import * as path from "node:path"
 
 import * as vscode from "vscode"
 
+import {FileCoverageDetail} from "vscode"
+
 import {Acton} from "./Acton"
 import {TestCommand, TestMode} from "./ActonCommand"
 
 export class ActonTestController implements Disposable {
     private readonly controller: vscode.TestController
     private readonly outputChannel: vscode.OutputChannel
+    private readonly coverageDetails: Map<string, vscode.StatementCoverage[]> = new Map()
 
     public constructor() {
         this.controller = vscode.tests.createTestController("actonTests", "Acton Tests")
@@ -25,6 +29,22 @@ export class ActonTestController implements Disposable {
             true,
         )
 
+        const coverageProfile = this.controller.createRunProfile(
+            "Coverage",
+            vscode.TestRunProfileKind.Coverage,
+            async (request, token) => {
+                await this.runHandler(request, token, true)
+            },
+            true,
+        )
+
+        coverageProfile.loadDetailedCoverage = (
+            _testRun,
+            fileCoverage,
+        ): Thenable<FileCoverageDetail[]> => {
+            return Promise.resolve(this.coverageDetails.get(fileCoverage.uri.toString()) ?? [])
+        }
+
         this.controller.resolveHandler = async item => {
             await (item ? this.discoverTestsInItem(item) : this.discoverAllTests())
         }
@@ -135,7 +155,11 @@ export class ActonTestController implements Disposable {
     private async runHandler(
         request: vscode.TestRunRequest,
         token: vscode.CancellationToken,
+        isCoverage: boolean = false,
     ): Promise<void> {
+        if (isCoverage) {
+            this.coverageDetails.clear()
+        }
         const run = this.controller.createTestRun(request)
         const queue: vscode.TestItem[] = []
 
@@ -154,7 +178,7 @@ export class ActonTestController implements Disposable {
                 break
             }
 
-            await this.runTestItem(test, run, token)
+            await this.runTestItem(test, run, token, isCoverage)
         }
 
         run.end()
@@ -164,6 +188,7 @@ export class ActonTestController implements Disposable {
         item: vscode.TestItem,
         run: vscode.TestRun,
         token: vscode.CancellationToken,
+        isCoverage: boolean = false,
     ): Promise<void> {
         if (token.isCancellationRequested) {
             return
@@ -181,10 +206,27 @@ export class ActonTestController implements Disposable {
         const workingDir = tomlUri ? path.dirname(tomlUri.fsPath) : path.dirname(uri.fsPath)
         const relativePath = path.relative(workingDir, uri.fsPath)
 
+        const coverageFile = isCoverage ? path.join(workingDir, "lcov.info") : ""
         const command =
             item.children.size > 0
-                ? new TestCommand(TestMode.FILE, relativePath)
-                : new TestCommand(TestMode.FUNCTION, relativePath, item.label)
+                ? new TestCommand(
+                      TestMode.FILE,
+                      relativePath,
+                      "",
+                      false,
+                      isCoverage,
+                      "lcov",
+                      coverageFile,
+                  )
+                : new TestCommand(
+                      TestMode.FUNCTION,
+                      relativePath,
+                      item.label,
+                      false,
+                      isCoverage,
+                      "lcov",
+                      coverageFile,
+                  )
 
         try {
             const {exitCode, stdout, stderr} = await Acton.getInstance().spawn(
@@ -199,6 +241,9 @@ export class ActonTestController implements Disposable {
 
             if (exitCode === 0) {
                 run.passed(item)
+                if (isCoverage && coverageFile) {
+                    await this.processCoverage(coverageFile, run, workingDir)
+                }
             } else {
                 const rawMessage = stderr || stdout || "Test failed"
 
@@ -220,6 +265,69 @@ export class ActonTestController implements Disposable {
         }
     }
 
+    private async processCoverage(
+        coverageFile: string,
+        run: vscode.TestRun,
+        workingDir: string,
+    ): Promise<void> {
+        try {
+            const content = await fs.readFile(coverageFile, "utf8")
+            const files = this.parseLcov(content)
+
+            for (const fileData of files) {
+                const absolutePath = path.isAbsolute(fileData.file)
+                    ? fileData.file
+                    : path.join(workingDir, fileData.file)
+                const uri = vscode.Uri.file(absolutePath)
+
+                const statementCoverage: vscode.StatementCoverage[] = fileData.lines.map(line => {
+                    return new vscode.StatementCoverage(
+                        line.count,
+                        new vscode.Range(line.line - 1, 0, line.line - 1, 0),
+                    )
+                })
+
+                this.coverageDetails.set(uri.toString(), statementCoverage)
+
+                const coveredLines = fileData.lines.filter(l => l.count > 0).length
+                const totalLines = fileData.lines.length
+
+                const fileCoverage = new vscode.FileCoverage(
+                    uri,
+                    new vscode.TestCoverageCount(coveredLines, totalLines),
+                )
+
+                run.addCoverage(fileCoverage)
+            }
+        } catch (error) {
+            console.error("Failed to process coverage", error)
+        }
+    }
+
+    private parseLcov(content: string): {file: string; lines: {line: number; count: number}[]}[] {
+        const files: {file: string; lines: {line: number; count: number}[]}[] = []
+        let currentFile: {file: string; lines: {line: number; count: number}[]} | null = null
+
+        for (const line of content.split(/\r?\n/)) {
+            if (line.startsWith("SF:")) {
+                currentFile = {file: line.slice(3).trim(), lines: []}
+            } else if (line.startsWith("DA:") && currentFile) {
+                const parts = line.slice(3).split(",")
+                if (parts.length >= 2) {
+                    const lineNum = Number.parseInt(parts[0], 10)
+                    const count = Number.parseInt(parts[1], 10)
+                    if (!Number.isNaN(lineNum) && !Number.isNaN(count)) {
+                        currentFile.lines.push({line: lineNum, count})
+                    }
+                }
+            } else if (line.startsWith("end_of_record") && currentFile) {
+                files.push(currentFile)
+                currentFile = null
+            }
+        }
+        return files
+    }
+
     private extractErrorLocation(cleanMessage: string): vscode.Location | undefined {
         const locationMatch = /at\s+(.+):(\d+):(\d+)/.exec(cleanMessage)
         if (!locationMatch) return undefined
diff --git a/package.json b/package.json
@@ -14,7 +14,7 @@
     ],
     "version": "0.5.1",
     "engines": {
-        "vscode": "^1.63.0"
+        "vscode": "^1.88.0"
     },
     "icon": "editors/code/src/assets/logo.png",
     "activationEvents": [
@@ -833,7 +833,7 @@
     "dependencies": {
         "@ton/core": "0.61.0",
         "@ton/crypto": "^3.3.0",
-        "@vscode/debugadapter": "^1.51.0",
+        "@vscode/debugadapter": "^1.68.0",
         "@vscode/debugprotocol": "^1.68.0",
         "buffer": "^6.0.3",
         "glob": "^11.1.0",
@@ -863,7 +863,7 @@
         "@types/node": "^22.2.0",
         "@types/react": "^18.3.3",
         "@types/react-dom": "^18.3.0",
-        "@types/vscode": "^1.63.0",
+        "@types/vscode": "^1.88.0",
         "@types/ws": "^8",
         "@vscode/test-cli": "^0.0.12",
         "@vscode/test-electron": "^2.4.1",
diff --git a/server/src/languages/tolk/inspections/index.ts b/server/src/languages/tolk/inspections/index.ts
@@ -8,7 +8,6 @@ import {UnusedVariableInspection} from "@server/languages/tolk/inspections/Unuse
 import {DeprecatedSymbolUsageInspection} from "@server/languages/tolk/inspections/DeprecatedSymbolUsageInspection"
 import {UnusedImportInspection} from "@server/languages/tolk/inspections/UnusedImportInspection"
 import {StructInitializationInspection} from "@server/languages/tolk/inspections/StructInitializationInspection"
-import {TypeCompatibilityInspection} from "@server/languages/tolk/inspections/TypeCompatibilityInspection"
 import {CannotReassignInspection} from "@server/languages/tolk/inspections/CannotReassignInspection"
 import {UnusedTopLevelDeclarationInspection} from "@server/languages/tolk/inspections/UnusedTopLevelDeclarationInspection"
 import {UnusedTypeParameterInspection} from "@server/languages/tolk/inspections/UnusedTypeParameterInspection"
@@ -29,7 +28,7 @@ export async function runTolkInspections(
         new UnusedTopLevelDeclarationInspection(),
         new DeprecatedSymbolUsageInspection(),
         new StructInitializationInspection(),
-        new TypeCompatibilityInspection(),
+        // new TypeCompatibilityInspection(),
         new CannotReassignInspection(),
         new NeedNotNullUnwrappingInspection(),
         new MissedSemicolonInspection(),
diff --git a/yarn.lock b/yarn.lock
@@ -2676,10 +2676,10 @@ __metadata:
   languageName: node
   linkType: hard
 
-"@types/vscode@npm:^1.63.0":
-  version: 1.104.0
-  resolution: "@types/vscode@npm:1.104.0"
-  checksum: 10c0/6d9dd470a8d430c65ce5069bc504b9b3f500fcfe13dc7e5df23994f70ea569707ea005936cf706d325b3abf28bfdc402eb8b8449c548b7428f1a6ebfe0d264b9
+"@types/vscode@npm:^1.88.0":
+  version: 1.107.0
+  resolution: "@types/vscode@npm:1.107.0"
+  checksum: 10c0/91dcefaa646bb586189f80f64e2d63f3d2189ff521ce44872652ffbb80d4420fede685a6dbd1fa274fb5ef54501bff23e66ee5f15df9205b12c5bd6d8429988f
   languageName: node
   linkType: hard
 
@@ -2998,7 +2998,7 @@ __metadata:
   languageName: node
   linkType: hard
 
-"@vscode/debugadapter@npm:^1.51.0":
+"@vscode/debugadapter@npm:^1.68.0":
   version: 1.68.0
   resolution: "@vscode/debugadapter@npm:1.68.0"
   dependencies:
@@ -12304,9 +12304,9 @@ __metadata:
     "@types/node": "npm:^22.2.0"
     "@types/react": "npm:^18.3.3"
     "@types/react-dom": "npm:^18.3.0"
-    "@types/vscode": "npm:^1.63.0"
+    "@types/vscode": "npm:^1.88.0"
     "@types/ws": "npm:^8"
-    "@vscode/debugadapter": "npm:^1.51.0"
+    "@vscode/debugadapter": "npm:^1.68.0"
     "@vscode/debugprotocol": "npm:^1.68.0"
     "@vscode/test-cli": "npm:^0.0.12"
     "@vscode/test-electron": "npm:^2.4.1"
