add tests

Author: apatoLZ

packages/react-router-devtools/src/vite/utils/data-functions-augment.test.ts

@@ -722,6 +722,99 @@ describe("transform", () => {
 	})
 })
 
+describe("aliased import transforms", () => {
+	it("should transform aliased loader import where local name is loader", () => {
+		const result = augmentDataFetchingFunctions(
+			`
+			import { partnerCategoryLoader as loader } from "./loaders";
+			export { loader };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withLoaderWrapper as _withLoaderWrapper } from "react-router-devtools/server";
+			import { partnerCategoryLoader as _loader } from "./loaders";
+			export const loader = _withLoaderWrapper(_loader, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased action import where local name is action", () => {
+		const result = augmentDataFetchingFunctions(
+			`
+			import { partnerCategoryAction as action } from "./actions";
+			export { action };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withActionWrapper as _withActionWrapper } from "react-router-devtools/server";
+			import { partnerCategoryAction as _action } from "./actions";
+			export const action = _withActionWrapper(_action, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased clientLoader import where local name is clientLoader", () => {
+		const result = augmentDataFetchingFunctions(
+			`
+			import { partnerCategoryClientLoader as clientLoader } from "./loaders";
+			export { clientLoader };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withClientLoaderWrapper as _withClientLoaderWrapper } from "react-router-devtools/client";
+			import { partnerCategoryClientLoader as _clientLoader } from "./loaders";
+			export const clientLoader = _withClientLoaderWrapper(_clientLoader, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased clientAction import where local name is clientAction", () => {
+		const result = augmentDataFetchingFunctions(
+			`
+			import { partnerCategoryClientAction as clientAction } from "./actions";
+			export { clientAction };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withClientActionWrapper as _withClientActionWrapper } from "react-router-devtools/client";
+			import { partnerCategoryClientAction as _clientAction } from "./actions";
+			export const clientAction = _withClientActionWrapper(_clientAction, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased loader import and update usages in code", () => {
+		const result = augmentDataFetchingFunctions(
+			`
+			import { partnerCategoryLoader as loader } from "./loaders";
+			const test = () => {
+				return loader();
+			}
+			export { loader };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withLoaderWrapper as _withLoaderWrapper } from "react-router-devtools/server";
+			import { partnerCategoryLoader as _loader } from "./loaders";
+			const test = () => {
+				return _loader();
+			};
+			export const loader = _withLoaderWrapper(_loader, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+})
+
 it("should transform the re-exports when it's re-exported from another file with multiple re-exports", () => {
 	const result = augmentDataFetchingFunctions(
 		`

packages/react-router-devtools/src/vite/utils/inject-context.test.ts

@@ -622,6 +622,99 @@ describe("transform", () => {
 	})
 })
 
+describe("aliased import transforms", () => {
+	it("should transform aliased loader import where local name is loader", () => {
+		const result = injectContext(
+			`
+			import { partnerCategoryLoader as loader } from "./loaders";
+			export { loader };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withLoaderContextWrapper as _withLoaderContextWrapper } from "react-router-devtools/context";
+			import { partnerCategoryLoader as _loader } from "./loaders";
+			export const loader = _withLoaderContextWrapper(_loader, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased action import where local name is action", () => {
+		const result = injectContext(
+			`
+			import { partnerCategoryAction as action } from "./actions";
+			export { action };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withActionContextWrapper as _withActionContextWrapper } from "react-router-devtools/context";
+			import { partnerCategoryAction as _action } from "./actions";
+			export const action = _withActionContextWrapper(_action, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased clientLoader import where local name is clientLoader", () => {
+		const result = injectContext(
+			`
+			import { partnerCategoryClientLoader as clientLoader } from "./loaders";
+			export { clientLoader };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withClientLoaderContextWrapper as _withClientLoaderContextWrapper } from "react-router-devtools/context";
+			import { partnerCategoryClientLoader as _clientLoader } from "./loaders";
+			export const clientLoader = _withClientLoaderContextWrapper(_clientLoader, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased clientAction import where local name is clientAction", () => {
+		const result = injectContext(
+			`
+			import { partnerCategoryClientAction as clientAction } from "./actions";
+			export { clientAction };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withClientActionContextWrapper as _withClientActionContextWrapper } from "react-router-devtools/context";
+			import { partnerCategoryClientAction as _clientAction } from "./actions";
+			export const clientAction = _withClientActionContextWrapper(_clientAction, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+
+	it("should transform aliased loader import and update usages in code", () => {
+		const result = injectContext(
+			`
+			import { partnerCategoryLoader as loader } from "./loaders";
+			const test = () => {
+				return loader();
+			}
+			export { loader };
+			`,
+			"test",
+			"/file/path"
+		)
+		const expected = removeWhitespace(`
+			import { withLoaderContextWrapper as _withLoaderContextWrapper } from "react-router-devtools/context";
+			import { partnerCategoryLoader as _loader } from "./loaders";
+			const test = () => {
+				return _loader();
+			};
+			export const loader = _withLoaderContextWrapper(_loader, "test");
+		`)
+		expect(removeWhitespace(result.code)).toStrictEqual(expected)
+	})
+})
+
 it("should transform the re-exports when it's re-exported from another file with multiple re-exports", () => {
 	const result = injectContext(
 		`

packages/react-router-devtools/src/vite/utils/middleware-augment.test.ts

@@ -630,4 +630,71 @@ describe("middleware augmentation", () => {
 			expect(result.code).toContain('"authMiddleware"')
 		})
 	})
+
+	describe("aliased import transforms", () => {
+		it("should transform aliased middleware import where local name is middleware", () => {
+			const result = augmentMiddlewareFunctions(
+				`
+				import { authMiddleware as middleware } from "./middlewares";
+				export { middleware };
+				`,
+				"test",
+				"/file/path"
+			)
+			// Should skip transformation for re-exported imported middleware
+			// The aliased import handler renames it but since it's imported and re-exported, it skips
+			expect(result.code).toContain("authMiddleware")
+		})
+
+		it("should transform aliased clientMiddleware import where local name is clientMiddleware", () => {
+			const result = augmentMiddlewareFunctions(
+				`
+				import { clientAuthMiddleware as clientMiddleware } from "./middlewares";
+				export { clientMiddleware };
+				`,
+				"test",
+				"/file/path"
+			)
+			// Should skip transformation for re-exported imported middleware
+			expect(result.code).toContain("clientAuthMiddleware")
+		})
+
+		it("should handle aliased middleware import used in array", () => {
+			const result = augmentMiddlewareFunctions(
+				`
+				import { someAuthFn as authMiddleware } from "./auth";
+				export const middleware = [authMiddleware];
+				`,
+				"test",
+				"/file/path"
+			)
+			const expected = removeWhitespace(`
+				import { withMiddlewareWrapperSingle as _withMiddlewareWrapperSingle } from "react-router-devtools/server";
+				import { someAuthFn as authMiddleware } from "./auth";
+				export const middleware = [
+					_withMiddlewareWrapperSingle(authMiddleware, "test", 0, "authMiddleware")
+				];
+			`)
+			expect(removeWhitespace(result.code)).toStrictEqual(expected)
+		})
+
+		it("should handle aliased clientMiddleware import used in array", () => {
+			const result = augmentMiddlewareFunctions(
+				`
+				import { someClientAuthFn as clientAuthMiddleware } from "./auth";
+				export const clientMiddleware = [clientAuthMiddleware];
+				`,
+				"test",
+				"/file/path"
+			)
+			const expected = removeWhitespace(`
+				import { withClientMiddlewareWrapperSingle as _withClientMiddlewareWrapperSingle } from "react-router-devtools/client";
+				import { someClientAuthFn as clientAuthMiddleware } from "./auth";
+				export const clientMiddleware = [
+					_withClientMiddlewareWrapperSingle(clientAuthMiddleware, "test", 0, "clientAuthMiddleware")
+				];
+			`)
+			expect(removeWhitespace(result.code)).toStrictEqual(expected)
+		})
+	})
 })