haskell-language-server-0.8.0.0: test/functional/ModuleName.hs

{-# LANGUAGE NamedFieldPuns      #-}
{-# LANGUAGE OverloadedStrings   #-}
{-# LANGUAGE ScopedTypeVariables #-}

module ModuleName
  ( tests
  )
where

import           Control.Applicative.Combinators (skipManyTill)
import           Control.Monad.IO.Class          (MonadIO (liftIO))
import qualified Data.Text.IO                    as T
import           Language.Haskell.LSP.Test       (anyMessage, documentContents,
                                                  executeCommand, fullCaps,
                                                  getCodeLenses, message,
                                                  openDoc, runSession)
import           Language.Haskell.LSP.Types      (ApplyWorkspaceEditRequest,
                                                  CodeLens (..))
import           System.FilePath                 ((<.>), (</>))
import           Test.Hls.Util                   (hlsCommand)
import           Test.Tasty                      (TestTree, testGroup)
import           Test.Tasty.HUnit                (testCase, (@?=))

tests :: TestTree
tests = testGroup
  "moduleName"
  [ testCase "Add module header to empty module" $ goldenTest "TEmptyModule.hs"
  , testCase "Fix wrong module name" $ goldenTest "TWrongModuleName.hs"
  , testCase "Must infer module name as Main, if the file name starts with a lowercase" $ goldenTest "mainlike.hs"
  ]

goldenTest :: FilePath -> IO ()
goldenTest input = runSession hlsCommand fullCaps testdataPath $ do
  doc                              <- openDoc input "haskell"
  -- getCodeLenses doc >>= liftIO . print . length
  [CodeLens { _command = Just c }] <- getCodeLenses doc
  executeCommand c
  _resp :: ApplyWorkspaceEditRequest <- skipManyTill anyMessage message
  edited <- documentContents doc
  -- liftIO $ T.writeFile (testdataPath </> input <.> "expected") edited
  expected <- liftIO $ T.readFile $ testdataPath </> input <.> "expected"
  liftIO $ edited @?= expected

testdataPath :: FilePath
testdataPath = "test/testdata/moduleName"