Skip to content

Commit a84b704

Browse files
qvalentinqvalentin
committed
refact: clean up test
1 parent 34d50ce commit a84b704

File tree

1 file changed

+120
-81
lines changed

1 file changed

+120
-81
lines changed

‎src/test/extension.test.ts‎

Lines changed: 120 additions & 81 deletions
Original file line numberDiff line numberDiff line change
@@ -1,87 +1,126 @@
11
import * as assert from "assert";
22
import * as path from "path";
3-
4-
// You can import and use all API from the 'vscode' module
5-
// as well as import your extension to test it
63
import * as vscode from "vscode";
7-
// import * as myExtension from '../../extension';
4+
5+
/**
6+
* Helper function to wait for extension activation with timeout
7+
*/
8+
async function waitForExtensionActivation(extensionId: string, timeout: number = 10000): Promise<vscode.Extension<any>> {
9+
const startTime = Date.now();
10+
11+
while (Date.now() - startTime < timeout) {
12+
const ext = vscode.extensions.getExtension(extensionId);
13+
if (ext?.isActive) {
14+
return ext;
15+
}
16+
17+
if (ext && !ext.isActive) {
18+
await ext.activate();
19+
return ext;
20+
}
21+
22+
await new Promise(resolve => setTimeout(resolve, 100));
23+
}
24+
25+
throw new Error(`Extension ${extensionId} failed to activate within ${timeout}ms`);
26+
}
27+
28+
/**
29+
* Helper function to open a Helm template file and wait for language server
30+
*/
31+
async function openHelmDocument(workspaceFolder: vscode.WorkspaceFolder): Promise<{ document: vscode.TextDocument; uri: vscode.Uri }> {
32+
const docPath = path.join(
33+
workspaceFolder.uri.fsPath,
34+
'src', 'test', 'fixtures', 'testChart', 'templates', 'deployment.yaml'
35+
);
36+
const docUri = vscode.Uri.file(docPath);
37+
const document = await vscode.workspace.openTextDocument(docUri);
38+
await vscode.window.showTextDocument(document);
39+
40+
// Give language server a moment to initialize
41+
await new Promise(resolve => setTimeout(resolve, 1000));
42+
43+
return { document, uri: docUri };
44+
}
845

