feat: support stringifying flows for Firefox

__snapshots__/PuppeteerStringifyExtension.test.ts.js

@@ -88,3 +88,15 @@ exports[
 }
 
 `;
+
+exports['PuppeteerStringifyExtension Firefox should stringify 1'] = `
+const puppeteer = require('puppeteer'); // v22.0.0 or later
+
+(async () => {
+  const browser = await puppeteer.launch({browser: 'firefox', protocol: 'webDriverBiDi'});
+  const page = await browser.newPage();
+  const timeout = 5000;
+  page.setDefaultTimeout(timeout);
+
+
+`;

src/PuppeteerStringifyExtension.ts

@@ -44,16 +44,30 @@ import {
 } from './SchemaUtils.js';
 import { formatJSONAsJS } from './JSONUtils.js';
 
+type TargetBrowser = 'chrome' | 'firefox';
+
 export class PuppeteerStringifyExtension extends StringifyExtension {
   #shouldAppendWaitForElementHelper = false;
+  #targetBrowser: TargetBrowser;
+
+  constructor(targetBrowser: TargetBrowser = 'chrome') {
+    super();
+    this.#targetBrowser = targetBrowser;
+  }
 
   override async beforeAllSteps(out: LineWriter, flow: UserFlow) {
     out.appendLine(
       "const puppeteer = require('puppeteer'); // v22.0.0 or later"
     );
     out.appendLine('');
     out.appendLine('(async () => {').startBlock();
-    out.appendLine('const browser = await puppeteer.launch();');
+    if (this.#targetBrowser === 'firefox') {
+      out.appendLine(
+        `const browser = await puppeteer.launch({browser: 'firefox', protocol: 'webDriverBiDi'});`
+      );
+    } else {
+      out.appendLine('const browser = await puppeteer.launch();');
+    }
     out.appendLine('const page = await browser.newPage();');
     out.appendLine(`const timeout = ${flow.timeout || defaultTimeout};`);
     out.appendLine('page.setDefaultTimeout(timeout);');

test/PuppeteerStringifyExtension.test.ts

@@ -95,4 +95,23 @@ describe('PuppeteerStringifyExtension', () => {
     await ext.stringifyStep(writer, step, flow);
     snapshot(writer.toString());
   });
+
+  describe('Firefox', () => {
+    const ext = new PuppeteerStringifyExtension('firefox');
+
+    it('should stringify', async () => {
+      const step = {
+        type: StepType.Click as const,
+        target: 'main',
+        selectors: ['aria/Test'],
+        offsetX: 1,
+        offsetY: 1,
+      };
+      const flow = { title: 'test', steps: [step] };
+
+      const writer = new InMemoryLineWriter('  ');
+      await ext.beforeAllSteps(writer, flow);
+      snapshot(writer.toString());
+    });
+  });
 });