fix(core): mute puppeteer headless new warnings

Summary: ## Before Puppeteer showing warning messages when running MemLab: {F1075018402} ## After No warning messages: {F1075018416} Differential Revision: D48610311 fbshipit-source-id: 9a535a89c9bbe31dd8bf836aadda09e1b9f0f6be
facebook · Aug 24, 2023 · a114be4 · a114be4
1 parent 85aceb0
commit a114be4
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/packages/api/src/__tests__/heap/lib/HeapParserTestUtils.ts b/packages/api/src/__tests__/heap/lib/HeapParserTestUtils.ts
@@ -63,6 +63,7 @@ async function dumpHeap(
   snapshotFile: string,
   leakInjector: () => void,
 ): Promise<void> {
+  utils.tryToMutePuppeteerWarning();
   const browser = await puppeteer.launch(config.puppeteerConfig);
   const page = await browser.newPage();
   // set page size

diff --git a/packages/api/src/lib/APIUtils.ts b/packages/api/src/lib/APIUtils.ts
@@ -24,6 +24,7 @@ async function getBrowser(
 ): Promise<Browser> {
   const runConfig = options.config ?? config;
   let browser: Browser;
+  utils.tryToMutePuppeteerWarning();
   if (runConfig.isLocalPuppeteer && !options.warmup) {
     try {
       browser = await puppeteer.connect({

diff --git a/packages/core/src/lib/Utils.ts b/packages/core/src/lib/Utils.ts
@@ -2092,6 +2092,16 @@ function objectToMap(object: StringRecord): Map<string, string> {
   return ret;
 }
 
+function tryToMutePuppeteerWarning() {
+  if (process.env['PUPPETEER_DISABLE_HEADLESS_WARNING'] == null) {
+    info.lowLevel(
+      'Setting env variable PUPPETEER_DISABLE_HEADLESS_WARNING=1 ' +
+        'to mute puppeteer warnings.',
+    );
+    process.env['PUPPETEER_DISABLE_HEADLESS_WARNING'] = '1';
+  }
+}
+
 export default {
   aggregateDominatorMetrics,
   applyToNodes,
@@ -2203,5 +2213,6 @@ export default {
   shouldShowMoreInfo,
   shuffleArray,
   throwError,
+  tryToMutePuppeteerWarning,
   upperCaseFirstCharacter,
 };