ueblueprint/tests/nodes.spec.js

import { test, expect } from "./fixtures/test.js"
import Configuration from "./../js/Configuration.js"
import EventNodes from "./resources/EventNodes.js"
import FlowControlNodes from "./resources/FlowControlNodes.js"
import InputNodes from "./resources/InputNodes.js"
import LegacyNodes from "./resources/LegacyNodes.js"
import MaterialNodes from "./resources/MaterialNodes.js"
import OperationsNodes from "./resources/OperationsNodes.js"
import OtherNodes from "./resources/OtherNodes.js"
import PCGNodes from "./resources/PCGNodes.js"

const nodeTests = [
    ...EventNodes.get(),
    ...FlowControlNodes.get(),
    ...InputNodes.get(),
    ...LegacyNodes.get(),
    ...MaterialNodes.get(),
    ...OperationsNodes.get(),
    ...OtherNodes.get(),
    ...PCGNodes.get()
]

/** @param {String[]} words */
function getFirstWordOrder(words) {
    return new RegExp(/\s*/.source + words.join(/[^\n]+\n\s*/.source) + /\s*/.source)
}

for (const nodeTest of nodeTests) {
    test.describe(nodeTest.name, () => {

        test.beforeAll(async ({ blueprintPage }) => {
            await blueprintPage.removeNodes()
            await blueprintPage.paste(nodeTest.value)
        })

        if (nodeTest.color) {
            test(
                `${nodeTest.name}: Has correct color`,
                async ({ blueprintPage }) => {
                    expect(
                        await blueprintPage.node.evaluate(node => node.entity.nodeColor().toString())
                    ).toBe(nodeTest.color.toString())
                }
            )
        }
        test(
            `${nodeTest.name}: Has correct delegate`,
            async ({ blueprintPage }) => {
                const delegate = blueprintPage.blueprintLocator.locator(
                    'ueb-node .ueb-node-top ueb-pin[data-type="delegate"]'
                )
                if (nodeTest.delegate) {
                    await expect(delegate).toBeVisible()
                } else {
                    await expect(delegate).toBeHidden()
                }
            }
        )
        if (nodeTest.title) {
            test(
                `${nodeTest.name}: Has title ${nodeTest.title}`,
                async ({ blueprintPage }) => expect(
                    await blueprintPage.node.evaluate(node => node.getNodeDisplayName())
                ).toBe(nodeTest.title)
            )
        }
        if (nodeTest.subtitle) {
            test(
                `${nodeTest.name}: Has expected subtitle ${nodeTest.subtitle}`,
                async ({ blueprintPage }) => await expect(blueprintPage.node.locator(".ueb-node-subtitle-text"))
                    .toHaveText(nodeTest.subtitle, { useInnerText: true })
            )
        }
        if (nodeTest.size) {
            test(
                `${nodeTest.name}: Has approximately the expected size`,
                async ({ blueprintPage }) => {
                    const expectedSize = await blueprintPage.node.evaluate(
                        (node, gridSize) => {
                            const bounding = node.getBoundingClientRect()
                            const expectedSize = [bounding.width / gridSize, bounding.height / gridSize]
                            return expectedSize
                        },
                        Configuration.gridSize
                    )
                    expect(Math.abs(nodeTest.size[0] - expectedSize[0])).toBeLessThan(1.5)
                    expect(Math.abs(nodeTest.size[1] - expectedSize[1])).toBeLessThan(1.5)
                    if (
                        Math.abs(nodeTest.size[0] - expectedSize[0]) > 0.6
                        || Math.abs(nodeTest.size[1] - expectedSize[1]) > 0.6
                    ) {
                        console.error(`Node "${nodeTest.name}" size does not match`)
                    }
                }
            )
        }
        if (nodeTest.icon !== undefined) {
            test(
                `${nodeTest.name}: Has the correct icon`,
                async ({ blueprintPage }) => expect(
                    await blueprintPage.node.evaluate(
                        node => node.entity.nodeIcon()?.strings.join("")
                    )
                ).toBe(nodeTest.icon?.strings.join(""))
            )
        }
        if (nodeTest.pins !== undefined) {
            test(
                `${nodeTest.name}: Has ${nodeTest.pins} pins`,
                async ({ blueprintPage }) => expect(
                    await blueprintPage.node.evaluate(
                        node => node.querySelectorAll("ueb-pin").length
                    )
                ).toBe(nodeTest.pins)
            )
        }
        if (nodeTest.pinNames) {
            test(
                `${nodeTest.name}: Has correct pin names`,
                async ({ blueprintPage }) => {
                    const innerTexts = await blueprintPage.node.locator(".ueb-pin-content .ueb-pin-name").allInnerTexts()
                    const pinNames = innerTexts.map(v => v.trim()).filter(v => v.length > 0)
                    expect(pinNames).toStrictEqual(nodeTest.pinNames)
                }
            )
        }
        test(
            `${nodeTest.name}: Expected development`,
            async ({ blueprintPage }) => expect(
                await blueprintPage.node.evaluate(node => node.entity.isDevelopmentOnly())
            ).toBe(nodeTest.development)
        )
        test(
            `${nodeTest.name}: Maintains the order of attributes`,
            async ({ blueprintPage }) => {
                const value = await blueprintPage.blueprintLocator.evaluate(blueprint => {
                    blueprint.selectAll()
                    return blueprint.template.getCopyInputObject().getSerializedText()
                })
                const words = value
                    .split("\n")
                    .map(row => row.match(/\s*("?\w+(\s+\w+)*).+/)?.[1])
                    .filter(v => v?.length > 0)
                expect(value).toMatch(getFirstWordOrder(words))
            }
        )
        if (nodeTest.variadic) {
            test(
                `${nodeTest.name}: Can add new pins`,
                async ({ blueprintPage }) => {
                    const variadic = blueprintPage.node.getByText("Add pin")
                    await expect(variadic).toBeVisible()
                    await variadic.click()
                    expect(await blueprintPage.node.locator("ueb-pin").all()).toHaveLength(nodeTest.pins + 1)
                }
            )
        }
        if (nodeTest.additionalTest) {
            test(
                `${nodeTest.name}: Additional tests`,
                async ({ blueprintPage }) =>
                    nodeTest.additionalTest(blueprintPage.node, await blueprintPage.node.locator("ueb-pin").all())
            )
        }
    })
}