HHY
ab60cac87c
- pillow>=12.3.0 依赖入库(支撑 AI 生成图持久化处理) - e2e 辅助脚本: _e2e_opt/_e2e_upload/_rec_ui - 巡检/审计/多结果/记录抽屉测试截图归档 shots/ - 远程仓库迁移至 git.djxx.club/hanzhicheng/menchuang
35 lines
1.7 KiB
JavaScript
35 lines
1.7 KiB
JavaScript
import puppeteer from 'puppeteer-core'
|
|
import path from 'path'
|
|
const browser = await puppeteer.launch({
|
|
executablePath: 'C:/Program Files (x86)/Microsoft/Edge/Application/msedge.exe',
|
|
headless: 'new', args: ['--window-size=1440,900', '--enable-unsafe-swiftshader'],
|
|
defaultViewport: { width: 1440, height: 900 },
|
|
})
|
|
const page = await browser.newPage()
|
|
const errs = []
|
|
page.on('pageerror', e => errs.push(String(e).slice(0, 120)))
|
|
const token = process.argv[2]
|
|
const sketch = process.argv[3]
|
|
await page.goto(`http://localhost:8000/ai_render?token=${encodeURIComponent(token)}`, { waitUntil: 'networkidle2' })
|
|
await new Promise(r => setTimeout(r, 2000))
|
|
await page.evaluate(() => {
|
|
[...document.querySelectorAll('.el-radio-button')].find(b => b.textContent.includes('工程大样图'))?.querySelector('input')?.click()
|
|
})
|
|
await new Promise(r => setTimeout(r, 800))
|
|
const input = await page.$('.el-upload input[type=file]')
|
|
await input.uploadFile(path.resolve(sketch))
|
|
await new Promise(r => setTimeout(r, 1200))
|
|
await page.evaluate(() => { [...document.querySelectorAll('button')].find(b => b.textContent.includes('AI 生成大样图'))?.click() })
|
|
let done = false
|
|
for (let i = 0; i < 110; i++) {
|
|
await new Promise(r => setTimeout(r, 2000))
|
|
done = await page.evaluate(() => !!document.querySelector('.bp-compare'))
|
|
if (done) break
|
|
}
|
|
// 打开生成记录抽屉
|
|
await page.evaluate(() => { [...document.querySelectorAll('button')].find(b => b.textContent.includes('生成记录'))?.click() })
|
|
await new Promise(r => setTimeout(r, 2000))
|
|
await page.screenshot({ path: '../shots/rec-drawer.png' })
|
|
console.log(done ? '生成+记录抽屉 OK' : '生成超时', '| 异常:', errs.length ? errs.join('|') : '无')
|
|
await browser.close()
|