fix(e2e): replace networkidle with load + element wait, wait for diagram text before assert

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: alswl

tests/e2e/editor.spec.ts

@@ -3,7 +3,8 @@ import { expect, test } from '@playwright/test';
 test.describe('DBML Editor - Basic Features', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
   });
 
   test('should load homepage', async ({ page }) => {
@@ -42,7 +43,8 @@ test.describe('DBML Editor - Basic Features', () => {
 test.describe('DBML Editor - Editing Features', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
   });
 
   test('should edit DBML code', async ({ page }) => {
@@ -97,7 +99,8 @@ Table orders {
 test.describe('DBML Editor - Import Features', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
   });
 
   test('should open import dialog', async ({ page }) => {
@@ -154,7 +157,8 @@ test.describe('DBML Editor - Import Features', () => {
 test.describe('DBML Editor - Export Features', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
   });
 
   test('should open export dialog', async ({ page }) => {
@@ -208,7 +212,8 @@ test.describe('DBML Editor - Export Features', () => {
 test.describe('DBML Editor - Error Handling', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
   });
 
   test('should display syntax error message', async ({ page }) => {

tests/e2e/er-diagram.spec.ts

@@ -3,7 +3,8 @@ import { expect, test } from '@playwright/test';
 test.describe('ER Diagram - Interaction', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
     await page.waitForSelector('svg', { timeout: 10000 });
   });
 
@@ -21,6 +22,12 @@ test.describe('ER Diagram - Interaction', () => {
   });
 
   test('should display table names', async ({ page }) => {
+    // 等待 ER 图节点和文字渲染完成（layout 与 X6 渲染为异步）
+    await expect(page.locator('svg g[data-cell-id]').first()).toBeVisible({
+      timeout: 15000,
+    });
+    await expect(page.locator('svg text').first()).toBeVisible({ timeout: 15000 });
+
     const tableNames = page.locator('svg text');
     const textCount = await tableNames.count();
     expect(textCount).toBeGreaterThan(0);
@@ -106,7 +113,8 @@ Ref: posts.user_id > users.id
 test.describe('ER Diagram - Layout', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
     await page.waitForSelector('svg', { timeout: 10000 });
   });
 
@@ -178,7 +186,8 @@ Table products {
 test.describe('ER Diagram - Visual', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
     await page.waitForSelector('svg', { timeout: 10000 });
     await page.waitForTimeout(1000);
   });
@@ -194,6 +203,7 @@ test.describe('ER Diagram - Visual', () => {
     const rectCount = await rects.count();
     expect(rectCount).toBeGreaterThan(0);
 
+    await expect(page.locator('svg text').first()).toBeVisible({ timeout: 15000 });
     const texts = page.locator('svg text');
     const textCount = await texts.count();
     expect(textCount).toBeGreaterThan(0);

tests/e2e/export.spec.ts

@@ -3,7 +3,8 @@ import { expect, test } from '@playwright/test';
 test.describe('Export - SQL Export Functionality', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
   });
 
   test('should open export modal', async ({ page }) => {

tests/e2e/toolbar.spec.ts

@@ -3,7 +3,8 @@ import { expect, test } from '@playwright/test';
 test.describe('Toolbar - Zoom Controls', () => {
   test.beforeEach(async ({ page }) => {
     await page.goto('/');
-    await page.waitForLoadState('networkidle');
+    await page.waitForLoadState('load');
+    await expect(page.locator('.react-shape-app')).toBeVisible({ timeout: 15000 });
     await page.waitForSelector('.zoom-toolbar', { timeout: 10000 });
     await page.waitForTimeout(2000); // Wait for initial render
   });