hnix-store-aterm-0.1.0.0: tests/Derivation.hs

module Derivation where

import Test.Tasty (TestTree, testGroup)
import Test.Tasty.Golden (goldenVsFile)

import System.Nix.Derivation.ATerm (parseTraditionalDerivation, buildTraditionalDerivation)

import Data.Default.Class (Default(def))
import Data.Attoparsec.Text.Lazy qualified
import Data.Text.Lazy.IO qualified
import Data.Text.Lazy.Builder qualified

processDerivation :: FilePath -> FilePath -> IO ()
processDerivation source dest = do
  contents <- Data.Text.Lazy.IO.readFile source
  either
    fail
    (Data.Text.Lazy.IO.writeFile dest
      . Data.Text.Lazy.Builder.toLazyText
      . buildTraditionalDerivation def
    )
    (Data.Attoparsec.Text.Lazy.parseOnly
      (parseTraditionalDerivation def)
      contents
    )

test_derivation :: TestTree
test_derivation =
  testGroup "golden" $ fmap mk [0 .. 1]
 where
  mk :: Int -> TestTree
  mk n =
    goldenVsFile
      ("derivation roundtrip of " <> drv)
      drv
      act
      (processDerivation drv act)
   where
    drv = fp <> show n <> ".drv"
    act = fp <> show n <> ".actual"
    fp  = "tests/example"