001code-html--cocos/scratch-vm/test/integration/tw_rejected_promise.js

const {test} = require('tap');
const fs = require('fs');
const path = require('path');
const VM = require('../../src/virtual-machine');
const Scratch = require('../../src/extension-support/tw-extension-api-common');

const commandFixture = fs.readFileSync(path.join(__dirname, '../fixtures/tw-rejected-promise-command.sb3'));
const reporterFixture = fs.readFileSync(path.join(__dirname, '../fixtures/tw-rejected-promise-reporter.sb3'));

class TestExtension {
    getInfo () {
        return {
            id: 'test123',
            name: 'test123',
            blocks: [
                {
                    blockType: Scratch.BlockType.COMMAND,
                    opcode: 'command',
                    text: 'return rejected promise'
                },
                {
                    blockType: Scratch.BlockType.REPORTER,
                    opcode: 'reporter',
                    text: 'return rejected promise'
                }
            ]
        };
    }
    command () {
        return Promise.reject(new Error('Test error 1'));
    }
    reporter () {
        return Promise.reject(new Error('Test error 2'));
    }
}

for (const enableCompiler of [true, false]) {
    test(`COMMAND returns rejected promise - ${enableCompiler ? 'compiler' : 'interpreter'}`, t => {
        const vm = new VM();
        vm.extensionManager.addBuiltinExtension('test123', TestExtension);

        vm.setCompilerOptions({
            enabled: enableCompiler
        });
        t.equal(vm.runtime.compilerOptions.enabled, enableCompiler);

        vm.loadProject(commandFixture).then(async () => {
            vm.greenFlag();

            for (let i = 0; i < 12; i++) {
                vm.runtime._step();
    
                // wait for promise rejection to be handled
                await Promise.resolve();
            }

            const stage = vm.runtime.getTargetForStage();
            t.equal(stage.lookupVariableByNameAndType('before', '').value, 10);
            t.equal(stage.lookupVariableByNameAndType('after', '').value, 10);
            t.end();
        });
    });

    test(`REPORTER returns rejected promise - ${enableCompiler ? 'compiler' : 'interpreter'}`, t => {
        const vm = new VM();
        vm.extensionManager.addBuiltinExtension('test123', TestExtension);

        vm.setCompilerOptions({
            enabled: enableCompiler
        });
        t.equal(vm.runtime.compilerOptions.enabled, enableCompiler);

        vm.loadProject(reporterFixture).then(async () => {
            vm.greenFlag();

            for (let i = 0; i < 12; i++) {
                vm.runtime._step();
    
                // wait for promise rejection to be handled
                await Promise.resolve();
            }

            const stage = vm.runtime.getTargetForStage();
            t.equal(stage.lookupVariableByNameAndType('before', '').value, 10);
            t.equal(stage.lookupVariableByNameAndType('after', '').value, 10);
            t.same(stage.lookupVariableByNameAndType('values', 'list').value, [
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2',
                'Error: Test error 2'
            ]);
            t.end();
        });
    });
}