test: add setup and cleanup functions

Author: mcarvin8

test/commands/acc-transformer/testCleanup.ts

@@ -0,0 +1,27 @@
+/* eslint-disable no-await-in-loop */
+import { rm } from 'node:fs/promises';
+import { resolve } from 'node:path';
+
+import { formatOptions } from '../../../src/utils/constants.js';
+import { sfdxConfigFile, inputJsons, defaultPath } from './testConstants.js';
+
+export async function postTestCleanup(): Promise<void> {
+  await rm(sfdxConfigFile);
+  await rm('force-app/main/default/classes/AccountProfile.cls');
+  await rm('packaged/triggers/AccountTrigger.trigger');
+  await rm('force-app', { recursive: true });
+  await rm('packaged', { recursive: true });
+
+  const pathsToRemove = formatOptions
+    .flatMap((format) =>
+      inputJsons.map(({ label }) => {
+        const extension = format === 'lcovonly' ? 'info' : 'xml';
+        return resolve(`${format}_${label}.${extension}`);
+      })
+    )
+    .concat(defaultPath);
+
+  for (const path of pathsToRemove) {
+    await rm(path).catch(() => {});
+  }
+}

test/commands/acc-transformer/testSetup.ts

@@ -0,0 +1,12 @@
+/* eslint-disable no-await-in-loop */
+import { mkdir, copyFile, writeFile } from 'node:fs/promises';
+
+import { sfdxConfigFile, baselineClassPath, baselineTriggerPath, configJsonString } from './testConstants.js';
+
+export async function preTestSetup(): Promise<void> {
+  await mkdir('force-app/main/default/classes', { recursive: true });
+  await mkdir('packaged/triggers', { recursive: true });
+  await copyFile(baselineClassPath, 'force-app/main/default/classes/AccountProfile.cls');
+  await copyFile(baselineTriggerPath, 'packaged/triggers/AccountTrigger.trigger');
+  await writeFile(sfdxConfigFile, configJsonString);
+}

test/commands/acc-transformer/transform.nut.ts

@@ -1,55 +1,27 @@
 /* eslint-disable no-await-in-loop */
 'use strict';
 
-import { copyFile, writeFile, rm, mkdir } from 'node:fs/promises';
 import { resolve } from 'node:path';
 
 import { execCmd, TestSession } from '@salesforce/cli-plugins-testkit';
 import { expect } from 'chai';
 import { formatOptions } from '../../../src/utils/constants.js';
-import {
-  baselineClassPath,
-  baselineTriggerPath,
-  configJsonString,
-  sfdxConfigFile,
-  inputJsons,
-  defaultPath,
-  invalidJson,
-} from './testConstants.js';
+import { inputJsons, invalidJson } from './testConstants.js';
 import { compareToBaselines } from './baselineCompare.js';
+import { postTestCleanup } from './testCleanup.js';
+import { preTestSetup } from './testSetup.js';
 
 describe('acc-transformer transform NUTs', () => {
   let session: TestSession;
 
   before(async () => {
     session = await TestSession.create({ devhubAuthStrategy: 'NONE' });
-    await mkdir('force-app/main/default/classes', { recursive: true });
-    await mkdir('packaged/triggers', { recursive: true });
-    await copyFile(baselineClassPath, 'force-app/main/default/classes/AccountProfile.cls');
-    await copyFile(baselineTriggerPath, 'packaged/triggers/AccountTrigger.trigger');
-    await writeFile(sfdxConfigFile, configJsonString);
+    await preTestSetup();
   });
 
   after(async () => {
     await session?.clean();
-    await rm(sfdxConfigFile);
-    await rm('force-app/main/default/classes/AccountProfile.cls');
-    await rm('packaged/triggers/AccountTrigger.trigger');
-    await rm('force-app', { recursive: true });
-    await rm('packaged', { recursive: true });
-
-    const pathsToRemove = formatOptions
-      .flatMap((format) =>
-        inputJsons.map(({ label }) => {
-          const extension = format === 'lcovonly' ? 'info' : 'xml';
-          return resolve(`${format}_${label}.${extension}`);
-        })
-      )
-      .concat(defaultPath);
-
-    for (const path of pathsToRemove) {
-      await rm(path).catch(() => {});
-    }
+    await postTestCleanup();
   });
 
   formatOptions.forEach((format) => {

test/commands/acc-transformer/transform.test.ts

@@ -1,36 +1,24 @@
 /* eslint-disable no-await-in-loop */
 'use strict';
 
-import { copyFile, writeFile, rm, mkdir } from 'node:fs/promises';
 import { resolve } from 'node:path';
 
 import { TestContext } from '@salesforce/core/testSetup';
 import { expect } from 'chai';
 import { stubSfCommandUx } from '@salesforce/sf-plugins-core';
 import TransformerTransform from '../../../src/commands/acc-transformer/transform.js';
 import { formatOptions } from '../../../src/utils/constants.js';
-import {
-  baselineClassPath,
-  baselineTriggerPath,
-  configJsonString,
-  sfdxConfigFile,
-  inputJsons,
-  defaultPath,
-  invalidJson,
-  deployCoverage,
-} from './testConstants.js';
+import { inputJsons, defaultPath, invalidJson, deployCoverage } from './testConstants.js';
 import { compareToBaselines } from './baselineCompare.js';
+import { postTestCleanup } from './testCleanup.js';
+import { preTestSetup } from './testSetup.js';
 
 describe('main', () => {
   const $$ = new TestContext();
   let sfCommandStubs: ReturnType<typeof stubSfCommandUx>;
 
   before(async () => {
-    await mkdir('force-app/main/default/classes', { recursive: true });
-    await mkdir('packaged/triggers', { recursive: true });
-    await copyFile(baselineClassPath, 'force-app/main/default/classes/AccountProfile.cls');
-    await copyFile(baselineTriggerPath, 'packaged/triggers/AccountTrigger.trigger');
-    await writeFile(sfdxConfigFile, configJsonString);
+    await preTestSetup();
   });
 
   beforeEach(() => {
@@ -42,24 +30,7 @@ describe('main', () => {
   });
 
   after(async () => {
-    await rm(sfdxConfigFile);
-    await rm('force-app/main/default/classes/AccountProfile.cls');
-    await rm('packaged/triggers/AccountTrigger.trigger');
-    await rm('force-app', { recursive: true });
-    await rm('packaged', { recursive: true });
-
-    const pathsToRemove = formatOptions
-      .flatMap((format) =>
-        inputJsons.map(({ label }) => {
-          const extension = format === 'lcovonly' ? 'info' : 'xml';
-          return resolve(`${format}_${label}.${extension}`);
-        })
-      )
-      .concat(defaultPath);
-
-    for (const path of pathsToRemove) {
-      await rm(path).catch(() => {});
-    }
+    await postTestCleanup();
   });
 
   formatOptions.forEach((format) => {