UpdateGenerationTests.hs

module Hydra.UpdateGenerationTests where

import Hydra.Kernel
import Hydra.Generation (createTestGroupLookup, generateGenerationTestSuite)
import Hydra.Ext.Haskell.TestingIo (haskellTestGenerator)
import qualified Hydra.Sources.Test.TestSuite as TestSuite
import qualified Hydra.Test.TestSuite as GenTests
import System.Exit (exitFailure)


updateGenerationTestsHaskell :: IO ()
updateGenerationTestsHaskell = do
  putStrLn "=== Generate Hydra generation tests ==="
  putStrLn ""

  -- Get the namespaces from TestSuite's term dependencies
  let testNamespaces = moduleTermDependencies TestSuite.module_

  -- Build the lookup function from namespaces and test group hierarchy
  let lookupFn = createTestGroupLookup testNamespaces GenTests.allTests

  -- Get the list of test modules explicitly
  let testModules = TestSuite.testSuiteModules

  -- Generate generation tests to src/gen-test/haskell
  let outputDir = "src/gen-test/haskell"

  putStrLn $ "Generating tests into: " ++ outputDir
  putStrLn ""

  success <- generateGenerationTestSuite haskellTestGenerator outputDir testModules lookupFn

  if success
    then do
      putStrLn ""
      putStrLn "=== Done! ==="
      putStrLn ""
      putStrLn "To view the generated tests:"
      putStrLn $ "  ls -R " ++ outputDir
      putStrLn ""
      putStrLn "To run the generated tests:"
      putStrLn "  stack test"
    else do
      putStrLn ""
      putStrLn "=== FAILED ==="
      exitFailure