Hugging Face's logo

Spaces:
yijin928
/
Test
Running on Zero

App Files Community
1
Test
File size: 3,074 Bytes 
b4e8d50
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
 

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()
        }
    }
});

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
    }
    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)