module Template 
    ( templateGroup
    ) where

import qualified Data.Map as M
import Control.Applicative ((<$>))
import Control.Monad (replicateM)
import Data.Monoid (mempty)

import Data.Binary
import Test.Framework (testGroup)
import Test.Framework.Providers.HUnit
import Test.Framework.Providers.QuickCheck2
import Test.HUnit
import Test.QuickCheck

import Text.Hakyll.Context (Context (..))
import Text.Hakyll.Internal.Template
import Text.Hakyll.Internal.Template.Template

-- Template test group.
templateGroup = testGroup "Template"
    [ testProperty "prop_template_encode_id" prop_template_encode_id
    , testProperty "prop_substitute_id" prop_substitute_id
    , testCase "test_substitute_1" test_substitute_1
    , testCase "test_substitute_2" test_substitute_2
    ]

-- | Generate arbitrary templates from a given length.
--
instance Arbitrary TemplateElement where
    arbitrary = oneof
        -- Random chunk
        [ Chunk <$> do
            string <- arbitrary
            let sanitized = filter (/= '$') string
            return $ if null sanitized then "foo" else sanitized
        -- Random identifier
        , fmap Identifier $
            choose (5, 10) >>= flip replicateM (choose ('a', 'z'))
        -- Escape character
        , return EscapeCharacter
        ]

-- | Make @Template@ testable.
instance Arbitrary Template where
    arbitrary = Template <$> arbitrary
    shrink = map Template . shrink . unTemplate

-- Test encoding/decoding of templates.
prop_template_encode_id :: Template -> Bool
prop_template_encode_id template = decode (encode template) == template

-- Check we get the same sting with empty substitutions.
prop_substitute_id string =
    regularSubstitute (fromString string) mempty == string

-- substitute test case 1.
test_substitute_1 =
    finalSubstitute template context @?= "Banana costs $4."
  where
    template = fromString "$product costs $$$price."
    context = Context $ M.fromList [("product", "Banana"), ("price", "4")]

-- substitute test case 2.
test_substitute_2 =
    regularSubstitute template context @?= "$$root is a special key."
  where
    template = fromString "$$root is a special $thing."
    context = Context $ M.fromList [("root", "foo"), ("thing", "key")]