Add more invalid identifier tests

jehna · Oct 6, 2024 · e31ab3a · e31ab3a
1 parent 4eca63a
commit e31ab3a
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/src/plugins/local-llm-rename/visit-all-identifiers.test.ts b/src/plugins/local-llm-rename/visit-all-identifiers.test.ts
@@ -213,3 +213,24 @@ test("should handle invalid identifiers", async () => {
   const result = await visitAllIdentifiers(code, async () => "this.kLength");
   assert.equal(result, "const thisKLength = 1;");
 });
+
+test("should handle reserved identifiers", async () => {
+  const code = `const a = 1`;
+  const result = await visitAllIdentifiers(code, async () => "static");
+  assert.equal(result, "const _static = 1;");
+});
+
+test("should handle multiple identifiers named the same", async () => {
+  const code = `
+const a = 1;
+const b = 1;
+`.trim();
+  const result = await visitAllIdentifiers(code, async () => "foo");
+  assert.equal(
+    result,
+    `
+const foo = 1;
+const _foo = 1;
+`.trim()
+  );
+});