66 lines
1.7 KiB
JavaScript
66 lines
1.7 KiB
JavaScript
|
const puppeteer = require("puppeteer");
|
||
|
|
const { spawn } = require("child_process");
|
||
|
|
const res = spawn("python", ["-m", "http.server", "8000"], { shell: true });
|
||
|
|
|
||
|
|
async function timeout(time) {
|
||
|
|
return new Promise((resolve) => setTimeout(resolve, time));
|
||
|
|
}
|
||
|
|
|
||
|
|
function cleanup(err) {
|
||
|
|
console.log("cleaning up");
|
||
|
|
res.kill();
|
||
|
|
if (err != null) {
|
||
|
|
console.error(err);
|
||
|
|
process.exit(1);
|
||
|
|
}
|
||
|
|
process.exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
async function waitForText(selector, text) {
|
||
|
|
let n = 0;
|
||
|
|
let ready = false;
|
||
|
|
while (n < 30) {
|
||
|
|
const res = await (await selector.getProperty("textContent")).jsonValue();
|
||
|
|
console.log(`waiting for text ${text} got ${res}`);
|
||
|
|
if (res == text) {
|
||
|
|
ready = true;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
await timeout(1000);
|
||
|
|
n += 1;
|
||
|
|
}
|
||
|
|
return ready;
|
||
|
|
}
|
||
|
|
|
||
|
|
async function runTest() {
|
||
|
|
const browser = await puppeteer.launch({
|
||
|
|
headless: false,
|
||
|
|
args: ["--enable-unsafe-webgpu"],
|
||
|
|
});
|
||
|
|
const page = await browser.newPage();
|
||
|
|
|
||
|
|
page
|
||
|
|
.on("console", (message) =>
|
||
|
|
console.log(`message from console ${message.text()}`),
|
||
|
|
)
|
||
|
|
.on("pageerror", ({ message }) =>
|
||
|
|
console.log(`error from page ${message}`),
|
||
|
|
);
|
||
|
|
|
||
|
|
const res = await page.goto("http://localhost:8000/examples/webgpu/efficientnet/index.html");
|
||
|
|
if (res.status() !== 200) throw new Error("Failed to load page");
|
||
|
|
|
||
|
|
const textSelector = await page.waitForSelector("#result");
|
||
|
|
const buttonSelector = await page.waitForSelector("input[type=button]");
|
||
|
|
const ready = await waitForText(textSelector, "ready");
|
||
|
|
if (!ready) throw new Error("Failed to load page");
|
||
|
|
|
||
|
|
await buttonSelector.evaluate((e) => e.click());
|
||
|
|
const done = await waitForText(textSelector, "hen");
|
||
|
|
if (!done) throw new Error("failed to get hen");
|
||
|
|
|
||
|
|
cleanup(null);
|
||
|
|
}
|
||
|
|
|
||
|
|
runTest().catch((err) => cleanup(err));
|