From 2ede182f5e0a75e7c6759aec0b24dcd49f6eff00 Mon Sep 17 00:00:00 2001
From: Sathya Gunasekaran <gsathya.ceg@gmail.com>
Date: Mon, 6 Mar 2023 19:53:57 +0000
Subject: [PATCH] [test] Enable gatingModule in tests

---
 .../src/__tests__/test-utils/generateTestsFromFixtures.ts   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts b/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts
index 3b976e2374..5440301508 100644
--- a/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts
+++ b/compiler/forget/src/__tests__/test-utils/generateTestsFromFixtures.ts
@@ -48,6 +48,7 @@ export default function generateTestsFromFixtures(
     options: {
       debug: boolean;
       enableOnlyOnUseForgetDirective: boolean;
+      gatingModule: string | null;
     }
   ) => string
 ) {
@@ -83,6 +84,7 @@ export default function generateTestsFromFixtures(
         let input: string | null = null;
         let debug = false;
         let enableOnlyOnUseForgetDirective = false;
+        let gatingModule: string | null = null;
 
         if (inputFile != null) {
           input = fs.readFileSync(inputFile, "utf8");
@@ -97,6 +99,9 @@ export default function generateTestsFromFixtures(
           if (lines[0]!.indexOf("@forgetDirective") !== -1) {
             enableOnlyOnUseForgetDirective = true;
           }
+          if (lines[0]!.indexOf("@gatingModule") !== -1) {
+            gatingModule = "ReactForgetFeatureFlag";
+          }
         }
 
         testCommand(basename, () => {
@@ -105,6 +110,7 @@ export default function generateTestsFromFixtures(
             receivedOutput = transform(input, basename, {
               debug,
               enableOnlyOnUseForgetDirective,
+              gatingModule,
             });
           } else {
             receivedOutput = "<<input deleted>>";