test(e2e): add paid media flow coverage

e2e/fixtures/api-mocks.ts

@@ -5,11 +5,14 @@ import {
   chatSendResponse,
   createPaymentOrderResponse,
   createChatSendResponse,
+  createPaidImageHistoryResponse,
   createWeeklyLimitChatSendResponse,
   e2eEmailUser,
   emptyChatHistoryResponse,
   emailLoginResponse,
   guestLoginResponse,
+  paidImageChatSendResponse,
+  paidImageUnlockHistoryResponse,
   paidPaymentOrderStatusResponse,
   paymentPlansResponse,
   userEntitlementsResponse,
@@ -18,20 +21,27 @@ import {
 
 interface MockCoreApisOptions {
   chatLimitTriggerAt?: number;
+  paidImageFlow?: boolean;
 }
 
 export async function mockCoreApis(
   page: Page,
   options: MockCoreApisOptions = {},
 ): Promise<void> {
-  const { chatLimitTriggerAt } = options;
+  const { chatLimitTriggerAt, paidImageFlow = false } = options;
   let sendCount = 0;
+  let paidImageRequested = false;
+  let paidImageUnlocked = false;
 
   await page.route("**/api/auth/guest", async (route) => {
     await route.fulfill({ json: apiEnvelope(guestLoginResponse) });
   });
 
   await page.route("**/api/auth/login", async (route) => {
+    if (paidImageFlow) {
+      paidImageRequested = true;
+    }
+
     await route.fulfill({ json: apiEnvelope(emailLoginResponse) });
   });
 
@@ -48,18 +58,62 @@ export async function mockCoreApis(
   });
 
   await page.route("**/api/chat/history**", async (route) => {
-    await route.fulfill({ json: apiEnvelope(emptyChatHistoryResponse) });
+    const response =
+      paidImageFlow && paidImageRequested
+        ? createPaidImageHistoryResponse(paidImageUnlocked)
+        : emptyChatHistoryResponse;
+
+    await route.fulfill({ json: apiEnvelope(response) });
   });
 
   await page.route("**/api/chat/send", async (route) => {
     sendCount += 1;
+    const requestBody = route.request().postDataJSON() as
+      | { message?: unknown }
+      | undefined;
+    const message =
+      typeof requestBody?.message === "string" ? requestBody.message : "";
 
     const response =
-      chatLimitTriggerAt != null && sendCount >= chatLimitTriggerAt
-        ? createWeeklyLimitChatSendResponse(sendCount, chatLimitTriggerAt)
-        : chatLimitTriggerAt != null
-          ? createChatSendResponse(sendCount)
-          : chatSendResponse;
+      paidImageFlow && message.includes("给我发图片")
+        ? paidImageChatSendResponse
+        : chatLimitTriggerAt != null && sendCount >= chatLimitTriggerAt
+          ? createWeeklyLimitChatSendResponse(sendCount, chatLimitTriggerAt)
+          : chatLimitTriggerAt != null
+            ? createChatSendResponse(sendCount)
+            : chatSendResponse;
+
+    if (paidImageFlow && message.includes("给我发图片")) {
+      paidImageRequested = true;
+      paidImageUnlocked = false;
+    }
+
+    await route.fulfill({ json: apiEnvelope(response) });
+  });
+
+  await page.route("**/api/chat/unlock-history", async (route) => {
+    if (paidImageFlow && paidImageRequested) {
+      paidImageUnlocked = true;
+    }
+
+    const response =
+      paidImageFlow && paidImageRequested
+        ? paidImageUnlockHistoryResponse
+        : {
+            unlocked: true,
+            reason: "no_locked_messages",
+            totalLocked: 0,
+            unlockedCount: 0,
+            privateCount: 0,
+            imageCount: 0,
+            voiceCount: 0,
+            requiredCredits: 0,
+            currentCredits: 0,
+            remainingCredits: 0,
+            shortfallCredits: 0,
+            costsByMessage: {},
+            messageIds: [],
+          };
 
     await route.fulfill({ json: apiEnvelope(response) });
   });
@@ -73,6 +127,10 @@ export async function mockCoreApis(
   });
 
   await page.route("**/api/payment/order-status**", async (route) => {
+    if (paidImageFlow && paidImageRequested) {
+      paidImageUnlocked = true;
+    }
+
     await route.fulfill({ json: apiEnvelope(paidPaymentOrderStatusResponse) });
   });