From 69eb9ccccc93d0ac4bf63e64c515e544c6dbff16 Mon Sep 17 00:00:00 2001
From: Sathya Gunasekaran <gsathya.ceg@gmail.com>
Date: Thu, 9 Nov 2023 12:00:34 +0000
Subject: [PATCH] [snap] Add option to set panicThreshold

---
 .../fixture-test-utils/src/compiler-utils.ts         | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/compiler/packages/fixture-test-utils/src/compiler-utils.ts b/compiler/packages/fixture-test-utils/src/compiler-utils.ts
index 54bb9977f5..aabeb47f0e 100644
--- a/compiler/packages/fixture-test-utils/src/compiler-utils.ts
+++ b/compiler/packages/fixture-test-utils/src/compiler-utils.ts
@@ -7,7 +7,10 @@
 
 import assert from "assert";
 import type { runReactForgetBabelPlugin as RunReactForgetBabelPlugin } from "babel-plugin-react-forget/src/Babel/RunReactForgetBabelPlugin";
-import { CompilationMode } from "babel-plugin-react-forget/src/Entrypoint";
+import {
+  CompilationMode,
+  PanicThresholdOptions,
+} from "babel-plugin-react-forget/src/Entrypoint";
 import type { Effect, ValueKind } from "babel-plugin-react-forget/src/HIR";
 import type { parseConfigPragma as ParseConfigPragma } from "babel-plugin-react-forget/src/HIR/Environment";
 import prettier from "prettier";
@@ -32,6 +35,7 @@ export function transformFixtureInput(
   let enableEmitFreeze = null;
   let compilationMode: CompilationMode = "all";
   let enableUseMemoCachePolyfill = false;
+  let panicThreshold: PanicThresholdOptions = "ALL_ERRORS";
 
   if (firstLine.indexOf("@compilationMode(annotation)") !== -1) {
     assert(
@@ -69,6 +73,10 @@ export function transformFixtureInput(
   if (firstLine.includes("@enableUseMemoCachePolyfill")) {
     enableUseMemoCachePolyfill = true;
   }
+  if (firstLine.includes("@panicThreshold(NONE)")) {
+    panicThreshold = "NONE";
+  }
+
   const config = parseConfigPragmaFn(firstLine);
   const result = pluginFn(
     input,
@@ -113,7 +121,7 @@ export function transformFixtureInput(
       logger: null,
       gating,
       instrumentForget,
-      panicThreshold: "ALL_ERRORS",
+      panicThreshold,
       noEmit: false,
       enableUseMemoCachePolyfill,
     },