fix(puppeteer-page): add existing head content to page

Read the existing head element content, add that content to the e2e page that is being created. This prevents a user losing script or style tags that may have been added to the page before using the setContent method. Fixes ionic-team#3229
Trendy · Feb 25, 2022 · 1b00732 · 1b00732
1 parent 5dccc85
commit 1b00732
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/testing/puppeteer/puppeteer-page.ts b/src/testing/puppeteer/puppeteer-page.ts
@@ -213,6 +213,10 @@ async function e2eSetContent(page: E2EPageInternal, html: string, options: WaitF
 
   const output: string[] = [];
 
+  const headContent: string = await page.$eval('head', async (headElement: Element) => {
+    return headElement ? headElement.innerHTML : '';
+  });
+
   const appScriptUrl = env.__STENCIL_APP_SCRIPT_URL__;
   if (typeof appScriptUrl !== 'string') {
     throw new Error('invalid e2eSetContent() app script url');
@@ -222,6 +226,8 @@ async function e2eSetContent(page: E2EPageInternal, html: string, options: WaitF
   output.push(`<html>`);
   output.push(`<head>`);
 
+  output.push(headContent);
+
   const appStyleUrl = env.__STENCIL_APP_STYLE_URL__;
   if (typeof appStyleUrl === 'string') {
     output.push(`<link rel="stylesheet" href="${appStyleUrl}">`);