Refactor setupTempDirectory to accept objects for dirs

Author: kraenhansen

packages/host/src/node/path-utils.test.ts

@@ -268,24 +268,30 @@ describe("getLibraryName", () => {
 describe("findPackageDependencyPaths", () => {
   it("should find package dependency paths", (context) => {
     const tempDir = setupTempDirectory(context, {
-      "node_modules/lib-a/package.json": JSON.stringify({
-        name: "lib-a",
-        main: "index.js",
-      }),
-      "node_modules/lib-a/index.js": "",
-      "test-package/node_modules/lib-b/package.json": JSON.stringify({
-        name: "lib-b",
-        main: "index.js",
-      }),
-      "test-package/node_modules/lib-b/index.js": "",
-      "test-package/package.json": JSON.stringify({
-        name: "test-package",
-        dependencies: {
-          "lib-a": "^1.0.0",
-          "lib-b": "^1.0.0",
+      "node_modules/lib-a": {
+        "package.json": JSON.stringify({
+          name: "lib-a",
+          main: "index.js",
+        }),
+        "index.js": "",
+      },
+      "test-package": {
+        "package.json": JSON.stringify({
+          name: "test-package",
+          dependencies: {
+            "lib-a": "^1.0.0",
+            "lib-b": "^1.0.0",
+          },
+        }),
+        "src/index.js": "console.log('Hello, world!')",
+        "node_modules/lib-b": {
+          "package.json": JSON.stringify({
+            name: "lib-b",
+            main: "index.js",
+          }),
+          "index.js": "",
         },
-      }),
-      "test-package/src/index.js": "console.log('Hello, world!')",
+      },
     });
 
     const result = findPackageDependencyPaths(

packages/host/src/node/test-utils.ts

@@ -3,23 +3,32 @@ import os from "node:os";
 import fs from "node:fs";
 import path from "node:path";
 
-export function setupTempDirectory(
-  context: TestContext,
-  files: Record<string, string>
-) {
+export interface FileMap {
+  [key: string]: string | FileMap;
+}
+
+function writeFiles(fromPath: string, files: FileMap) {
+  for (const [filePath, content] of Object.entries(files)) {
+    const fullPath = path.join(fromPath, filePath);
+    fs.mkdirSync(path.dirname(fullPath), { recursive: true });
+    if (typeof content === "string") {
+      fs.writeFileSync(fullPath, content, "utf8");
+    } else {
+      writeFiles(fullPath, content);
+    }
+  }
+}
+
+export function setupTempDirectory(context: TestContext, files: FileMap) {
   const tempDirectoryPath = fs.realpathSync(
-    fs.mkdtempSync(path.join(os.tmpdir(), "babel-transform-test-"))
+    fs.mkdtempSync(path.join(os.tmpdir(), "react-native-node-api-test-"))
   );
 
   context.after(() => {
     fs.rmSync(tempDirectoryPath, { recursive: true, force: true });
   });
 
-  for (const [filePath, content] of Object.entries(files)) {
-    const fullPath = path.join(tempDirectoryPath, filePath);
-    fs.mkdirSync(path.dirname(fullPath), { recursive: true });
-    fs.writeFileSync(fullPath, content, "utf8");
-  }
+  writeFiles(tempDirectoryPath, files);
 
   return tempDirectoryPath;
 }