Tests for semantic tokens

Author: diyaayay

language-server/src/features/semantic-tokens.test.ts

@@ -0,0 +1,98 @@
+import { beforeAll, afterAll, describe, expect, test } from "vitest";
+import { ConfigurationRequest, SemanticTokensRequest, DidChangeConfigurationNotification } from "vscode-languageserver";
+import { getTestClient, initializeServer, openDocument, closeDocument } from "../test-utils.js";
+import semanticTokensFeature from "./semantic-tokens.js";
+import workspace from "./workspace.js";
+import documentSettings from "./document-settings.js";
+
+import type { Connection } from "vscode-languageserver";
+
+
+describe("Feature - Semantic Tokens", () => {
+  let client: Connection;
+  let documentUri: string;
+
+  beforeAll(async () => {
+    client = getTestClient([
+      semanticTokensFeature,
+      documentSettings,
+      workspace
+    ]);
+
+    const init = {};
+    const settings = { "schemaFilePatterns": ["**/subject.schema.json"] };
+    await initializeServer(client, init, settings);
+  });
+
+  afterAll(async () => {
+    await closeDocument(client, documentUri);
+  });
+
+  test("semantic tokens on a watched file", async () => {
+    documentUri = await openDocument(client, "subject.schema.json", `{"$schema":"http://json-schema.org/draft-07/schema#",
+"type": "string",
+"minLength": 10,
+"maxLength": 5
+}`);
+
+    const response = await client.sendRequest(SemanticTokensRequest.type, {
+      textDocument: { uri: documentUri }
+    });
+
+    expect(response?.data).to.eql(expectedData);
+  });
+
+  test("no semantic tokens", async () => {
+    documentUri = await openDocument(client, "subject.schema.json", `{
+"type": "string",
+"minLength": 10,
+"maxLength": 5}`);
+
+    const response = await client.sendRequest(SemanticTokensRequest.type, {
+      textDocument: { uri: documentUri }
+    });
+
+    expect(response?.data).to.eql([]);
+  });
+
+  test("no semantic tokens on an unwatched file", async () => {
+    documentUri = await openDocument(client, "subjectB.schema.json", `{"$schema":"http://json-schema.org/draft-07/schema#",
+      "type": "string",
+      "minLength": 10,
+      "maxLength": 5
+      }`);
+
+    const response = await client.sendRequest(SemanticTokensRequest.type, {
+      textDocument: { uri: documentUri }
+    });
+
+    expect(response?.data).to.eql([]);
+  });
+
+  test("change in watch file patterns refreshes tokens", async () => {
+    documentUri = await openDocument(client, "subjectC.schema.json", `{"$schema":"http://json-schema.org/draft-07/schema#",
+"type": "string",
+"minLength": 10,
+"maxLength": 5
+}`);
+
+    client.onRequest(ConfigurationRequest.type, () => {
+      return [{ "schemaFilePatterns": ["**/subjectC.schema.json"] }];
+    });
+
+    await client.sendNotification(DidChangeConfigurationNotification.type, { settings: { "schemaFilePatterns": ["**/subjectC.schema.json"] } });
+    const response = await client.sendRequest(SemanticTokensRequest.type, {
+      textDocument: { uri: documentUri }
+    });
+
+    expect(response?.data).to.eql(expectedData);
+  });
+});
+
+const expectedData = [
+  0, 1, 9, 0, 0, 1, 0,
+  6, 0, 0, 1, 0, 11, 0,
+  0, 1, 0, 11, 0, 0
+];
+
+