feat: rewrite test files

FuelLabs · Jul 12, 2024 · 04dd9d5 · 04dd9d5
1 parent da35aad
commit 04dd9d5
Show file tree

Hide file tree

Showing 5 changed files with 44 additions and 0 deletions.
diff --git a/packages/create-fuels/src/lib/rewriteTemplateFiles.test.ts b/packages/create-fuels/src/lib/rewriteTemplateFiles.test.ts
@@ -58,4 +58,33 @@ describe('rewriteTemplateFiles', () => {
     expect(fuelsConfig).not.toContain(/\n\W+forcPath: 'fuels-forc',/g);
     expect(fuelsConfig).not.toContain(/\n\W+fuelCorePath: 'fuels-core',/g);
   });
+
+  it('should rewrite the test files', () => {
+    const testDir = join(paths.template, 'test');
+    const programs = [
+      {
+        program: 'contract',
+        capitalised: 'Contract',
+      },
+      {
+        program: 'predicate',
+        capitalised: 'Predicate',
+      },
+      {
+        program: 'script',
+        capitalised: 'Script',
+      },
+    ];
+
+    rewriteTemplateFiles(paths.template);
+
+    programs.forEach(({ program, capitalised }) => {
+      const testFilePath = join(testDir, `${program}.test.ts`);
+      const testFileContents = readFileSync(testFilePath, 'utf-8');
+
+      expect(testFileContents).not.toContain('@group node');
+      expect(testFileContents).not.toContain('@group browser');
+      expect(testFileContents).toContain(`${capitalised} Testing`);
+    });
+  });
 });
diff --git a/packages/create-fuels/src/lib/rewriteTemplateFiles.ts b/packages/create-fuels/src/lib/rewriteTemplateFiles.ts
@@ -17,4 +17,16 @@ export const rewriteTemplateFiles = (templateDir: string) => {
   contents = contents.replace(/\n\W+forcPath: 'fuels-forc',/g, '');
   contents = contents.replace(/\n\W+fuelCorePath: 'fuels-core',/g, '');
   writeFileSync(fuelsConfigFilePath, contents);
+
+  // tests
+  const testDir = join(templateDir, 'test');
+  const programs = ['contract', 'predicate', 'script'];
+  programs.forEach((program) => {
+    const testFilePath = join(testDir, `${program}.test.ts`);
+    contents = readFileSync(testFilePath, 'utf-8');
+    const capitalisedProgram = program.charAt(0).toUpperCase() + program.slice(1);
+    contents = contents.replace(/@group node/g, `${capitalisedProgram} Testing`);
+    contents = contents.replace(/@group browser/g, '');
+    writeFileSync(testFilePath, contents);
+  });
 };
diff --git a/templates/nextjs/test/contract.test.ts b/templates/nextjs/test/contract.test.ts
@@ -12,6 +12,7 @@ import bytecode from '../src/sway-api/contracts/TestContractAbi.hex';
 
 /**
  * @group node
+ * @group browser
  *
  * Tests for the contract program type within the TS SDK. Here we will test the deployment of
  * our contract, and the result of call it's functions.

diff --git a/templates/nextjs/test/predicate.test.ts b/templates/nextjs/test/predicate.test.ts
@@ -11,6 +11,7 @@ import { TestPredicateAbi__factory, TestPredicateAbiInputs } from '../src/sway-a
 
 /**
  * @group node
+ * @group browser
  *
  * Tests for the predicate program type within the TS SDK. Here we will test the use of our predicate
  * in a transaction.

diff --git a/templates/nextjs/test/script.test.ts b/templates/nextjs/test/script.test.ts
@@ -11,6 +11,7 @@ import { TestScriptAbi__factory } from '../src/sway-api';
 
 /**
  * @group node
+ * @group browser
  *
  * Tests for the script program type within the TS SDK. Here we will test the use of our script
  * function call.