test: Add regression tests for #2403

haskell · Jan 10, 2022 · ea73091 · ea73091
1 parent aa5379d
commit ea73091
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/ghcide/test/exe/Main.hs b/ghcide/test/exe/Main.hs
@@ -4527,6 +4527,15 @@ localCompletionTests = [
         ,("abcdefgh", CiFunction, "abcdefgh", True, False, Nothing)
         ,("abcdefghi", CiFunction, "abcdefghi", True, False, Nothing)
         ],
+    completionTest
+        "type family"
+        ["{-# LANGUAGE DataKinds, TypeFamilies #-}"
+        ,"type family Bar a"
+        ,"a :: Ba"
+        ]
+        (Position 2 7)
+        [("Bar", CiStruct, "Bar", True, False, Nothing)
+        ],
     completionTest
         "class method"
         [

diff --git a/test/functional/Completion.hs b/test/functional/Completion.hs
@@ -61,6 +61,17 @@ tests = testGroup "completions" [
              item ^. detail @?= Just "Data.Maybe"
              item ^. kind @?= Just CiModule
 
+     , testCase "completes type family" $ runSession (hlsCommand <> " --test") fullCaps "test/testdata/completion" $ do
+         doc <- openDoc "CompletionTypeFamily.hs" "haskell"
+
+         compls <- getCompletions doc (Position 9 16)
+         let item = head $ filter ((== "Bar") . (^. label)) compls
+         liftIO $ do
+             item ^. label @?= "Bar"
+             item ^. kind @?= Just CiStruct
+             item ^. detail @?= Just ":: Bar"
+             item ^. insertTextFormat @?= Just Snippet
+             item ^. insertText @?= Just "Bar"
      , testCase "completes qualified imports" $ runSession (hlsCommand <> " --test") fullCaps "test/testdata/completion" $ do
          doc <- openDoc "Completion.hs" "haskell"
 

diff --git a/test/testdata/completion/CompletionTypeFamily.hs b/test/testdata/completion/CompletionTypeFamily.hs
@@ -0,0 +1,11 @@
+-- Based on #2403
+{-# LANGUAGE DataKinds #-}
+{-# LANGUAGE TypeFamilies #-}
+
+module MyLib where
+
+type family Bar a
+
+class Foo a where
+  foo :: a -> Ba
+  foo = undefined