test(e2e): add mock chat quantity limit coverage
This commit is contained in:
@@ -3,6 +3,8 @@ import type { Page } from "@playwright/test";
|
||||
import {
|
||||
apiEnvelope,
|
||||
chatSendResponse,
|
||||
createChatSendResponse,
|
||||
createWeeklyLimitChatSendResponse,
|
||||
e2eEmailUser,
|
||||
emptyChatHistoryResponse,
|
||||
emailLoginResponse,
|
||||
@@ -12,7 +14,17 @@ import {
|
||||
vipStatusResponse,
|
||||
} from "./test-data";
|
||||
|
||||
export async function mockCoreApis(page: Page): Promise<void> {
|
||||
interface MockCoreApisOptions {
|
||||
chatLimitTriggerAt?: number;
|
||||
}
|
||||
|
||||
export async function mockCoreApis(
|
||||
page: Page,
|
||||
options: MockCoreApisOptions = {},
|
||||
): Promise<void> {
|
||||
const { chatLimitTriggerAt } = options;
|
||||
let sendCount = 0;
|
||||
|
||||
await page.route("**/api/auth/guest", async (route) => {
|
||||
await route.fulfill({ json: apiEnvelope(guestLoginResponse) });
|
||||
});
|
||||
@@ -38,7 +50,16 @@ export async function mockCoreApis(page: Page): Promise<void> {
|
||||
});
|
||||
|
||||
await page.route("**/api/chat/send", async (route) => {
|
||||
await route.fulfill({ json: apiEnvelope(chatSendResponse) });
|
||||
sendCount += 1;
|
||||
|
||||
const response =
|
||||
chatLimitTriggerAt != null && sendCount >= chatLimitTriggerAt
|
||||
? createWeeklyLimitChatSendResponse(sendCount, chatLimitTriggerAt)
|
||||
: chatLimitTriggerAt != null
|
||||
? createChatSendResponse(sendCount)
|
||||
: chatSendResponse;
|
||||
|
||||
await route.fulfill({ json: apiEnvelope(response) });
|
||||
});
|
||||
|
||||
await page.route("**/api/payment/plans**", async (route) => {
|
||||
|
||||
@@ -89,6 +89,44 @@ export const chatSendResponse = {
|
||||
},
|
||||
};
|
||||
|
||||
export function createChatSendResponse(sendCount: number) {
|
||||
return {
|
||||
...chatSendResponse,
|
||||
reply: `Hello from the E2E boyfriend. Reply ${sendCount}.`,
|
||||
messageId: `msg_e2e_reply_${String(sendCount).padStart(3, "0")}`,
|
||||
timestamp: chatSendResponse.timestamp + sendCount,
|
||||
};
|
||||
}
|
||||
|
||||
export function createWeeklyLimitChatSendResponse(
|
||||
usedMessageCount: number,
|
||||
limit: number,
|
||||
) {
|
||||
return {
|
||||
reply: "",
|
||||
audioUrl: "",
|
||||
messageId: "",
|
||||
isGuest: true,
|
||||
timestamp: chatSendResponse.timestamp + usedMessageCount,
|
||||
image: {
|
||||
type: null,
|
||||
url: null,
|
||||
},
|
||||
lockDetail: {
|
||||
locked: true,
|
||||
showContent: false,
|
||||
showUpgrade: true,
|
||||
reason: "weekly_limit",
|
||||
hint: "Free message limit reached.",
|
||||
detail: {
|
||||
type: "weekly_msg_limit",
|
||||
usedThisWeek: usedMessageCount,
|
||||
limit,
|
||||
},
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
export const paymentPlansResponse = {
|
||||
plans: [
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user