chore: add tests

apify · Jun 18, 2024 · f20915e · f20915e
1 parent 137ed57
commit f20915e
Show file tree

Hide file tree

Showing 3 changed files with 63 additions and 1 deletion.
diff --git a/test/core/playwright_utils.test.ts b/test/core/playwright_utils.test.ts
@@ -12,7 +12,7 @@ let serverAddress = 'http://localhost:';
 let port: number;
 let server: Server;
 
-const launchContext = { launchOptions: { headless: true } };
+const launchContext = { launchOptions: { headless: false } };
 
 beforeAll(async () => {
     [server, port] = await runExampleComServer();
@@ -159,6 +159,22 @@ describe('playwrightUtils', () => {
         }
     });
 
+    test('parseWithCheerio() iframe expansion works', async () => {
+        const browser = await launchPlaywright(launchContext);
+
+        try {
+            const page = await browser.newPage();
+            await page.goto(new URL('/special/outside-iframe', serverAddress).toString());
+
+            const $ = await playwrightUtils.parseWithCheerio(page);
+
+            const headings = $('h1').map((i, el) => $(el).text()).get();
+            expect(headings).toEqual(['Outside iframe', 'In iframe']);
+        } finally {
+            await browser.close();
+        }
+    });
+
     describe('blockRequests()', () => {
         let browser: Browser = null;
         beforeAll(async () => {

diff --git a/test/core/puppeteer_utils.test.ts b/test/core/puppeteer_utils.test.ts
@@ -160,6 +160,22 @@ describe('puppeteerUtils', () => {
             }
         });
 
+        test('parseWithCheerio() iframe expansion works', async () => {
+            const browser = await launchPuppeteer(launchContext);
+
+            try {
+                const page = await browser.newPage();
+                await page.goto(new URL('/special/outside-iframe', serverAddress).toString());
+
+                const $ = await puppeteerUtils.parseWithCheerio(page);
+
+                const headings = $('h1').map((i, el) => $(el).text()).get();
+                expect(headings).toEqual(['Outside iframe', 'In iframe']);
+            } finally {
+                await browser.close();
+            }
+        });
+
         describe('blockRequests()', () => {
             let browser: Browser = null;
             beforeAll(async () => {

diff --git a/test/shared/_helper.ts b/test/shared/_helper.ts
@@ -172,6 +172,28 @@ console.log('Hello world!');
     </div>
 </body>
 </html>`,
+    outsideIframe: `
+    <!DOCTYPE html>
+    <html>
+        <head>
+            <title>Outside iframe</title>
+        </head>
+        <body>
+            <h1>Outside iframe</h1>
+            <iframe src="./inside-iframe"></iframe>
+        </body>
+    </html>`,
+    insideIframe: `
+    <!DOCTYPE html>
+    <html>
+        <head>
+            <title>In iframe</title>
+        </head>
+        <body>
+            <h1>In iframe</h1>
+            <p>Some content from inside of an iframe.</p>
+        </body>
+    </html>`,
 };
 
 export async function runExampleComServer(): Promise<[Server, number]> {
@@ -268,6 +290,14 @@ export async function runExampleComServer(): Promise<[Server, number]> {
         special.get('/cloudflareBlocking', async (_req, res) => {
             res.type('html').status(403).send(responseSamples.cloudflareBlocking);
         });
+
+        special.get('/outside-iframe', (_req, res) => {
+            res.type('html').send(responseSamples.outsideIframe);
+        });
+
+        special.get('/inside-iframe', (_req, res) => {
+            res.type('html').send(responseSamples.insideIframe);
+        });
     })();
 
     // "cacheable" site with one page, scripts and stylesheets