leporello-js/test/utils.js

import {parse, print_debug_node, load_modules} from '../src/parse_js.js'
import {eval_tree, eval_frame} from '../src/eval.js'
import {COMMANDS} from '../src/cmd.js'

Object.assign(globalThis, {log: console.log})

export const parse_modules = (entry, modules) => 
  load_modules(entry, module_name => modules[module_name])

export const assert_code_evals_to = (codestring, expected) => {
  const parse_result = parse(codestring)
  assert_equal(parse_result.ok, true)
  const tree = eval_tree(parse_result.node)
  const frame = eval_frame(tree)
  const result = frame.children[frame.children.length - 1].result
  assert_equal({ok: result.ok, value: result.value}, {ok: true, value: expected})
  return frame
}

export const assert_code_error = (codestring, error) => {
  const parse_result = parse(codestring)
  assert_equal(parse_result.ok, true)
  const tree = eval_tree(parse_result.node)
  const frame = eval_frame(tree)
  const result = frame.children[frame.children.length - 1].result
  assert_equal(result.ok, false)
  assert_equal(result.error, error)
}

export const test_initial_state = (code, state) => {
  return COMMANDS.get_initial_state(
    {
      ...state,
      files: typeof(code) == 'object' ? code : { '' : code},
      entrypoint: '',
      current_module: '',
    },
  )
}

export const stringify = val => 
  JSON.stringify(val, (key, value) => {
    // TODO do not use instanceof because currently not implemented in parser
    if(value?.constructor == Set){
      return [...value]
    } else {
      return value
    }
  }, 2)

export const assert_equal = (exp, actual) => {
  if(typeof(exp) == 'object' && typeof(actual) == 'object'){
    const exp_json = stringify(exp)
    const act_json = stringify(actual)
    if(exp_json != act_json){
      throw new Error(`FAIL: ${exp_json} != ${act_json}`)
    }
  } else {
    if(exp != actual){
      throw new Error(`FAIL: ${exp} != ${actual}`)
    } 
  }
}

export const print_debug_ct_node = node => {
  const do_print = node => {
    const {id, fn, ok, value, error, args, has_more_children} = node
    const res = {id, fn: fn?.name, ok, value, error, args, has_more_children}
    if(node.children == null) {
      return res
    } else {
      const next_children = node.children.map(do_print)
      return {...res, children: next_children}
    }
  }
  return stringify(do_print(node))
}

export const test = (message, test, only = false) => {
  return {
    message, 
    test: Object.defineProperty(test, 'name', {value: message}),
    only,
  }
}

export const test_only = (message, t) => test(message, t, true)

// Wrap to Function constructor to hide from calltree view
// TODO in calltree view, hide fn which has special flag set (see
// filter_calltree)
export const run = Object.defineProperty(new Function('tests', `
    const run_test = t => {
      try {
        t.test()
      } catch(e) {
        if(globalThis.process != null) {
          // In node.js runner, fail fast
          console.error('Failed: ' + t.message)
          throw e
        } else {
          return e
        }
      }
    }

    // If not run in node, then dont apply filter
    const filter = globalThis.process && globalThis.process.argv[2]

    if(filter == null) {

      const only = tests.find(t => t.only)
      const tests_to_run = only == null ? tests : [only]

      // Exec each test. After all tests are done, we rethrow first failer if
      // any. So we will mark root calltree node if one of tests failed
      const failure = tests_to_run.reduce(
        (failure, t) => {
          const next_failure = run_test(t)
          return failure ?? next_failure
        },
        null
      )

      if(failure != null) {
        throw failure
      } else {
        if(globalThis.process != null) {
          console.log('Ok')
        }
      }

    } else {
      const test = tests.find(t => t.message.includes(filter))
      if(test == null) {
        throw new Error('test not found')
      } else {
        run_test(test)
        if(globalThis.process != null) {
          console.log('Ok')
        }
      }
    }
`), 'name', {value: 'run'})