From b710e93fca79a4d77579b3ef014c53463d1e8523 Mon Sep 17 00:00:00 2001
From: waleedcodes <110061477+waleedcodes@users.noreply.github.com>
Date: Fri, 9 Oct 2026 21:26:00 +0500
Subject: [PATCH 1/3] test(codelens): scaffold codelens provider test suite

---
 tests/codeLensProvider.test.ts | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 tests/codeLensProvider.test.ts

diff --git a/tests/codeLensProvider.test.ts b/tests/codeLensProvider.test.ts
new file mode 100644
index 0000000..2e09537
--- /dev/null
+++ b/tests/codeLensProvider.test.ts
@@ -0,0 +1,30 @@
+import { describe, expect, it } from 'vitest';
+import * as vscode from 'vscode';
+import { ProofCodeCodeLensProvider } from '../src/ui/codeLensProvider';
+import { VerificationReport } from '../src/types';
+
+describe('ProofCodeCodeLensProvider', () => {
+  it('should return empty array when no report is set', () => {
+    const provider = new ProofCodeCodeLensProvider();
+    const doc: any = {
+      fileName: '/workspace/src/auth.ts',
+      lineCount: 10,
+      lineAt: () => ({ text: 'export function getUser() {}' })
+    };
+
+    const lenses = provider.provideCodeLenses(doc, {} as any);
+    expect(lenses).toEqual([]);
+  });
+
+  it('should provide CodeLenses for changed functions with caller info and risks', () => {
+    const provider = new ProofCodeCodeLensProvider();
+    const mockReport: Partial<VerificationReport> = {
+      summary: {
+        filesChanged: 1,
+        linesAdded: 10,
+        linesDeleted: 2,
+        symbolsAffected: 1,
+        apiRoutesAffected: 0,
+        testsAdded: 0,
+        testsAffected: 0,
+        highRiskCount: 1,

From 7b3db7bd71ac18c2eca5896493166953cb3339b2 Mon Sep 17 00:00:00 2001
From: waleedcodes <110061477+waleedcodes@users.noreply.github.com>
Date: Fri, 9 Oct 2026 21:26:00 +0500
Subject: [PATCH 2/3] test(codelens): test empty return when no report exists

---
 tests/codeLensProvider.test.ts | 36 ++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/tests/codeLensProvider.test.ts b/tests/codeLensProvider.test.ts
index 2e09537..a76b717 100644
--- a/tests/codeLensProvider.test.ts
+++ b/tests/codeLensProvider.test.ts
@@ -28,3 +28,39 @@ describe('ProofCodeCodeLensProvider', () => {
         testsAdded: 0,
         testsAffected: 0,
         highRiskCount: 1,
+        mediumRiskCount: 0,
+        lowRiskCount: 0
+      },
+      impact: {
+        changedFiles: ['/workspace/src/auth.ts'],
+        changedSymbols: [
+          {
+            symbolName: 'getUser',
+            kind: 'function',
+            filePath: '/workspace/src/auth.ts',
+            callers: [
+              { callerName: 'ProfilePage', filePath: '/src/Profile.tsx', line: 10, hasTest: false },
+              { callerName: 'Dashboard', filePath: '/src/Dashboard.tsx', line: 12, hasTest: true }
+            ],
+            isApiRoute: false
+          }
+        ],
+        affectedFiles: ['/src/Profile.tsx', '/src/Dashboard.tsx'],
+        affectedCallers: [],
+        affectedRoutes: [],
+        totalCallersCount: 2,
+        untestedCallersCount: 1
+      },
+      risks: [
+        {
+          id: 'risk-1',
+          ruleId: 'PC-AUTH-001',
+          category: 'authorization',
+          title: 'Missing Auth',
+          severity: 'HIGH',
+          description: 'No auth',
+          file: '/workspace/src/auth.ts',
+          line: 3,
+          snippet: 'const user = db.user()',
+          evidenceTrace: [],
+          recommendation: 'Fix it'

From 07b43665c9e6842e9bbf8899ae9489e8d80f88d1 Mon Sep 17 00:00:00 2001
From: waleedcodes <110061477+waleedcodes@users.noreply.github.com>
Date: Fri, 9 Oct 2026 21:26:00 +0500
Subject: [PATCH 3/3] test(codelens): test lens title generation with callers
 and risk tags

---
 tests/codeLensProvider.test.ts | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/tests/codeLensProvider.test.ts b/tests/codeLensProvider.test.ts
index a76b717..74d56ff 100644
--- a/tests/codeLensProvider.test.ts
+++ b/tests/codeLensProvider.test.ts
@@ -64,3 +64,25 @@ describe('ProofCodeCodeLensProvider', () => {
           snippet: 'const user = db.user()',
           evidenceTrace: [],
           recommendation: 'Fix it'
+        }
+      ]
+    };
+
+    provider.setReport(mockReport as VerificationReport);
+
+    const doc: any = {
+      fileName: '/workspace/src/auth.ts',
+      lineCount: 5,
+      lineAt: (i: number) => {
+        if (i === 1) return { text: 'export function getUser() {' };
+        return { text: '}' };
+      }
+    };
+
+    const lenses = provider.provideCodeLenses(doc, {} as any);
+    expect(lenses.length).toBe(1);
+    expect(lenses[0].command?.title).toContain('2 callers (1 untested)');
+    expect(lenses[0].command?.title).toContain('1 risk');
+    expect(lenses[0].command?.command).toBe('proofcode.openDashboard');
+  });
+});