File size: 3,132 Bytes
46dc982 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import {app} from "../../../scripts/app.js";
import {api} from "../../../scripts/api.js";
let watched_nodes = {}
let resolve = undefined
let testURL = api.apiURL("/VHS_test")
let errors = []
api.addEventListener("executed", async function ({detail}) {
if (watched_nodes && watched_nodes[detail?.node]) {
if (detail?.output?.unfinished_batch) {
return
}
let requestBody = {tests: watched_nodes[detail.node], output: detail.output}
try {
let req = await fetch(api.apiURL("/VHS_test"),
{method: "POST", body: JSON.stringify(requestBody)});
let testResult = await req.json()
if (testResult.length != 0) {
errors.push(testResult)
}
} catch(e) {
errors.push(e)
}
if (!(watched_nodes.length -= 1)) {
resolve()
}
}
});
const workflowService = app.extensionManager.workflow
async function runTest(file) {
if (!file?.name?.endsWith(".json")) {
return false
}
let workflow = JSON.parse(await file.text())
await app.loadGraphData(workflow)
//NOTE: API is not used so workflow data is actually processed
watched_nodes = workflow.tests
errors = []
let p = new Promise((r) => resolve = r)
await app.queuePrompt()
//block until execution completes
await p
watched_nodes = {}
if (errors.length > 0) {
app.ui.dialog.show("Failed " + errors.length + " tests:\n" + errors)
return true
}
await workflowService.closeWorkflow(workflowService.activeWorkflow, {warnIfUnsaved: false})
return false
}
let iconOverride = document.createElement("style")
iconOverride.innerHTML = `.VHSTestIcon:before {content: '🧪';}`
document.body.append(iconOverride)
let testSidebar = {id: 'VHStest', title: 'VHS Test', icon: 'VHSTestIcon', type: 'custom',
render: (e) => {
e.innerHTML = `Select a folder containing tests
<input>
Or select a single test
<input>
`
const folderInput = e.children[0]
const fileInput = e.children[1]
Object.assign(folderInput, {
type: "file",
webkitdirectory: true,
onchange: async function() {
const startTime = Date.now()
let failedTests = false
for(const file of this.files) {
failedTests ||= await runTest(file)
}
this.value=""
if (!failedTests) {
console.log("All tests passed in " + ((Date.now() - startTime)/1000) + "s")
}
},
});
Object.assign(fileInput, {
type: "file",
accept: ".json",
onchange: async function() {
if (this.files.length) {
if(!(await runTest(this.files[0]))) {
console.log("Test complete")
}
this.value=""
}
},
});
}}
app.extensionManager.registerSidebarTab(testSidebar)
|