diff --git a/examples/quick-start-llm-python/tests/e2e/ui.spec.js b/examples/quick-start-llm-python/tests/e2e/ui.spec.js index 67d8e3cfc7..5404a07110 100644 --- a/examples/quick-start-llm-python/tests/e2e/ui.spec.js +++ b/examples/quick-start-llm-python/tests/e2e/ui.spec.js @@ -6,12 +6,6 @@ const chatgptTraceBasedTest = require('./definitions/chatgpt'); const { runTracebasedTest } = require('./tracetest'); -const timeToWait = 10_000; // 10 seconds - -function sleep(ms) { - return new Promise(resolve => setTimeout(resolve, ms)); -} - test('generated summarized test for Gemini', async ({ page }) => { // Go to Streamlit app await page.goto('http://localhost:8501/'); @@ -25,14 +19,17 @@ test('generated summarized test for Gemini', async ({ page }) => { // Click on button to call summarization rule await page.getByRole('button', { name: 'Summarize' }).click(); - // Wait for time - await sleep(timeToWait); + // Wait link to appear + await page.getByText('Trace ID').waitFor({ state: 'visible' }); // Capture TraceID - const traceIDLabel = await page.getByRole('link', { name: 'Trace ID' }); - expect(traceIDLabel).toHaveText('Trace ID'); + const traceIDElement = await page.getByRole('link'); + console.log(traceIDElement.innerHTML()); + + // const traceIDLabel = await page.getByRole('link', { name: 'Trace ID' }); + // expect(traceIDLabel).toHaveText('Trace ID'); - console.log(traceIDLabel); + // console.log(traceIDLabel); // const traceID = (traceIDLabel || '').replace('Trace ID:', '').trim(); diff --git a/examples/quick-start-llm-python/tests/playwright.config.js b/examples/quick-start-llm-python/tests/playwright.config.js index 878d75cc7a..7bc2de9f66 100644 --- a/examples/quick-start-llm-python/tests/playwright.config.js +++ b/examples/quick-start-llm-python/tests/playwright.config.js @@ -24,7 +24,8 @@ module.exports = defineConfig({ /* Opt out of parallel tests on CI. */ workers: process.env.CI ? 1 : undefined, /* Reporter to use. See https://playwright.dev/docs/test-reporters */ - reporter: 'html', + // reporter: 'html', + reporter: 'list', /* Shared settings for all the projects below. See https://playwright.dev/docs/api/class-testoptions. */ use: { /* Base URL to use in actions like `await page.goto('/')`. */