module Main where

import Test.Tasty
import Test.Tasty.Golden
import System.FilePath
import System.IO (hPutStrLn, stderr)
import HsImport (hsImport, hsImportSpec)
import qualified HsImport.Args as A

main = defaultMain tests

tests :: TestTree
tests = testGroup "Tests" [moduleTests, symbolTests]

moduleTests :: TestTree
moduleTests = testGroup "Module Tests"
   [ test "ModuleTest0" $ A.defaultArgs { A.moduleName = "Foo.Bar" }
   , test "ModuleTest1" $ A.defaultArgs { A.moduleName = "Foo.Bar" }
   , test "ModuleTest2" $ A.defaultArgs { A.moduleName = "Foo.Bar.Blub" }
   , test "ModuleTest3" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest4" $ A.defaultArgs { A.moduleName = "Ugah.Argh2" }
   , test "ModuleTest5" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest6" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest7" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest8" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest9" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest10" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest11" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest12" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest13" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest14" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest15" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest16" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest17" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest18" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest19" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest20" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest21" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest22" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   , test "ModuleTest23" $ A.defaultArgs { A.moduleName = "Control.Monad", A.qualifiedName = "CM" }
   , test "ModuleTest24" $ A.defaultArgs { A.moduleName = "Control.Monad", A.qualifiedName = "CM" }
   , test "ModuleTest25" $ A.defaultArgs { A.moduleName = "Control.Monad", A.qualifiedName = "Control.Monad" }
   , test "ModuleTest26" $ A.defaultArgs { A.moduleName = "Control.Monad" }
   ]


symbolTests :: TestTree
symbolTests = testGroup "Symbol Tests"
   [ test "SymbolTest0" $ A.defaultArgs { A.moduleName = "Foo.Bar", A.symbolName = "foo" }
   , test "SymbolTest1" $ A.defaultArgs { A.moduleName = "Foo.Bar", A.symbolName = "foo" }
   , test "SymbolTest2" $ A.defaultArgs { A.moduleName = "Foo.Bar.Blub", A.symbolName = "foo" }
   , test "SymbolTest3" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "when" }
   , test "SymbolTest4" $ A.defaultArgs { A.moduleName = "Ugah.Argh2", A.symbolName = "argh" }
   , test "SymbolTest5" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "when" }
   , test "SymbolTest6" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "unless" }
   , test "SymbolTest7" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "unless" }
   , test "SymbolTest8" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "unless" }
   , test "SymbolTest9" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "unless" }
   , test "SymbolTest10" $ A.defaultArgs { A.moduleName = "Control.Applicative", A.symbolName = "<$>" }
   , test "SymbolTest11" $ A.defaultArgs { A.moduleName = "Control.Applicative", A.symbolName = "<$>" }
   , test "SymbolTest12" $ A.defaultArgs { A.moduleName = "Control.Applicative", A.symbolName = "<$>" }
   , test "SymbolTest13" $ A.defaultArgs { A.moduleName = "Control.Applicative", A.symbolName = "<*" }
   , test "SymbolTest14" $ A.defaultArgs { A.moduleName = "Control.Applicative", A.symbolName = "<*" }
   , test "SymbolTest15" $ A.defaultArgs { A.moduleName = "Control.Applicative", A.symbolName = "*>" }
   , test "SymbolTest16" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "when", A.qualifiedName = "CM" }
   , test "SymbolTest17" $ A.defaultArgs { A.moduleName = "Control.Monad", A.symbolName = "when", A.qualifiedName = "CM" }
   , test "SymbolTest18" $ A.defaultArgs { A.moduleName = "Data.Text", A.symbolName = "Text" }
   , test "SymbolTest19" $ A.defaultArgs { A.moduleName = "Data.List", A.symbolName = "foldl'" }
   , test "SymbolTest20" $ A.defaultArgs { A.moduleName = "Data.Maybe", A.symbolName = "Maybe", A.all = True }
   , test "SymbolTest21" $ A.defaultArgs { A.moduleName = "Data.Maybe", A.symbolName = "Maybe", A.all = True }
   , test "SymbolTest22" $ A.defaultArgs { A.moduleName = "Data.Maybe", A.symbolName = "Maybe", A.with = ["Just"] }
   , test "SymbolTest23" $ A.defaultArgs { A.moduleName = "Data.Maybe", A.symbolName = "Maybe", A.all = True, A.with = ["Just"] }
   , test "SymbolTest24" $ A.defaultArgs { A.moduleName = "Data.Maybe", A.symbolName = "Maybe", A.with = ["Just"] }
   , test "SymbolTest25" $ A.defaultArgs { A.moduleName = "Data.Maybe", A.symbolName = "Maybe", A.with = ["Nothing", "Just"] }
   ]


test :: String -> A.HsImportArgs -> TestTree
test testName args =
   goldenVsFileDiff testName diff goldenFile outputFile command
   where
      command = do
         spec <- hsImportSpec (args { A.inputSrcFile = inputFile, A.outputSrcFile = outputFile })
         case spec of
              Left error  -> hPutStrLn stderr ("hsimport: " ++ error)
              Right spec_ -> hsImport spec_

      diff ref new = ["diff", "-u", ref, new]

      goldenFile = "tests" </> "goldenFiles" </> testName <.> "hs"
      outputFile = "tests" </> "outputFiles" </> testName <.> "hs"
      inputFile  = "tests" </> "inputFiles"  </> testName <.> "hs"