language-puppet-1.5.1: tests/Interpreter/Function/AssertPrivateSpec.hs

{-# LANGUAGE OverloadedLists #-}

module Interpreter.Function.AssertPrivateSpec where

import Helpers

main :: IO ()
main = hspec spec

evalWithScope ::
  ([PValue] -> InterpreterMonad PValue) ->
  -- | caller scope
  Text ->
  -- | module scope
  Text ->
  -- | function args
  [Expression] ->
  Either String PValue
evalWithScope apFunc callerScope moduleScope =
  (_Left %~ show) . view _1 . ctxEval . (mapM resolveExpression >=> apFunc)
  where
    ctxEval = pureEval' mempty state0 Nothing
    state0 = dummyInitialState & curScope .~ [ContClass moduleScope, ContClass callerScope]

spec :: Spec
spec = withStdlibFunction "assert_private" $ \apFunc -> do
  let errorWith a b = case a of
        Right x -> fail ("Should have failed, got this instead: " ++ show x)
        Left rr -> rr `shouldContain` b
  it "should work when called from inside module" (evalWithScope apFunc "bar" "bar" [] `shouldBe` Right PUndef)
  it "should fail with the default message" (evalWithScope apFunc "bar" "baz" [] `errorWith` "is private")
  it "should fail with an explicit failure message" (evalWithScope apFunc "bar" "baz" ["lalala"] `errorWith` "lalala")