Add support for single identifier module abbreviations.

nojaf · nojaf · commit 48e41944359a · 2022-11-29T15:27:27.000-06:00
diff --git a/tests/ParallelTypeCheckingTests/Code/FileContentMapping.fs b/tests/ParallelTypeCheckingTests/Code/FileContentMapping.fs
@@ -28,6 +28,11 @@ let visitLongIdent (lid: LongIdent) =
     | [ _ ] -> []
     | lid -> [ FileContentEntry.PrefixedIdentifier(longIdentToPath true lid) ]
 
+let visitLongIdentForModuleAbbrev (lid: LongIdent) =
+    match lid with
+    | [] -> []
+    | lid -> [ FileContentEntry.PrefixedIdentifier(longIdentToPath false lid) ]
+
 let visitSynAttribute (a: SynAttribute) : FileContentEntry list =
     [ yield! visitSynLongIdent a.TypeName; yield! visitSynExpr a.ArgExpr ]
 
@@ -55,7 +60,7 @@ let visitSynModuleDecl (decl: SynModuleDecl) : FileContentEntry list =
     | SynModuleDecl.ModuleAbbrev (longId = longId) ->
         // I believe this is enough
         // A module abbreviation doesn't appear to be exposed as part of the current module/namespace
-        visitLongIdent longId
+        visitLongIdentForModuleAbbrev longId
     | SynModuleDecl.NamespaceFragment _ -> []
     | SynModuleDecl.Exception(exnDefn = SynExceptionDefn (exnRepr = SynExceptionDefnRepr (attributes = attributes
                                                                                           caseName = caseName
@@ -79,7 +84,7 @@ let visitSynModuleSigDecl (md: SynModuleSigDecl) =
             yield FileContentEntry.NestedModule(ident.idText, lc visitSynModuleSigDecl decls)
         ]
     | SynModuleSigDecl.NestedModule _ -> failwith "A nested module cannot have multiple identifiers"
-    | SynModuleSigDecl.ModuleAbbrev _ -> failwith "no support for module abbreviations"
+    | SynModuleSigDecl.ModuleAbbrev (longId = longId) -> visitLongIdentForModuleAbbrev longId
     | SynModuleSigDecl.Val (valSig, _) -> visitSynValSig valSig
     | SynModuleSigDecl.Types (types = types) -> lc visitSynTypeDefnSig types
     | SynModuleSigDecl.Exception(exnSig = SynExceptionSig (exnRepr = SynExceptionDefnRepr (attributes = attributes
diff --git a/tests/ParallelTypeCheckingTests/Tests/FileContentMappingTests.fs b/tests/ParallelTypeCheckingTests/Tests/FileContentMappingTests.fs
@@ -106,3 +106,18 @@ module Z =
     match content with
     | [ TopLevelNamespace "X" [ NestedModule "Z" [] ] ] -> Assert.Pass()
     | content -> Assert.Fail($"Unexpected content: {content}")
+
+[<Test>]
+let ``Single ident module abbreviation`` () =
+    let content =
+        getContent
+            true
+            """
+module A
+
+module B = C
+"""
+
+    match content with
+    | [ TopLevelNamespace "" [ PrefixedIdentifier "C" ] ] -> Assert.Pass()
+    | content -> Assert.Fail($"Unexpected content: {content}")
diff --git a/tests/ParallelTypeCheckingTests/Tests/Scenarios.fs b/tests/ParallelTypeCheckingTests/Tests/Scenarios.fs
@@ -465,6 +465,8 @@ exception BakedInMemberConstraintName of string
                     """
 namespace FSharp.Compiler.CodeAnalysis
 
+open FSharp.Compiler
+
 module Tc = CheckExpressions
 """
                     (set [| 0 |])
