{-# LANGUAGE TemplateHaskell #-}
{-# LANGUAGE OverloadedStrings #-}
module PrettyTests (tests) where

import           Test.Tasty
import           Test.Tasty.HUnit
import           Test.Tasty.TH

import           Nix.Expr
import           Nix.Pretty

case_indented_antiquotation :: Assertion
case_indented_antiquotation = do
  assertPretty (mkIndentedStr 0 "echo $foo")   "''echo $foo''"
  assertPretty (mkIndentedStr 0 "echo ${foo}") "''echo ''${foo}''"

case_string_antiquotation :: Assertion
case_string_antiquotation = do
  assertPretty (mkStr "echo $foo")   "\"echo \\$foo\""
  assertPretty (mkStr "echo ${foo}") "\"echo \\${foo}\""

case_function_params :: Assertion
case_function_params =
  assertPretty (mkFunction (mkParamset [] True) (mkInt 3)) "{ ... }:\n  3"

case_paths :: Assertion
case_paths = do
  assertPretty (mkPath False "~/test.nix") "~/test.nix"
  assertPretty (mkPath False "/test.nix")  "/test.nix"
  assertPretty (mkPath False "./test.nix") "./test.nix"

tests :: TestTree
tests = $testGroupGenerator

--------------------------------------------------------------------------------
assertPretty :: NExpr -> String -> Assertion
assertPretty e s =
  assertEqual ("When pretty-printing " ++ show e) s . show $ prettyNix e