import re import pytest from playwright.async_api import async_playwright, expect from common.base_page_test import BasePageTest from common.config import Config class GenTableTest(BasePageTest): """代码生成业务测试""" async def test_gen_code_flow(self) -> None: """测试代码生成全流程：导入 -> 编辑 -> 预览 -> 删除""" table_name = 'sys_post' # 1. 导航到代码生成页面 await self.navigate_to_gen_page() # 确保环境清理 (如果存在 sys_post 则先删除) await self.ensure_table_not_exists(table_name) # 2. 导入表 await self.import_table(table_name) # 3. 编辑表 await self.edit_table(table_name, '测试生成') # 4. 预览代码 await self.preview_code(table_name) await self.page.wait_for_timeout(1000) # 5. 删除表 (清理环境) await self.delete_table(table_name) async def navigate_to_gen_page(self) -> None: """导航到代码生成页面""" await self.goto_page(Config.frontend_url + '/tool/gen') await self.wait_for_selector('.app-container') async def ensure_table_not_exists(self, table_name: str) -> None: """确保表不存在 (如果存在则删除)""" await self.delete_table(table_name, verify=False) async def search_table(self, table_name: str) -> None: """搜索表""" # 确保在主搜索表单中搜索 search_form = self.page.locator('.el-form').first await search_form.get_by_role('textbox', name='表名称').fill(table_name) await search_form.get_by_role('button', name='搜索').click() # 等待加载 loading = self.page.locator('.el-loading-mask') if await loading.count() > 0: await expect(loading.first).to_be_hidden(timeout=10000) await self.page.wait_for_timeout(300) async def import_table(self, table_name: str) -> None: """导入表""" await self.page.get_by_role('button', name='导入').click() dialog = self.page.locator("div[role='dialog'][aria-label='导入表']") await dialog.wait_for() # 搜索要导入的表 await dialog.get_by_placeholder('请输入表名称').fill(table_name) await dialog.get_by_role('button', name='搜索').click() # 等待搜索结果 await dialog.locator(f"tr:has-text('{table_name}')").wait_for() # 选中行 row = dialog.locator('tr').filter(has=self.page.get_by_text(table_name, exact=True)) # 点击复选框 checkbox = row.locator('.el-checkbox') await checkbox.click() # 验证已选中 # Element Plus checkbox 选中时，最外层 label.el-checkbox 会有 is-checked 类 await expect(checkbox).to_have_class(re.compile(r'is-checked')) await self.page.wait_for_timeout(500) await dialog.get_by_role('button', name='确 定').click() # 检查是否有"请选择要导入的表"错误 try: await expect(self.page.get_by_text('请选择要导入的表')).to_be_visible(timeout=2000) print("ERROR: Selection failed, '请选择要导入的表' appeared.") except AssertionError: pass # 等待一会，让弹窗自动关闭 await self.page.wait_for_timeout(5000) # 如果弹窗还在，尝试关闭它以免阻塞后续操作 if await dialog.is_visible(): print('WARNING: Import dialog still visible after timeout. Forcing close.') # Check for error messages if await self.page.locator('.el-message--error').count() > 0: msg = await self.page.locator('.el-message--error').all_inner_texts() print(f'ERROR MESSAGE: {msg}') # 点击取消关闭弹窗 await dialog.get_by_role('button', name='取 消').click() await expect(dialog).to_be_hidden() # 手动刷新列表 await self.page.get_by_role('button', name='搜索').click() # 验证导入成功 (搜索并在列表中看到) # 使用 .app-container 限定在主页面表格，避免匹配到弹窗中的隐藏行 await self.search_table(table_name) row = self.page.locator('.app-container .el-table__body-wrapper tbody tr').filter( has=self.page.get_by_text(table_name, exact=True) ) for _i in range(5): try: await expect(row.first).to_be_visible(timeout=3000) break except AssertionError: await self.search_table(table_name) await expect(row.first).to_be_visible(timeout=5000) async def edit_table(self, table_name: str, remark: str) -> None: """编辑表""" await self.search_table(table_name) row = self.page.locator(f"tbody tr:has-text('{table_name}')") # 点击编辑 (操作列第2个按钮，索引1) # 按钮顺序: 预览, 编辑, 删除, 同步, 生成 await row.locator('button').nth(1).click() # 等待编辑页面 (tab页) await self.page.wait_for_selector("div[role='tablist']") # 修改基本信息 -> 表描述 # 确保在基本信息 Tab await self.page.get_by_text('基本信息').click() await self.page.get_by_role('textbox', name='表描述').fill(remark) # 提交 await self.page.get_by_role('button', name='提交').click() # 验证回到列表 await self.wait_for_selector('.app-container') # 验证描述已更新 await self.search_table(table_name) await self.wait_for_selector(f"tbody tr:has-text('{remark}')") async def preview_code(self, table_name: str) -> None: """预览代码""" await self.search_table(table_name) row = self.page.locator(f"tbody tr:has-text('{table_name}')") # 点击预览 (操作列第1个按钮，索引0) await row.locator('button').nth(0).click() # 等待预览弹窗 dialog = self.page.locator("div[role='dialog'][aria-label='代码预览']") await dialog.wait_for() # 验证存在代码内容 # pre 可能有多个（多tab），只检查可见的 await dialog.locator('pre:visible').first.wait_for() content = await dialog.locator('pre:visible').first.text_content() assert 'class' in content or 'import' in content or 'package' in content # 关闭预览 (点击右上角关闭按钮) await dialog.locator('.el-dialog__headerbtn').click() async def delete_table(self, table_name: str, verify: bool = True) -> None: """删除表 (支持删除多条重复数据)""" # 循环删除直到不存在 for _i in range(5): await self.search_table(table_name) # 使用 strict matching # 限制在 .app-container .el-table__body-wrapper 以避免固定列导致的重复 以及 避免匹配到弹窗中的行 row = self.page.locator('.app-container .el-table__body-wrapper tbody tr').filter( has=self.page.get_by_text(table_name, exact=True) ) count = await row.count() if count == 0: break # 针对第一行操作 # 使用 force=True 确保点击，防止遮挡 btns = row.first.locator('button') await btns.nth(2).click(force=True) # 处理确认弹窗 await self.page.get_by_role('button', name='确定').click() # 等待删除成功提示 # 使用 specific selector 避免匹配到代码预览中的文本 await expect(self.page.locator('.el-message__content').filter(has_text='删除成功')).to_be_visible( timeout=5000 ) # 等待提示消失，防止干扰下一次操作 await expect(self.page.locator('.el-message__content').filter(has_text='删除成功')).to_be_hidden( timeout=5000 ) if verify: # 验证删除成功 # 重新搜索验证不存在 await self.search_table(table_name) # 使用 strict matching row = self.page.locator('.app-container .el-table__body-wrapper tbody tr').filter( has=self.page.get_by_text(table_name, exact=True) ) # 期望找不到或者 count为0 await expect(row).to_have_count(0) @pytest.mark.asyncio async def test_gen_table_page() -> None: """测试代码生成页面功能""" async with async_playwright() as p: test_instance = GenTableTest() await test_instance.setup(p) try: await test_instance.test_gen_code_flow() finally: await test_instance.teardown()