test: speed up prisma-generator tests by pnpm-installing to a test-specific store once, then offline-installing from that store for each test going forward

Author: WimTibackx

.gitignore

@@ -8,3 +8,6 @@ dist
 coverage
 .build
 .test
+
+
+/.pnpm-test-store

packages/schema/tests/generator/prisma-generator.test.ts

@@ -6,22 +6,30 @@ import path from 'path';
 import tmp from 'tmp';
 import { loadDocument } from '../../src/cli/cli-util';
 import { PrismaSchemaGenerator } from '../../src/plugins/prisma/schema-generator';
-import { execSync } from '../../src/utils/exec-utils';
 import { loadModel } from '../utils';
+import { preparePackageJson, initProjectDir } from '../../../../script/test-utils';
 
 tmp.setGracefulCleanup();
 
 describe('Prisma generator test', () => {
     let origDir: string;
 
+    let packageJsonContents: string;
+
+    beforeAll(async () => {
+        packageJsonContents = preparePackageJson({
+            "prisma": "^5.0.0"
+        });
+        console.log(`Got my packageJsonContents`);
+    })
+
     beforeEach(() => {
         origDir = process.cwd();
         const r = tmp.dirSync({ unsafeCleanup: true });
         console.log(`Project dir: ${r.name}`);
         process.chdir(r.name);
 
-        execSync('npm init -y', { stdio: 'ignore' });
-        execSync('npm install prisma');
+        initProjectDir(r.name, packageJsonContents);
     });
 
     afterEach(() => {

script/test-utils.ts

@@ -0,0 +1,53 @@
+import path from 'node:path';
+import fs from 'node:fs';
+import { tmpdir } from 'node:os';
+import { execSync } from 'node:child_process';
+
+export const PNPM_STORE_PATH = path.resolve(__dirname, '../.pnpm-test-store');
+export const NPM_RC_FILE = '.npmrc';
+export const NPM_RC_CONTENTS = `store-dir = ${PNPM_STORE_PATH}`;
+export const PACKAGE_JSON_FILE = 'package.json';
+export const PACKAGE_JSON_CONTENTS = '{"name":"test-project","version":"1.0.0"}';
+
+export function preparePackageJson(dependencies: {[key: string]: string} = {}, devDependencies: {[key: string]: string} = {}): string {
+  const tmpDir = fs.mkdtempSync(path.join(tmpdir(), 'zenstack-test-'));
+  try {
+  const packageJsonContents = 
+`{
+  "name":"test-project",
+  "version":"1.0.0",
+  "dependencies": {
+    ${Object.entries(dependencies).map(([k, v]) => `"${k}": "${v}"`).join(',\n')}
+  },
+  "devDependencies": {
+    ${Object.entries(devDependencies).map(([k, v]) => `"${k}": "${v}"`).join(',\n')}
+  }
+}`;
+
+  initProjectDir(tmpDir, packageJsonContents, false);
+
+  return packageJsonContents;
+  } finally {
+    fs.rmSync(tmpDir, {recursive: true, force: true});
+    console.log(`Loaded dependencies into store via temp dir ${tmpDir}`);
+  }
+}
+
+function execCmdSync(cmd: string, path: string) {
+  console.log(`Running: ${cmd}, in ${path}`);
+  try {
+      execSync(cmd, { cwd: path, stdio: 'ignore' });
+  } catch (err) {
+      console.error(`Test project scaffolding cmd error: ${err}`);
+      throw err;
+  }
+}
+
+export function initProjectDir(projectDir: string, packageJsonContents: string, offline: boolean = true) {
+  if (!fs.existsSync(projectDir)) {
+    fs.mkdirSync(projectDir, { recursive: true });
+  }
+  fs.writeFileSync(path.join(projectDir, PACKAGE_JSON_FILE), packageJsonContents, { flag: 'w+' });
+  fs.writeFileSync(path.join(projectDir, NPM_RC_FILE), NPM_RC_CONTENTS, { flag: 'w+' });
+  execCmdSync(`pnpm install ${offline ? '--offline ' : ''}--ignore-workspace`, projectDir);
+}