EachSpec.hs

{-# LANGUAGE OverloadedLists #-}

module Interpreter.Function.EachSpec (spec, main) where

import Helpers

main :: IO ()
main = hspec spec

spec :: Spec
spec = do
  let getCatalog x = case pureCatalog x of
        Left rr -> fail rr
        Right (y, _) -> pure y
  describe "should be callable as" $ do
    let checkEnsure f v c =
          getResource (RIdentifier "file" f) c >>= getAttribute "ensure" >>= \a -> a `shouldBe` v
        checks :: [(Text, PValue)] -> FinalCatalog -> IO ()
        checks lst c = mapM_ (\(f, v) -> checkEnsure f v c) lst
    it "each on an array selecting each value" $
      getCatalog "$a = [1,2,3]\n $a.each |$v| {\n file { \"/file_$v\": ensure => present } \n } "
        >>= checks
          [ ("/file_1", "present"),
            ("/file_2", "present"),
            ("/file_3", "present")
          ]
    it "each on an array selecting each value - function call style" $
      getCatalog "$a = [1,2,3]\n each ($a) |$index, $v| {\n file { \"/file_$v\": ensure => present }\n }"
        >>= checks
          [ ("/file_1", "present"),
            ("/file_2", "present"),
            ("/file_3", "present")
          ]
    it "each on an array with index" $
      getCatalog "$a = [present, absent, present]\n $a.each |$k,$v| {\n file { \"/file_$k\": ensure => $v }\n }"
        >>= checks
          [ ("/file_0", "present"),
            ("/file_1", "absent"),
            ("/file_2", "present")
          ]
    it "each on a hash selecting entries" $
      getCatalog "$a = {'a'=>'present','b'=>'absent','c'=>'present'}\n $a.each |$e| {\n $num = $e[0]\n file { \"/file_${num}\": ensure => $e[1] }\n }"
        >>= checks
          [ ("/file_a", "present"),
            ("/file_b", "absent"),
            ("/file_c", "present")
          ]
    it "each on a hash selecting key and value" $
      getCatalog "$a = {'a'=>present,'b'=>absent,'c'=>present}\n $a.each |$k, $v| {\n file { \"/file_$k\": ensure => $v }\n }"
        >>= checks
          [ ("/file_a", "present"),
            ("/file_b", "absent"),
            ("/file_c", "present")
          ]
    it "each on a hash selecting key and value (using captures-last parameter)" $ do
      pending
      getCatalog "$a = {'a'=>present,'b'=>absent,'c'=>present}\n $a.each |*$kv| {\n file { \"/file_${kv[0]}\": ensure => $kv[1] }\n }"
        >>= checks
          [ ("/file_a", "present"),
            ("/file_b", "absent"),
            ("/file_c", "present")
          ]
  describe "should produce receiver" $
    it "each checking produced value using single expression" $ do
      pending
      c <- getCatalog "$a = [1, 3, 2]\n $b = $a.each |$x| { \"unwanted\" }\n $u = $b[1]\n file { \"/file_${u}\":\n ensure => present\n }"
      getResource (RIdentifier "file" "/file_3") c >>= getAttribute "ensure" >>= \a -> a `shouldBe` "present"