Generate unit or E2E test files for existing code
/add-test [type] [target]
type — unit (default) or e2etarget — File path or feature name to testCreates src/path/__tests__/file.test.ts:
import * as assert from 'assert';
import { functionToTest } from '../file.js';
suite('FeatureName Test Suite', () => {
suite('functionName', () => {
test('should handle normal input', () => {
const result = functionToTest('input');
assert.strictEqual(result, 'expected');
});
test('should handle edge case', () => {
const result = functionToTest('');
assert.strictEqual(result, undefined);
});
test('should throw on invalid input', () => {
assert.throws(() => functionToTest(null), /error message/);
});
});
suite('async function', () => {
test('should resolve with data', async () => {
const result = await asyncFunction();
assert.deepStrictEqual(result, { key: 'value' });
});
});
});
When mocking is needed, use sinon:
import * as sinon from 'sinon';
let sandbox: sinon.SinonSandbox;
setup(() => {
sandbox = sinon.createSandbox();
});
teardown(() => {
sandbox.restore();
});
Creates tests/e2e/specs/feature.test.ts:
import { test as base, createTmpDir, expect, GitFixture, MaxTimeout } from '../baseTest.js';
const test = base.extend({
vscodeOptions: [
{
vscodeVersion: process.env.VSCODE_VERSION ?? 'stable',
setup: async () => {
const repoDir = await createTmpDir();
const git = new GitFixture(repoDir);
await git.init();
await git.commit('Initial commit', 'README.md', '# Test');
return repoDir;
},
},
{ scope: 'worker' },
],
});
test.describe('Feature Name', () => {
test.describe.configure({ mode: 'serial' });
test.afterEach(async ({ vscode }) => {
await vscode.gitlens.resetUI();
});
test('should display feature correctly', async ({ vscode }) => {
await vscode.gitlens.openGitLensSidebar();
await expect(vscode.page.getByRole('heading')).toContainText('Expected');
});
});
__tests__/ directory if neededassert.strictEqual(), assert.deepStrictEqual(), assert.ok(), assert.throws()Use the MCP server to explore, then write the test. Don't guess at selectors — verify them live.
/inspect-live to launch VS Code and discover the right selectors:
launch {}
execute_command { command: "gitlens.showHomeView" }
aria_snapshot {} # See all UI elements and roles
inspect_dom { selector: "h1", in_webview: true } # Find webview content
screenshot {} # Visual verification
inspect_dom to verify element text/visibilityevaluate to check extension runtime statescreenshot to visually confirm UI stateexpect(locator).toBeVisible(), .toContainText(), .toHaveCount()Use getGitLensWebview(title, purpose) to get a FrameLocator for webview content:
const webview = await vscode.gitlens.getGitLensWebview('Home', 'webviewView');
await expect(webview!.locator('h1')).toContainText('Expected heading');
await expect(webview!.getByRole('button', { name: /Try Pro/i })).toBeVisible();
Available webviews: Home, Graph, Graph Details, Inspect, Visual File History, Interactive Rebase.
Purpose is webviewView (sidebar/panel) or webviewPanel (editor tab) or customEditor.
// Simulate Pro subscription for the test
using _ = await vscode.gitlens.startSubscriptionSimulation({
state: 6 /* SubscriptionState.Paid */,
planId: 'pro',
});
// Pro features now accessible — auto-reverts when scope exits
await git.init()
await git.commit(message, fileName, content)
await git.branch(name)
await git.checkout(name, create?)
await git.tag(name, { message?, ref? })
await git.stash(message?)
await git.worktree(path, branch)
await git.addRemote(name, url)
await git.merge(branch, message?)
pnpm run test -- --grep "FeatureName" # Unit
pnpm run test:e2e -- tests/e2e/specs/file.test.ts # E2E
For detailed test running patterns, output interpretation, and debugging: see docs/testing.md.