From c642de322cf3892cdec4544b42ce5c3527180871 Mon Sep 17 00:00:00 2001
From: Sathya Gunasekaran <gsathya.ceg@gmail.com>
Date: Mon, 6 Mar 2023 19:53:54 +0000
Subject: [PATCH] [test] Enable "use forget" directive in tests

---
 .../test-utils/generateTestsFromFixtures.ts   | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts b/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts
index 573832f4ec..3b976e2374 100644
--- a/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts
+++ b/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts
@@ -42,7 +42,14 @@ expect.extend({
 
 export default function generateTestsFromFixtures(
   fixturesPath: string,
-  transform: (input: string, file: any, options: { debug: boolean }) => string
+  transform: (
+    input: string,
+    file: any,
+    options: {
+      debug: boolean;
+      enableOnlyOnUseForgetDirective: boolean;
+    }
+  ) => string
 ) {
   const files = fs.readdirSync(fixturesPath);
   const fixtures = matchInputOutputFixtures(files, fixturesPath);
@@ -75,6 +82,8 @@ export default function generateTestsFromFixtures(
         let testCommand = test;
         let input: string | null = null;
         let debug = false;
+        let enableOnlyOnUseForgetDirective = false;
+
         if (inputFile != null) {
           input = fs.readFileSync(inputFile, "utf8");
           const lines = input.split("\n");
@@ -85,12 +94,18 @@ export default function generateTestsFromFixtures(
           if (lines[0]!.indexOf("@skip") !== -1) {
             testCommand = test.skip;
           }
+          if (lines[0]!.indexOf("@forgetDirective") !== -1) {
+            enableOnlyOnUseForgetDirective = true;
+          }
         }
 
         testCommand(basename, () => {
           let receivedOutput;
           if (input !== null) {
-            receivedOutput = transform(input, basename, { debug });
+            receivedOutput = transform(input, basename, {
+              debug,
+              enableOnlyOnUseForgetDirective,
+            });
           } else {
             receivedOutput = "<<input deleted>>";
           }