squiggle/packages/squiggle-lang/__tests__/Reducer/Reducer_TestHelpers.res

module Expression = Reducer.Expression
module ExpressionValue = ReducerInterface.ExpressionValue
module ErrorValue = Reducer_ErrorValue

open Jest
open Expect

let unwrapRecord = rValue =>
  rValue->Belt.Result.flatMap(value => switch value {
  | ExpressionValue.EvRecord(aRecord) => Ok(aRecord)
  | _ => ErrorValue.RETodo("TODO: External bindings must be returned")->Error
  }
)

let expectParseToBe = (expr: string, answer: string) =>
  Reducer.parse(expr)->Expression.toStringResult->expect->toBe(answer)

let expectParseOuterToBe = (expr: string, answer: string) =>
  Reducer.parseOuter(expr)->Expression.toStringResult->expect->toBe(answer)

let expectParsePartialToBe = (expr: string, answer: string) =>
  Reducer.parsePartial(expr)->Expression.toStringResult->expect->toBe(answer)

let expectEvalToBe = (expr: string, answer: string) =>
  Reducer.evaluate(expr)->ExpressionValue.toStringResult->expect->toBe(answer)

let expectEvalBindingsToBe = (expr: string, bindings: Reducer.externalBindings, answer: string) =>
  Reducer.evaluateUsingOptions(expr, ~externalBindings=Some(bindings), ~isPartial=None, ~environment=None)
  ->ExpressionValue.toStringResult
  ->expect
  ->toBe(answer)

let expectEvalPartialBindingsToBe = (
  expr: string,
  bindings: Reducer.externalBindings,
  answer: string,
) =>
  Reducer.evaluateUsingOptions(expr, ~externalBindings=Some(bindings), ~isPartial=Some(true), ~environment=None)
  ->unwrapRecord
  ->ExpressionValue.toStringResultRecord
  ->expect
  ->toBe(answer)

let testParseToBe = (expr, answer) => test(expr, () => expectParseToBe(expr, answer))
let testParseOuterToBe = (expr, answer) => test(expr, () => expectParseOuterToBe(expr, answer))
let testParsePartialToBe = (expr, answer) => test(expr, () => expectParsePartialToBe(expr, answer))
let testDescriptionParseToBe = (desc, expr, answer) =>
  test(desc, () => expectParseToBe(expr, answer))

let testEvalToBe = (expr, answer) => test(expr, () => expectEvalToBe(expr, answer))
let testDescriptionEvalToBe = (desc, expr, answer) => test(desc, () => expectEvalToBe(expr, answer))
let testEvalBindingsToBe = (expr, bindingsList, answer) =>
  test(expr, () => expectEvalBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))
let testEvalPartialBindingsToBe = (expr, bindingsList, answer) =>
  test(expr, () => expectEvalPartialBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))

module MySkip = {
  let testParseToBe = (expr, answer) => Skip.test(expr, () => expectParseToBe(expr, answer))
  let testParseOuterToBe = (expr, answer) =>
    Skip.test(expr, () => expectParseOuterToBe(expr, answer))
  let testParsePartialToBe = (expr, answer) =>
    Skip.test(expr, () => expectParsePartialToBe(expr, answer))
  let testEvalToBe = (expr, answer) => Skip.test(expr, () => expectEvalToBe(expr, answer))
  let testEvalBindingsToBe = (expr, bindingsList, answer) =>
    Skip.test(expr, () => expectEvalBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))
  let testEvalPartialBindingsToBe = (expr, bindingsList, answer) =>
    Skip.test(expr, () =>
      expectEvalPartialBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer)
    )
}
module MyOnly = {
  let testParseToBe = (expr, answer) => Only.test(expr, () => expectParseToBe(expr, answer))
  let testParseOuterToBe = (expr, answer) =>
    Only.test(expr, () => expectParseOuterToBe(expr, answer))
  let testParsePartialToBe = (expr, answer) =>
    Only.test(expr, () => expectParsePartialToBe(expr, answer))
  let testEvalToBe = (expr, answer) => Only.test(expr, () => expectEvalToBe(expr, answer))
  let testEvalBindingsToBe = (expr, bindingsList, answer) =>
    Only.test(expr, () => expectEvalBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer))
  let testEvalPartialBindingsToBe = (expr, bindingsList, answer) =>
    Only.test(expr, () =>
      expectEvalPartialBindingsToBe(expr, bindingsList->Js.Dict.fromList, answer)
    )
}