{-# LANGUAGE OverloadedStrings #-}

module IntegrationTests where

import           System.IO                 (hClose)
import           System.IO.Temp            (withSystemTempFile)
import           Test.HUnit

import qualified HBF.Compiler              as C
import           HBF.Types
import           Helper

squaresPath :: FilePath
squaresPath = "tests/squares.bf"

squaresResult :: String
squaresResult = concatMap ((++ "\n") . show) sq
  where
    sq :: [Int]
    sq = takeWhile (<= 10000) $ map (\n -> n * n) [0 ..]

compile :: FilePath -> IO (Program Optimized)
compile inpath =
  withSystemTempFile "hbftests.bfc" $ \outpath handle -> do
    hClose handle
    cOpts <- C.unsafeParse ["--output", outpath, inpath]
    Right _ <- C.compile cOpts
    C.loadFile outpath

unit_compileAndEvalSquares :: Assertion
unit_compileAndEvalSquares = do
  program <- compile squaresPath
  (_,mock) <- execProgram program (mkMockIO [])
  mockOutputS mock @?= squaresResult