pybricks-code/src/mpy/sagas.test.ts

// SPDX-License-Identifier: MIT
// Copyright (c) 2020-2022 The Pybricks Authors

import { AsyncSaga } from '../../test';
import { compile, didCompile, didFailToCompile } from './actions';
import mpy from './sagas';

test('compiler works', async () => {
    const saga = new AsyncSaga(mpy);

    saga.put(compile('print("hello!")', 6, []));

    const action = await saga.take();
    expect(didCompile.matches(action)).toBeTruthy();
    const { data } = action as ReturnType<typeof didCompile>;
    expect(data[0]).toBe('M'.charCodeAt(0));
    expect(data[1]).toBe(6); // ABI version
    expect(data[2]).toBe(0); // flags
    expect(data[3]).toBe(31); // small int bits
});

test('compiler error works', async () => {
    const saga = new AsyncSaga(mpy);

    saga.put(compile('syntax error!', 6, []));

    const action = await saga.take();
    expect(didFailToCompile.matches(action)).toBeTruthy();
    const { err } = action as ReturnType<typeof didFailToCompile>;
    expect(err).toMatchInlineSnapshot(`
        [
          "Traceback (most recent call last):",
          "  File "main.py", line 1",
          "SyntaxError: invalid syntax",
        ]
    `);

    await saga.end();
});