{-# LANGUAGE OverloadedStrings #-}

module Hydra.Dsl.TypesSpec where

import Hydra.Kernel
import Hydra.Dsl.Types

import qualified Test.Hspec as H


check :: Type Kv -> Type Kv -> H.Expectation
check = H.shouldBe

checkFunctionSyntax :: H.SpecWith ()
checkFunctionSyntax = do
  H.describe "Check function syntax" $ do

    H.it "Function arrows are supported" $ do
      check
        ("a" --> "b")
        (function (var "a") (var "b"))
      check
        (string --> int32)
        (function string int32)

    H.it "Function arrows are right-associative" $ do
      check
        ("a" --> "b" --> "c")
        ("a" --> ("b" --> "c"))

    H.it "Functions bind less tightly than application" $ do
      check
        ("a" @@ "b" --> "c" @@ "d")
        (("a" @@ "b") --> ("c" @@ "d"))

checkHelperFunctions :: H.SpecWith ()
checkHelperFunctions = do
  H.describe "Check helper functions" $ do

    H.it "Check n-ary functions" $ do
      check
        (functionN ["a", "b"])
        (function "a" "b")
      check
        (functionN [int32, string, boolean])
        (function int32 $ function string boolean)

spec :: H.Spec
spec = do
  checkFunctionSyntax
  checkHelperFunctions