pi-lens 2.0.7 → 2.0.8

package/clients/typescript-client.test.ts

@@ -1,120 +1,126 @@
-import { describe, it, expect, beforeEach, afterEach } from "vitest";
-import { TypeScriptClient } from "./typescript-client.js";
+import { afterEach, beforeEach, describe, expect, it } from "vitest";
 import { createTempFile, setupTestEnvironment } from "./test-utils.js";
+import { TypeScriptClient } from "./typescript-client.js";
 
 describe("TypeScriptClient", () => {
-  let client: TypeScriptClient;
-  let tmpDir: string;
-  let cleanup: () => void;
-
-  beforeEach(() => {
-    client = new TypeScriptClient();
-    ({ tmpDir, cleanup } = setupTestEnvironment("pi-lens-ts-test-"));
-  });
-
-  afterEach(() => {
-    cleanup();
-  });
-
-  afterEach(() => {
-    cleanup();
-  });
-
-  describe("isTypeScriptFile", () => {
-    it("should recognize TypeScript files", () => {
-      expect(client.isTypeScriptFile("test.ts")).toBe(true);
-      expect(client.isTypeScriptFile("test.tsx")).toBe(true);
-    });
-
-    it("should also recognize JavaScript files (for type checking)", () => {
-      expect(client.isTypeScriptFile("test.js")).toBe(true);
-      expect(client.isTypeScriptFile("test.jsx")).toBe(true);
-    });
-
-    it("should not recognize non-JS/TS files", () => {
-      expect(client.isTypeScriptFile("test.py")).toBe(false);
-      expect(client.isTypeScriptFile("test.md")).toBe(false);
-    });
-  });
-
-  describe("updateFile and getDiagnostics", () => {
-    it("should detect type errors", () => {
-      const content = `
+	let client: TypeScriptClient;
+	let tmpDir: string;
+	let cleanup: () => void;
+
+	beforeEach(() => {
+		client = new TypeScriptClient();
+		({ tmpDir, cleanup } = setupTestEnvironment("pi-lens-ts-test-"));
+	});
+
+	afterEach(() => {
+		cleanup();
+	});
+
+	afterEach(() => {
+		cleanup();
+	});
+
+	describe("isTypeScriptFile", () => {
+		it("should recognize TypeScript files", () => {
+			expect(client.isTypeScriptFile("test.ts")).toBe(true);
+			expect(client.isTypeScriptFile("test.tsx")).toBe(true);
+		});
+
+		it("should also recognize JavaScript files (for type checking)", () => {
+			expect(client.isTypeScriptFile("test.js")).toBe(true);
+			expect(client.isTypeScriptFile("test.jsx")).toBe(true);
+		});
+
+		it("should not recognize non-JS/TS files", () => {
+			expect(client.isTypeScriptFile("test.py")).toBe(false);
+			expect(client.isTypeScriptFile("test.md")).toBe(false);
+		});
+	});
+
+	describe("updateFile and getDiagnostics", () => {
+		it("should detect type errors", () => {
+			const content = `
 const x: number = "string"; // Type error
 `;
-      const filePath = createTempFile(tmpDir, "test.ts", content);
+			const filePath = createTempFile(tmpDir, "test.ts", content);
 
-      client.updateFile(filePath, content);
-      const diags = client.getDiagnostics(filePath);
+			client.updateFile(filePath, content);
+			const diags = client.getDiagnostics(filePath);
 
-      expect(diags.length).toBeGreaterThan(0);
-      expect(diags.some(d => d.message.includes("string"))).toBe(true);
-    });
+			expect(diags.length).toBeGreaterThan(0);
+			expect(diags.some((d) => d.message.includes("string"))).toBe(true);
+		});
 
-    it("should not report errors for valid code", () => {
-      const content = `
+		it("should not report errors for valid code", () => {
+			const content = `
 const x: number = 42;
 const y: string = "hello";
 `;
-      const filePath = createTempFile(tmpDir, "test.ts", content);
+			const filePath = createTempFile(tmpDir, "test.ts", content);
 
-      client.updateFile(filePath, content);
-      const diags = client.getDiagnostics(filePath);
+			client.updateFile(filePath, content);
+			const diags = client.getDiagnostics(filePath);
 
-      // May have warnings but no errors for valid code
-      const errors = diags.filter(d => d.severity === 1); // Error severity
-      expect(errors.length).toBe(0);
-    });
+			// May have warnings but no errors for valid code
+			const errors = diags.filter((d) => d.severity === 1); // Error severity
+			expect(errors.length).toBe(0);
+		});
 
-    it("should detect undefined variables", () => {
-      const content = `
+		it("should detect undefined variables", () => {
+			const content = `
 function test() {
   return undefinedVariable;
 }
 `;
-      const filePath = createTempFile(tmpDir, "test.ts", content);
+			const filePath = createTempFile(tmpDir, "test.ts", content);
 
-      client.updateFile(filePath, content);
-      const diags = client.getDiagnostics(filePath);
+			client.updateFile(filePath, content);
+			const diags = client.getDiagnostics(filePath);
 
-      expect(diags.some(d => d.message.includes("undefined"))).toBe(true);
-    });
+			expect(diags.some((d) => d.message.includes("undefined"))).toBe(true);
+		});
 
-    it("should detect missing function arguments", () => {
-      const content = `
+		it("should detect missing function arguments", () => {
+			const content = `
 function add(a: number, b: number): number {
   return a + b;
 }
 const result = add(1);
 `;
-      const filePath = createTempFile(tmpDir, "test.ts", content);
-
-      client.updateFile(filePath, content);
-      const diags = client.getDiagnostics(filePath);
-
-      expect(diags.some(d => d.message.includes("Expected"))).toBe(true);
-    });
-  });
-
-  describe("diagnostic severity", () => {
-    it("should have correct severity levels", () => {
-      const diags = [
-        {
-          range: { start: { line: 0, character: 0 }, end: { line: 0, character: 10 } },
-          severity: 1, // Error
-          message: "Error message",
-        },
-        {
-          range: { start: { line: 1, character: 0 }, end: { line: 1, character: 10 } },
-          severity: 2, // Warning
-          message: "Warning message",
-        },
-      ];
-
-      // Test that diagnostics have expected structure
-      expect(diags[0].severity).toBe(1); // Error
-      expect(diags[1].severity).toBe(2); // Warning
-      expect(diags[0].message).toContain("Error");
-    });
-  });
+			const filePath = createTempFile(tmpDir, "test.ts", content);
+
+			client.updateFile(filePath, content);
+			const diags = client.getDiagnostics(filePath);
+
+			expect(diags.some((d) => d.message.includes("Expected"))).toBe(true);
+		});
+	});
+
+	describe("diagnostic severity", () => {
+		it("should have correct severity levels", () => {
+			const diags = [
+				{
+					range: {
+						start: { line: 0, character: 0 },
+						end: { line: 0, character: 10 },
+					},
+					severity: 1, // Error
+					message: "Error message",
+				},
+				{
+					range: {
+						start: { line: 1, character: 0 },
+						end: { line: 1, character: 10 },
+					},
+					severity: 2, // Warning
+					message: "Warning message",
+				},
+			];
+
+			// Test that diagnostics have expected structure
+			expect(diags[0].severity).toBe(1); // Error
+			expect(diags[1].severity).toBe(2); // Warning
+			expect(diags[0].message).toContain("Error");
+		});
+	});
 });