hnix/tests/TestCommon.hs

65 lines
2 KiB
Haskell

module TestCommon where
import Data.Text (Text, unpack)
import Nix
import System.Environment
import System.IO
import System.Posix.Files
import System.Posix.Temp
import System.Process
import Test.Tasty.HUnit
hnixEvalFile :: FilePath -> [String] -> IO (NValueNF (Lazy IO))
hnixEvalFile file incls = do
parseResult <- parseNixFileLoc file
case parseResult of
Failure err ->
error $ "Parsing failed for file `" ++ file ++ "`.\n" ++ show err
Success expr -> do
setEnv "TEST_VAR" "foo"
runLazyM $ normalForm =<< evalLoc (Just file) incls expr
hnixEvalFileOpts :: Options -> FilePath -> IO (NValueNF (Lazy IO))
hnixEvalFileOpts opts file = do
parseResult <- parseNixFileLoc file
case parseResult of
Failure err ->
error $ "Parsing failed for file `" ++ file ++ "`.\n" ++ show err
Success expr -> do
setEnv "TEST_VAR" "foo"
runLazyM $ evaluateExpression opts (Just file) evalLoc normalForm expr
hnixEvalText :: Text -> [String] -> IO (NValueNF (Lazy IO))
hnixEvalText src incls = case parseNixText src of
Failure err ->
error $ "Parsing failed for expressien `"
++ unpack src ++ "`.\n" ++ show err
Success expr ->
runLazyM $ normalForm =<< eval Nothing incls expr
nixEvalString :: String -> IO String
nixEvalString expr = do
(fp,h) <- mkstemp "nix-test-eval"
hPutStr h expr
hClose h
res <- nixEvalFile fp
removeLink fp
return res
nixEvalFile :: FilePath -> IO String
nixEvalFile fp = readProcess "nix-instantiate" ["--eval", fp] ""
assertEvalFileMatchesNix :: FilePath -> Assertion
assertEvalFileMatchesNix fp = do
hnixVal <- (++"\n") . printNix <$> hnixEvalFile fp []
nixVal <- nixEvalFile fp
assertEqual fp nixVal hnixVal
assertEvalMatchesNix :: Text -> Assertion
assertEvalMatchesNix expr = do
hnixVal <- (++"\n") . printNix <$> hnixEvalText expr []
nixVal <- nixEvalString expr'
assertEqual expr' nixVal hnixVal
where
expr' = unpack expr