feat(e2e): support for custom data when using mockRecords

This commit is contained in:
Zeke Zhang 2023-12-01 14:58:45 +08:00
parent bfeaf456b9
commit 6fc644d506

View File

@ -173,7 +173,18 @@ const _test = base.extend<{
mockCollections: <T = any>(collectionSettings: CollectionSetting[]) => Promise<T>; mockCollections: <T = any>(collectionSettings: CollectionSetting[]) => Promise<T>;
mockCollection: <T = any>(collectionSetting: CollectionSetting) => Promise<T>; mockCollection: <T = any>(collectionSetting: CollectionSetting) => Promise<T>;
mockRecord: <T = any>(collectionName: string, data?: any) => Promise<T>; mockRecord: <T = any>(collectionName: string, data?: any) => Promise<T>;
mockRecords: <T = any>(collectionName: string, count?: number, data?: any) => Promise<T[]>; mockRecords: {
/**
* @param collectionName -
* @param count -
*/
<T = any>(collectionName: string, count?: number): Promise<T[]>;
/**
* @param collectionName -
* @param data -
*/
<T = any>(collectionName: string, data?: any[]): Promise<T[]>;
};
createCollections: (collectionSettings: CollectionSetting | CollectionSetting[]) => Promise<void>; createCollections: (collectionSettings: CollectionSetting | CollectionSetting[]) => Promise<void>;
}>({ }>({
mockPage: async ({ page }, use) => { mockPage: async ({ page }, use) => {
@ -242,7 +253,11 @@ const _test = base.extend<{
} }
}, },
mockRecords: async ({ page }, use) => { mockRecords: async ({ page }, use) => {
const mockRecords = async (collectionName: string, count = 3, data?: any) => { const mockRecords = async (collectionName: string, count: any = 3, data?: any) => {
if (_.isArray(count)) {
data = count;
count = data.length;
}
return createRandomData(collectionName, count, data); return createRandomData(collectionName, count, data);
}; };