File size: 1,392 Bytes
867b17d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import { expect as baseExpect, test as baseTest } from '@playwright/test';
import { createAuthenticatedContext, type UserContext } from './helpers';
import { getUnixTime } from 'date-fns';
interface Fixtures {
adaContext: UserContext;
babbageContext: UserContext;
curieContext: UserContext;
}
export const test = baseTest.extend<{}, Fixtures>({
adaContext: [
async ({ browser }, use, workerInfo) => {
const ada = await createAuthenticatedContext({
browser,
name: `ada-${workerInfo.workerIndex}-${getUnixTime(new Date())}`,
});
await use(ada);
await ada.context.close();
},
{ scope: 'worker' },
],
babbageContext: [
async ({ browser }, use, workerInfo) => {
const babbage = await createAuthenticatedContext({
browser,
name: `babbage-${workerInfo.workerIndex}-${getUnixTime(new Date())}`,
});
await use(babbage);
await babbage.context.close();
},
{ scope: 'worker' },
],
curieContext: [
async ({ browser }, use, workerInfo) => {
const curie = await createAuthenticatedContext({
browser,
name: `curie-${workerInfo.workerIndex}-${getUnixTime(new Date())}`,
chatModel: 'chat-model-reasoning',
});
await use(curie);
await curie.context.close();
},
{ scope: 'worker' },
],
});
export const expect = baseExpect;
|