{-# LANGUAGE BangPatterns #-}
module Main (main) where

import Data.List
import Data.Ord
import Frelude
import Test.Tasty
import Test.Tasty.HUnit
import qualified Data.Language.Ruby as Ruby
import qualified Rubyhs.References as Ruby
import qualified Data.Aeson as Aeson
import System.FilePath
import Data.Map (Map)
import Data.HashMap.Strict (HashMap)
import Data.Set (Set)
import System.Directory (withCurrentDirectory)
import Debug.Trace
import System.IO
import qualified Data.ByteString.Lazy.Char8 as ByteString

main :: IO ()
main = withCurrentDirectory "test" $ defaultMain $ testGroup "Unit tests" tests

tests :: [TestTree]
tests = go <$> ["mod", "simple"]
  where
  go :: String -> TestTree
  go s = testCase s $ do
    let basename = "tests" </> s
    begin <- Ruby.parseFile @Ruby.Begin $ basename <.> "rb"
    json  <- decodeFile @(HashMap Ruby.Node (Set Ruby.Node)) $ basename <.> "json"
    let (Ruby.Result refs) = Ruby.references begin
    when (refs /= json)
      $ assertFailure
      $ "Expected " <> convertString (Aeson.encode refs) <> " but got " <> convertString (Aeson.encode json)

decodeFile :: FromJSON a => FilePath -> IO a
decodeFile p = Aeson.eitherDecodeFileStrict' p >>= \case
  Left err -> error err
  Right a -> pure a