{-# LANGUAGE OverloadedStrings #-}

module Jikka.Core.Convert.ANormalSpec
  ( spec,
  )
where

import Jikka.Common.Alpha
import Jikka.Common.Error
import Jikka.Core.Convert.ANormal (run)
import Jikka.Core.Language.BuiltinPatterns
import Jikka.Core.Language.Expr
import Test.Hspec

run' :: Program -> Either Error Program
run' = flip evalAlphaT 0 . run

spec :: Spec
spec = describe "run" $ do
  it "works" $ do
    let prog =
          ResultExpr $
            Plus'
              (Let "x" IntTy Lit1 (Var "x"))
              (App (Lam "x" IntTy (Var "x")) Lit1)
    let expected =
          ResultExpr $
            Let "x$0" IntTy Lit1 $
              Let "$3" (Fun1STy IntTy) (Lam "x$1" IntTy (Var "x$1")) $
                Let "$2" (Fun1STy IntTy) (Var "$3") $
                  Let "$4" IntTy (App (Var "$2") Lit1) $
                    Plus' (Var "x$0") (Var "$4")
    run' prog `shouldBe` Right expected