Merge pull request #645 from alanz/multi-main

Multi main

Author: alanz

submodules/ghc-mod

@@ -146,6 +146,8 @@ spec = do
             )
             Nothing
 
+  -- -----------------------------------
+
   describe "multi-server setup" $
     it "doesn't have clashing commands on two servers" $ do
       let getCommands = runSession hieCommand "test/testdata" $ do
@@ -156,6 +158,8 @@ spec = do
       List uuids2 <- getCommands
       liftIO $ forM_ (zip uuids1 uuids2) (uncurry shouldNotBe)
 
+  -- -----------------------------------
+
   describe "code actions" $
     it "provide hlint suggestions" $ runSession hieCommand "test/testdata" $ do
       doc <- openDoc "ApplyRefact2.hs" "haskell"
@@ -180,4 +184,24 @@ spec = do
       liftIO $ do
         length cmds `shouldBe` 1
         evaluateCmd ^. title `shouldBe` "Apply hint:Evaluate"
-      
+
+  -- -----------------------------------
+
+  describe "multiple main modules" $
+    it "Can load one file at a time, when more than one Main module exists"
+                                  -- $ runSession hieCommand "test/testdata" $ do
+                                  $ runSession hieCommandVomit "test/testdata" $ do
+      _doc <- openDoc "ApplyRefact2.hs" "haskell"
+      _diagsRspHlint <- skipManyTill anyNotification notification :: Session PublishDiagnosticsNotification
+      diagsRspGhc   <- skipManyTill anyNotification notification :: Session PublishDiagnosticsNotification
+      let (List diags) = diagsRspGhc ^. params . diagnostics
+
+      liftIO $ length diags `shouldBe` 2
+
+      _doc2 <- openDoc "HaReRename.hs" "haskell"
+      _diagsRspHlint2 <- skipManyTill anyNotification notification :: Session PublishDiagnosticsNotification
+      -- errMsg <- skipManyTill anyNotification notification :: Session ShowMessageNotification
+      diagsRsp2 <- skipManyTill anyNotification notification :: Session PublishDiagnosticsNotification
+      let (List diags2) = diagsRsp2 ^. params . diagnostics
+
+      liftIO $ show diags2 `shouldBe` "[]"

test/Functional.hs

@@ -10,6 +10,7 @@ module TestUtils
   , makeRequest
   , runIGM
   , hieCommand
+  , hieCommandVomit
   ) where
 
 import           Control.Exception
@@ -120,6 +121,9 @@ stackYaml =
 hieCommand :: String
 hieCommand = "stack exec --stack-yaml=" ++ stackYaml ++ " hie -- --lsp -d -l test-logs/functional-hie.log"
 
+hieCommandVomit :: String
+hieCommandVomit = hieCommand ++ " --vomit"
+
 -- |Choose a resolver based on the current compiler, otherwise HaRe/ghc-mod will
 -- not be able to load the files
 readResolver :: IO String

test/TestUtils.hs

@@ -1,2 +1,2 @@
 main = undefined
-foo x = id x
+foo x = id x