946
suite("Extension Test Suite", () => {
10-
vscode.window.showInformationMessage('Start all tests.');
11-
12-
test('Sample test', () => {
13-
assert.strictEqual(-1, [1, 2, 3].indexOf(5));
14-
assert.strictEqual(-1, [1, 2, 3].indexOf(0));
15-
});
16-
17-
test('Extension should be loaded', async function () {
18-
this.timeout(500000)
19-
const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
20-
assert.ok(workspaceFolder, 'No workspace folder found');
21-
const docPath = path.join(workspaceFolder.uri.fsPath, 'src', 'test', 'fixtures', 'testChart', 'templates', 'deployment.yaml');
22-
const docUri = vscode.Uri.file(docPath);
23-
const document = await vscode.workspace.openTextDocument(docUri);
24-
await vscode.window.showTextDocument(document);
25-
26-
await new Promise(resolve => setTimeout(resolve, 5000));
27-
28-
const ext = vscode.extensions.getExtension('helm-ls.helm-ls');
29-
assert.ok(ext, 'Extension not found in vscode.extensions');
30-
31-
console.log('Extension found:', ext.id);
32-
console.log('Is Active:', ext.isActive);
33-
34-
if (!ext.isActive) {
35-
await ext.activate();
36-
console.log('Extension activated manually.');
37-
}
38-
39-
assert.ok(ext.isActive, 'Extension failed to activate');
40-
});
41-
42-
test('Hover tests', async function () {
43-
this.timeout(500000)
44-
const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
45-
assert.ok(workspaceFolder, 'No workspace folder found');
46-
const docPath = path.join(workspaceFolder.uri.fsPath, 'src', 'test', 'fixtures', 'testChart', 'templates', 'deployment.yaml');
47-
const docUri = vscode.Uri.file(docPath);
48-
const document = await vscode.workspace.openTextDocument(docUri);
49-
await vscode.window.showTextDocument(document);
50-
51-
await new Promise(resolve => setTimeout(resolve, 5000));
52-
53-
console.log("Testing hover")
54-
55-
// Test hover on a Helm property: .Values.replicaCount at line 9
56-
const helmPosition = new vscode.Position(8, 25); // Position inside 'replicaCount'
57-
const helmHovers = await vscode.commands.executeCommand<vscode.Hover[]>(
58-
'vscode.executeHoverProvider',
59-
docUri,
60-
helmPosition
61-
);
62-
63-
assert.strictEqual(helmHovers.length, 1, 'Expected one hover for Helm property');
64-
const helmHoverContent = helmHovers[0].contents[0] as vscode.MarkdownString;
65-
// Assuming replicaCount is 1 in values.yaml
66-
assert.ok(helmHoverContent.value.includes('1'), `Hover content for Helm property should show the value. Got ${helmHoverContent.value}`);
67-
assert.ok(
68-
helmHoverContent.value.includes('values.yaml'),
69-
'Hover content for Helm property should mention the source file.'
70-
);
71-
72-
// Test hover on a YAML property: spec at line 7
73-
const yamlPosition = new vscode.Position(6, 3); // Position inside 'spec'
74-
const yamlHovers = await vscode.commands.executeCommand<vscode.Hover[]>(
75-
'vscode.executeHoverProvider',
76-
docUri,
77-
yamlPosition
78-
);
79-
80-
assert.strictEqual(yamlHovers.length, 1, 'Expected one hover for YAML property');
81-
const yamlHoverContent = yamlHovers[0].contents[0] as vscode.MarkdownString;
82-
assert.ok(
83-
yamlHoverContent.value.includes('Specification of the desired behavior of the Deployment.'),
84-
`Hover content for YAML property should show documentation. Got ${yamlHoverContent.value}`
85-
);
86-
});
47+
vscode.window.showInformationMessage('Starting Helm LS extension tests');
48+
49+
test("Extension loads and activates", async function () {
50+
this.timeout(30000); // 30 seconds should be plenty
51+
52+
const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
53+
assert.ok(workspaceFolder, 'No workspace folder found');
54+
55+
// Open a Helm file to trigger activation
56+
await openHelmDocument(workspaceFolder);
57+
58+
// Wait for extension to activate
59+
const ext = await waitForExtensionActivation('helm-ls.helm-ls');
60+
assert.ok(ext, 'Extension should be activated');
61+
assert.strictEqual(ext.id, 'helm-ls.helm-ls');
62+
});
63+
64+
test("Hover support for Helm templates", async function () {
65+
this.timeout(30000);
66+
67+
const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
68+
assert.ok(workspaceFolder, 'No workspace folder found');
69+
70+
// Ensure extension is activated
71+
await waitForExtensionActivation('helm-ls.helm-ls');
72+
73+
const { uri: docUri } = await openHelmDocument(workspaceFolder);
74+
75+
// Test hover on .Values.replicaCount (line 9, assuming it's around column 25)
76+
const helmPosition = new vscode.Position(8, 25);
77+
const helmHovers = await vscode.commands.executeCommand<vscode.Hover[]>(
78+
'vscode.executeHoverProvider',
79+
docUri,
80+
helmPosition
81+
);
82+
83+
assert.ok(helmHovers && helmHovers.length > 0, 'Should have hover information for Helm property');
84+
85+
const helmHoverContent = helmHovers[0].contents[0] as vscode.MarkdownString;
86+
assert.ok(helmHoverContent, 'Hover content should exist');
87+
88+
// Check for expected content (replicaCount value from values.yaml)
89+
assert.ok(
90+
helmHoverContent.value.includes('1') || helmHoverContent.value.includes('replicaCount'),
91+
`Hover should show replicaCount value or reference. Got: ${helmHoverContent.value}`
92+
);
93+
});
94+
95+
test("Hover support for YAML schema", async function () {
96+
this.timeout(30000);
97+
98+
const workspaceFolder = vscode.workspace.workspaceFolders?.[0];
99+
assert.ok(workspaceFolder, 'No workspace folder found');
100+
101+
// Ensure extension is activated
102+
await waitForExtensionActivation('helm-ls.helm-ls');
103+
104+
const { uri: docUri } = await openHelmDocument(workspaceFolder);
105+
106+
// Test hover on 'spec' property (line 7)
107+
const yamlPosition = new vscode.Position(6, 3);
108+
const yamlHovers = await vscode.commands.executeCommand<vscode.Hover[]>(
109+
'vscode.executeHoverProvider',
110+
docUri,
111+
yamlPosition
112+
);
113+
114+
assert.ok(yamlHovers && yamlHovers.length > 0, 'Should have hover information for YAML property');
115+
116+
const yamlHoverContent = yamlHovers[0].contents[0] as vscode.MarkdownString;
117+
assert.ok(yamlHoverContent, 'YAML hover content should exist');
118+
119+
// Check for Kubernetes documentation
120+
const content = yamlHoverContent.value.toLowerCase();
121+
assert.ok(
122+
content.includes('deployment') || content.includes('spec') || content.includes('specification'),
123+
`Hover should show Kubernetes schema info. Got: ${yamlHoverContent.value}`
124+
);
125+
});
87126
});

0 commit comments

Comments
 (0)