Add unit tests

Author: nickserv

src/LineWriterImpl.ts

@@ -0,0 +1,47 @@
+/**
+    Copyright 2022 Google LLC
+    Licensed under the Apache License, Version 2.0 (the "License");
+    you may not use this file except in compliance with the License.
+    You may obtain a copy of the License at
+        https://www.apache.org/licenses/LICENSE-2.0
+    Unless required by applicable law or agreed to in writing, software
+    distributed under the License is distributed on an "AS IS" BASIS,
+    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    See the License for the specific language governing permissions and
+    limitations under the License.
+ */
+
+import { LineWriter } from "@puppeteer/replay"
+
+export class LineWriterImpl implements LineWriter {
+  #indentation: string
+  #currentIndentation = 0
+  #lines: string[] = []
+
+  constructor(indentation: string) {
+    this.#indentation = indentation
+  }
+
+  appendLine(line: string): LineWriter {
+    const indentedLine = line
+      ? this.#indentation.repeat(this.#currentIndentation) + line.trimEnd()
+      : ""
+    this.#lines.push(indentedLine)
+    return this
+  }
+
+  startBlock(): LineWriter {
+    this.#currentIndentation++
+    return this
+  }
+
+  endBlock(): LineWriter {
+    this.#currentIndentation--
+    return this
+  }
+
+  toString(): string {
+    // Scripts should end with a final blank line.
+    return this.#lines.join("\n")
+  }
+}

src/index.ts

@@ -46,9 +46,9 @@ export class Extension implements StringifyExtension {
     switch (step.type) {
       case "click":
         out.appendLine(
-          `userEvent.click(${JSON.stringify(
-            stringifySelector(step.selectors[0]),
-          )})`,
+          `await userEvent.click(${stringifySelector(step.selectors[0])}${
+            step.button === "secondary" ? ", { buttons: 2 }" : ""
+          })`,
         )
         break
       case "navigate":

src/test.ts

@@ -1,11 +1,71 @@
 import { readFile } from "fs/promises"
 import { join } from "path"
-import { stringify, UserFlow } from "@puppeteer/replay"
+import {
+  stringify,
+  type Selector,
+  type Step,
+  type UserFlow,
+} from "@puppeteer/replay"
 import { Extension } from "."
 import flow from "./fixtures/Example.json"
+import { LineWriterImpl } from "./LineWriterImpl"
 
-test("Extension", async () => {
-  expect(
-    await stringify(flow as UserFlow, { extension: new Extension() }),
-  ).toBe(await readFile(join(__dirname, "fixtures/example.test.js"), "utf8"))
+const extension = new Extension()
+const selectors: Selector[] = [["aria/Test"], ["#test"]]
+
+describe("Extension", () => {
+  test("stringify", async () => {
+    expect(await stringify(flow as UserFlow, { extension })).toBe(
+      await readFile(join(__dirname, "fixtures/example.test.js"), "utf8"),
+    )
+  })
+
+  describe("stringifyStep", () => {
+    test.each<[Step, string]>([
+      [
+        {
+          type: "click",
+          selectors,
+          offsetX: 0,
+          offsetY: 0,
+        },
+        'await userEvent.click(screen.getByText("Test"))',
+      ],
+      [
+        {
+          type: "click",
+          selectors,
+          button: "secondary",
+          offsetX: 0,
+          offsetY: 0,
+        },
+        'await userEvent.click(screen.getByText("Test"), { buttons: 2 })',
+      ],
+      [
+        {
+          type: "navigate",
+          url: "https://example.com/",
+          assertedEvents: [
+            {
+              type: "navigation",
+              url: "https://example.com/",
+              title: "Example Domain",
+            },
+          ],
+        },
+        'expect(location.href).toBe("https://example.com/")\nexpect(document.title).toBe("Example Domain")',
+      ],
+      [
+        {
+          type: "waitForElement",
+          selectors,
+        },
+        'await waitFor(() => screen.getByText("Test"))',
+      ],
+    ])("%p", async (step, expected) => {
+      const writer = new LineWriterImpl("  ")
+      await extension.stringifyStep(writer, step, { title: "", steps: [step] })
+      expect(writer.toString()).toBe(expected)
+    })
+  })
 })