Add Join Record to more tests

tests/join_form.spec.ts

@@ -14,6 +14,7 @@ import {
   submitAndCheckToast,
   expectSuccess,
   sampleJoinRecord,
+  findJoinRecord,
 } from "@/tests/util";
 import { isDeepStrictEqual } from "util";
 
@@ -89,8 +90,22 @@ test("confirm city", async ({ page }) => {
 
   await submitConfirmationDialogExpected(page, 2000);
 
+  // Check that the Join Record's code is correct.
+  let joinRecord = await findJoinRecord(page);
+  let code = "409"
+  if (joinRecord.code !== code) {
+    throw new Error(`JoinRecord code (${joinRecord.code}) did not match expected code (${code})`);
+  }
+
   await page.locator("[name='confirm']").click();
 
+  // Make sure the JoinRecord updated properly
+  await page.waitForTimeout(1000);
+  joinRecord = await findJoinRecord(page);
+  if (joinRecord.code !== "201") {
+    throw new Error(`JoinRecord code (${joinRecord.code}) did not match expected code (201)`);
+  }
+
   await expectSuccess(page, 1000);
 });
 
@@ -358,4 +373,11 @@ test("fail nj", async ({ page }) => {
     page,
     "Non-NYC registrations are not supported at this time",
   );
+
+  // Check that the Join Record's code is correct.
+  let joinRecord = await findJoinRecord(page);
+  let code = "400"
+  if (joinRecord.code !== code) {
+    throw new Error(`JoinRecord code (${joinRecord.code}) did not match expected code (${code})`);
+  }
 });

tests/util.ts

@@ -1,3 +1,4 @@
+import { getJoinRecordFromS3 } from "@/app/join_record";
 import { JoinRecord } from "@/app/types";
 import { JoinFormValues } from "@/components/JoinForm/JoinForm";
 import { expect, Page } from "@playwright/test";
@@ -163,3 +164,19 @@ export async function submitConfirmationDialogExpected(
   await page.waitForTimeout(timeout);
   await expect(page.locator("[id='alert-dialog-title']")).toBeVisible();
 }
+
+export async function findJoinRecord(page: Page): Promise<JoinRecord> {
+  const joinRecordKey = await page.getAttribute(
+    '[data-testid="test-join-record-key"]',
+    "data-state",
+  );
+
+  if (joinRecordKey === null) {
+    throw new Error("Got null join record");
+  }
+
+  // This is wacky because playwright has to run this and therefore
+  // needs our dotenv.
+  const joinRecord: JoinRecord = await getJoinRecordFromS3(joinRecordKey);
+  return joinRecord;
+}