Test Fourmolu with CLI option

Author: georgefst

plugins/hls-fourmolu-plugin/hls-fourmolu-plugin.cabal

@@ -41,6 +41,8 @@ test-suite tests
   hs-source-dirs:   test
   main-is:          Main.hs
   ghc-options:      -Wall -threaded -rtsopts -with-rtsopts=-N
+  build-tool-depends:
+    fourmolu:fourmolu
   build-depends:
     , base
     , aeson

plugins/hls-fourmolu-plugin/test/Main.hs

@@ -3,6 +3,9 @@ module Main
   ( main
   ) where
 
+import           Data.Aeson
+import           Data.Functor
+import           Ide.Plugin.Config
 import qualified Ide.Plugin.Fourmolu as Fourmolu
 import           Language.LSP.Test
 import           Language.LSP.Types
@@ -16,15 +19,21 @@ fourmoluPlugin :: PluginDescriptor IdeState
 fourmoluPlugin = Fourmolu.descriptor "fourmolu"
 
 tests :: TestTree
-tests = testGroup "fourmolu"
-  [ goldenWithFourmolu "formats correctly" "Fourmolu" "formatted" $ \doc -> do
-      formatDoc doc (FormattingOptions 4 True Nothing Nothing Nothing)
-  , goldenWithFourmolu "formats imports correctly" "Fourmolu" "formatted" $ \doc -> do
-      formatDoc doc (FormattingOptions 4 True Nothing Nothing Nothing)
-  ]
+tests =
+  testGroup "fourmolu" $
+    [False, True] <&> \cli ->
+      testGroup
+        (if cli then "cli" else "lib")
+        [ goldenWithFourmolu cli "formats correctly" "Fourmolu" "formatted" $ \doc -> do
+            formatDoc doc (FormattingOptions 4 True Nothing Nothing Nothing)
+        , goldenWithFourmolu cli "formats imports correctly" "Fourmolu" "formatted" $ \doc -> do
+            formatDoc doc (FormattingOptions 4 True Nothing Nothing Nothing)
+        ]
 
-goldenWithFourmolu :: TestName -> FilePath -> FilePath -> (TextDocumentIdentifier -> Session ()) -> TestTree
-goldenWithFourmolu title path desc = goldenWithHaskellDocFormatter fourmoluPlugin "fourmolu" def title testDataDir path desc "hs"
+goldenWithFourmolu :: Bool -> TestName -> FilePath -> FilePath -> (TextDocumentIdentifier -> Session ()) -> TestTree
+goldenWithFourmolu cli title path desc = goldenWithHaskellDocFormatter fourmoluPlugin "fourmolu" conf title testDataDir path desc "hs"
+ where
+  conf = def{plcConfig = (\(Object obj) -> obj) $ object ["cli" .= cli]}
 
 testDataDir :: FilePath
 testDataDir = "test" </> "testdata"