arepl-backend/pythonExecutors.test.ts

JS interface to python evaluator for AREPL

/*global suite, test*/ //comment for eslint

// This test uses TDD Mocha. see https://mochajs.org/ for help
// http://ricostacruz.com/cheatsheets/mocha-tdd

// The module 'assert' provides assertion methods from node
import * as assert from 'assert'

import { PythonExecutors } from './pythonExecutors'

suite("PythonExecutors", () => {
    let pyExecutors = new PythonExecutors()
    let input = {
        evalCode: "",
        filePath: "",
        usePreviousVariables: false,
        show_global_vars: true,
        default_filter_vars: [],
        default_filter_types: ["<class 'module'>", "<class 'function'>"]
    }
    const pythonStartupTime = 3000
    const num_executors = 2

    suiteSetup(function () {
        this.timeout(pythonStartupTime + 500)
    })

    setup(function (done) {
        pyExecutors.onPrint = () => { }
        pyExecutors.onStderr = () => { }
        pyExecutors.onResult = () => { }
        pyExecutors.start(num_executors)
        done()
    })

    teardown(function(){
        pyExecutors.stop(true)
    })

    test("can do multiple executions", function (done) {
        // we do three test runs because given that only two executors exist:
        // if first fails: something is wrong with first executor
        // if second fails: something is wrong with second executor
        // if thid fails: logic that waits for a executor to become free is broken
        let num_results = 0
        pyExecutors.onResult = (result) => {
            num_results+=1
            if(num_results == 1){
                assert.strictEqual(result.userVariables['x'], 1)
                input.evalCode = "x=2"
                pyExecutors.execCode(input)
            }
            else if(num_results == 2){
                assert.strictEqual(result.userVariables['x'], 2)
                input.evalCode = "x=3"
                pyExecutors.execCode(input)
            }
            else if(num_results > num_executors){
                assert.strictEqual(result.userVariables['x'], 3)
                done()
            }
        }
        input.evalCode = "x=1"
        pyExecutors.execCode(input)
    })

    test("last execution takes precedence", function (done) {
        pyExecutors.onResult = (result) => {
            assert.strictEqual(result.userVariables['x'], 2)
            done()
        }
        input.evalCode = "x=1"
        pyExecutors.execCode(input)
        input.evalCode = "x=2"
        pyExecutors.execCode(input)
    })

})