undo e2e (#300)

e2e-tests/fixtures/write-index-2.md

@@ -0,0 +1,15 @@
+OK, I'm going to do some writing now...
+
+<dyad-write path="src/pages/Index.tsx" description="write-description">
+const Index = () => {
+  return (
+    <div>
+        Testing:write-index(2)!
+    </div>
+  );
+};
+
+export default Index;
+</dyad-write>
+
+And it's done!

e2e-tests/helpers/test_helper.ts

@@ -73,10 +73,18 @@ class PageObject {
     await this.getRetryButton().click();
   }
 
+  async clickUndo() {
+    await this.getUndoButton().click();
+  }
+
   private getRetryButton() {
     return this.page.getByRole("button", { name: "Retry" });
   }
 
+  private getUndoButton() {
+    return this.page.getByRole("button", { name: "Undo" });
+  }
+
   async sendPrompt(prompt: string) {
     await this.page
       .getByRole("textbox", { name: "Ask Dyad to build..." })
@@ -84,7 +92,7 @@ class PageObject {
     await this.page
       .getByRole("textbox", { name: "Ask Dyad to build..." })
       .fill(prompt);
-    await this.page.getByRole("button", { name: "Start new chat" }).click();
+    await this.page.getByRole("button", { name: "Send message" }).click();
     await this.waitForChatCompletion();
   }
 

e2e-tests/undo.spec.ts

@@ -0,0 +1,25 @@
+import { test } from "./helpers/test_helper";
+import { expect } from "@playwright/test";
+
+test("undo", async ({ po }) => {
+  await po.setUp({ autoApprove: true });
+  await po.sendPrompt("tc=write-index");
+  await po.sendPrompt("tc=write-index-2");
+
+  const iframe = po.getPreviewIframeElement();
+  await expect(
+    iframe.contentFrame().getByText("Testing:write-index(2)!"),
+  ).toBeVisible({
+    // This can be pretty slow because it's waiting for the app to build.
+    timeout: 15_000,
+  });
+
+  await po.clickUndo();
+
+  await expect(
+    iframe.contentFrame().getByText("Testing:write-index!"),
+  ).toBeVisible({
+    // Also, could be slow.
+    timeout: 15_000,
+  });
+});