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

{-# LANGUAGE OverloadedLists #-}

module Interpreter.Function.JoinKeysToValuesSpec (spec) where

import qualified Data.Foldable as F
import Data.Monoid
import Helpers
import Puppet.Interpreter
import Test.Hspec

spec :: Spec
spec = withStdlibFunction "join_keys_to_values" $ \jkvFunc -> it "Should work as expected" $ do
  let eval h s = case dummyEval (jkvFunc [PHash h, PString s]) of
        Left rr -> Left (renderToString (getError rr))
        Right (PArray vals) -> Right (F.toList vals)
        Right v -> Left ("Expected an array, not: " <> renderToString v)
  eval [] "" `shouldBe` Right []
  eval [] ":" `shouldBe` Right []
  eval [("key", "value")] "" `shouldBe` Right ["keyvalue"]
  eval [("key", "value")] ":" `shouldBe` Right ["key:value"]
  eval [("key", PUndef)] ":" `shouldBe` Right ["key:"]
  case eval [("key1", "value1"), ("key2", "value2")] ":" of
    Left rr -> fail rr
    Right lst -> lst `shouldMatchList` ["key1:value1", "key2:value2"]