{-# LANGUAGE DataKinds #-}
{-# LANGUAGE DuplicateRecordFields #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE TypeOperators #-}
{-# LANGUAGE ViewPatterns #-}
module Utils where
import Control.DeepSeq (deepseq)
import qualified Control.Exception as E
import Control.Lens hiding (failing, (<.>), (.=))
import Control.Monad (unless)
import Control.Monad.IO.Class
import Data.Aeson
import Data.Foldable
import Data.Function (on)
import qualified Data.Map as M
import Data.Maybe
import Data.Text (Text)
import qualified Data.Text as T
import qualified Data.Text.IO as T
import qualified Ide.Plugin.Config as Plugin
import Ide.Plugin.Tactic as Tactic
import Language.LSP.Types
import Language.LSP.Types.Lens hiding (actions, applyEdit, capabilities, executeCommand, id, line, message, name, rename, title)
import System.Directory (doesFileExist)
import System.FilePath
import Test.Hls
import Test.Hspec
import Test.Hspec.Formatters (FailureReason(ExpectedButGot))
import Test.Tasty.HUnit (Assertion, HUnitFailure(..))
import Wingman.FeatureSet (FeatureSet, allFeatures, prettyFeatureSet)
import Wingman.LanguageServer (mkShowMessageParams)
import Wingman.Types
plugin :: PluginDescriptor IdeState
plugin = Tactic.descriptor "tactics"
------------------------------------------------------------------------------
-- | Get a range at the given line and column corresponding to having nothing
-- selected.
--
-- NB: These coordinates are in "file space", ie, 1-indexed.
pointRange :: Int -> Int -> Range
pointRange
(subtract 1 -> line)
(subtract 1 -> col) =
Range (Position line col) (Position line $ col + 1)
------------------------------------------------------------------------------
-- | Get the title of a code action.
codeActionTitle :: (Command |? CodeAction) -> Maybe Text
codeActionTitle InL{} = Nothing
codeActionTitle (InR(CodeAction title _ _ _ _ _ _ _)) = Just title
------------------------------------------------------------------------------
-- | Make a tactic unit test.
mkTest
:: Foldable t
=> String -- ^ The test name
-> FilePath -- ^ The file to load
-> Int -- ^ Cursor line
-> Int -- ^ Cursor columnn
-> t ( Bool -> Bool -- Use 'not' for actions that shouldnt be present
, TacticCommand -- An expected command ...
, Text -- ... for this variable
) -- ^ A collection of (un)expected code actions.
-> SpecWith (Arg Bool)
mkTest name fp line col ts = it name $ do
runSessionWithServer plugin tacticPath $ do
setFeatureSet allFeatures
doc <- openDoc fp "haskell"
_ <- waitForDiagnostics
actions <- getCodeActions doc $ pointRange line col
let titles = mapMaybe codeActionTitle actions
for_ ts $ \(f, tc, var) -> do
let title = tacticTitle tc var
liftIO $
(title `elem` titles) `shouldSatisfy` f
setFeatureSet :: FeatureSet -> Session ()
setFeatureSet features = do
let unObject (Object obj) = obj
unObject _ = undefined
def_config = def :: Plugin.Config
config =
def_config
{ Plugin.plugins = M.fromList [("tactics",
def { Plugin.plcConfig = unObject $ object ["features" .= prettyFeatureSet features] }
)] <> Plugin.plugins def_config }
sendNotification SWorkspaceDidChangeConfiguration $
DidChangeConfigurationParams $
toJSON config
mkGoldenTest
:: (Text -> Text -> Assertion)
-> FeatureSet
-> TacticCommand
-> Text
-> Int
-> Int
-> FilePath
-> SpecWith ()
mkGoldenTest eq features tc occ line col input =
it (input <> " (golden)") $ do
runSessionWithServer plugin tacticPath $ do
setFeatureSet features
doc <- openDoc input "haskell"
_ <- waitForDiagnostics
actions <- getCodeActions doc $ pointRange line col
Just (InR CodeAction {_command = Just c})
<- pure $ find ((== Just (tacticTitle tc occ)) . codeActionTitle) actions
executeCommand c
_resp <- skipManyTill anyMessage (message SWorkspaceApplyEdit)
edited <- documentContents doc
let expected_name = input <.> "expected"
-- Write golden tests if they don't already exist
liftIO $ (doesFileExist expected_name >>=) $ flip unless $ do
T.writeFile expected_name edited
expected <- liftIO $ T.readFile expected_name
liftIO $ edited `eq` expected
mkShowMessageTest
:: FeatureSet
-> TacticCommand
-> Text
-> Int
-> Int
-> FilePath
-> UserFacingMessage
-> SpecWith ()
mkShowMessageTest features tc occ line col input ufm =
it (input <> " (golden)") $ do
runSessionWithServer plugin tacticPath $ do
setFeatureSet features
doc <- openDoc input "haskell"
_ <- waitForDiagnostics
actions <- getCodeActions doc $ pointRange line col
Just (InR CodeAction {_command = Just c})
<- pure $ find ((== Just (tacticTitle tc occ)) . codeActionTitle) actions
executeCommand c
NotificationMessage _ _ err <- skipManyTill anyMessage (message SWindowShowMessage)
liftIO $ err `shouldBe` mkShowMessageParams ufm
goldenTest :: TacticCommand -> Text -> Int -> Int -> FilePath -> SpecWith ()
goldenTest = mkGoldenTest shouldBe allFeatures
goldenTestNoWhitespace :: TacticCommand -> Text -> Int -> Int -> FilePath -> SpecWith ()
goldenTestNoWhitespace = mkGoldenTest shouldBeIgnoringSpaces allFeatures
shouldBeIgnoringSpaces :: Text -> Text -> Assertion
shouldBeIgnoringSpaces = assertFun f ""
where
f = (==) `on` T.unwords . T.words
assertFun
:: Show a
=> (a -> a -> Bool)
-> String -- ^ The message prefix
-> a -- ^ The expected value
-> a -- ^ The actual value
-> Assertion
assertFun eq preface expected actual =
unless (eq actual expected) $ do
(prefaceMsg
`deepseq` expectedMsg
`deepseq` actualMsg
`deepseq`
E.throwIO
(HUnitFailure Nothing $ show $ ExpectedButGot prefaceMsg expectedMsg actualMsg))
where
prefaceMsg
| null preface = Nothing
| otherwise = Just preface
expectedMsg = show expected
actualMsg = show actual
------------------------------------------------------------------------------
-- | Don't run a test.
failing :: Applicative m => String -> b -> m ()
failing _ _ = pure ()
tacticPath :: FilePath
tacticPath = "test/golden"
executeCommandWithResp :: Command -> Session (ResponseMessage 'WorkspaceExecuteCommand)
executeCommandWithResp cmd = do
let args = decode $ encode $ fromJust $ cmd ^. arguments
execParams = ExecuteCommandParams Nothing (cmd ^. command) args
request SWorkspaceExecuteCommand execParams