From faf891a00ba4db63ad65d141d4c5ac8655136535 Mon Sep 17 00:00:00 2001
From: Gareth Jones <3151613+G-Rath@users.noreply.github.com>
Date: Wed, 11 Jun 2025 08:06:48 +1200
Subject: [PATCH] fix(prefer-importing-jest-globals): preserve `as` imports

---
 .../prefer-importing-jest-globals.test.ts     | 54 +++++++++++++++++++
 src/rules/prefer-importing-jest-globals.ts    |  9 +++-
 2 files changed, 62 insertions(+), 1 deletion(-)

diff --git a/src/rules/__tests__/prefer-importing-jest-globals.test.ts b/src/rules/__tests__/prefer-importing-jest-globals.test.ts
index 33b331793..e71ead351 100644
--- a/src/rules/__tests__/prefer-importing-jest-globals.test.ts
+++ b/src/rules/__tests__/prefer-importing-jest-globals.test.ts
@@ -104,6 +104,60 @@ ruleTester.run('prefer-importing-jest-globals', rule, {
         },
       ],
     },
+    {
+      code: dedent`
+        import { describe as context } from '@jest/globals';
+        context("suite", () => {
+          test("foo");
+          expect(true).toBeDefined();
+        })
+      `,
+      output: dedent`
+        import { describe as context, expect, test } from '@jest/globals';
+        context("suite", () => {
+          test("foo");
+          expect(true).toBeDefined();
+        })
+      `,
+      parserOptions: { sourceType: 'module' },
+      errors: [
+        {
+          endColumn: 7,
+          column: 3,
+          line: 3,
+          messageId: 'preferImportingJestGlobal',
+        },
+      ],
+    },
+    {
+      code: dedent`
+        import { describe as context } from '@jest/globals';
+        describe("something", () => {
+          context("suite", () => {
+            test("foo");
+            expect(true).toBeDefined();
+          })
+        })
+      `,
+      output: dedent`
+        import { describe, describe as context, expect, test } from '@jest/globals';
+        describe("something", () => {
+          context("suite", () => {
+            test("foo");
+            expect(true).toBeDefined();
+          })
+        })
+      `,
+      parserOptions: { sourceType: 'module' },
+      errors: [
+        {
+          endColumn: 9,
+          column: 1,
+          line: 2,
+          messageId: 'preferImportingJestGlobal',
+        },
+      ],
+    },
     {
       code: dedent`
         jest.useFakeTimers();
diff --git a/src/rules/prefer-importing-jest-globals.ts b/src/rules/prefer-importing-jest-globals.ts
index 66afbd3f0..612640ede 100644
--- a/src/rules/prefer-importing-jest-globals.ts
+++ b/src/rules/prefer-importing-jest-globals.ts
@@ -120,7 +120,14 @@ export default createRule({
                   specifier.type === AST_NODE_TYPES.ImportSpecifier &&
                   specifier.imported?.name
                 ) {
-                  functionsToImport.add(specifier.imported.name);
+                  let importName = specifier.imported.name;
+                  const local = getAccessorValue(specifier.local);
+
+                  if (local !== importName) {
+                    importName = `${importName} as ${local}`;
+                  }
+
+                  functionsToImport.add(importName);
                 }
 
                 if (specifier.type === AST_NODE_TYPES.ImportDefaultSpecifier) {