Add class plugin test for _

haskell · Jan 7, 2021 · 645f141 · 645f141
1 parent 379f590
commit 645f141
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/test/functional/Class.hs b/test/functional/Class.hs
@@ -48,6 +48,9 @@ tests = testGroup
   , glodenTest "Creates a placeholder for multiple methods 2" "T3" "2"
     $ \(_:mmAction:_) -> do
       executeCodeAction mmAction
+  , glodenTest "Creates a placeholder for a method starting with '_'" "T4" ""
+    $ \(_:mmAction:_) -> do
+      executeCodeAction mmAction
   ]
 
 _CACodeAction :: Prism' CAResult CodeAction
@@ -60,14 +63,18 @@ classPath = "test" </> "testdata" </> "class"
 
 glodenTest :: String -> FilePath -> FilePath -> ([CodeAction] -> Session ()) -> TestTree
 glodenTest name fp deco execute
-  = goldenVsStringDiff name goldenGitDiff (classPath </> fp <.> deco <.> "expected" <.> "hs")
+  = goldenVsStringDiff name goldenGitDiff (classPath </> fpWithDeco <.> "expected" <.> "hs")
     $ runSession hlsCommand fullCaps classPath
     $ do
       doc <- openDoc (fp <.> "hs") "haskell"
       _ <- waitForDiagnosticsFromSource doc "typecheck"
       actions <- concatMap (^.. _CACodeAction) <$> getAllCodeActions doc
       execute actions
       BS.fromStrict . T.encodeUtf8 <$> getDocumentEdit doc
+  where
+    fpWithDeco
+      | deco == "" = fp
+      | otherwise  = fp <.> deco
 
 goldenGitDiff :: FilePath -> FilePath -> [String]
 goldenGitDiff fRef fNew = ["git", "diff", "--no-index", "--text", "--exit-code", fRef, fNew]
diff --git a/test/testdata/class/T4.hs b/test/testdata/class/T4.hs
@@ -0,0 +1,7 @@
+module T4 where
+
+class Test a where
+  _f :: a
+  {-# MINIMAL f #-}
+
+instance Test Int where