Strip parentheses from AsTypeName of array returning functions and pr…

…operties
rubberduck-vba · Aug 24, 2019 · 617b27f · 617b27f
1 parent 3a915a4
commit 617b27f
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 2 deletions.
diff --git a/Rubberduck.Parsing/VBA/DeclarationResolving/DeclarationSymbolsListener.cs b/Rubberduck.Parsing/VBA/DeclarationResolving/DeclarationSymbolsListener.cs
@@ -93,6 +93,7 @@ private IEnumerable<IAnnotation> FindGeneralAnnotations(int firstLine)
 
                 var isByRef = argContext.BYREF() != null || argContext.BYVAL() == null;
                 var isParamArray = argContext.PARAMARRAY() != null;
+
                 result = new ParameterDeclaration(
                     new QualifiedMemberName(_qualifiedModuleName, identifierName),
                     _parentDeclaration,
@@ -385,9 +386,12 @@ public override void EnterFunctionStmt(VBAParser.FunctionStmtContext context)
                     : asTypeClause.type().GetText()
                 : SymbolList.TypeHintToTypeName[typeHint];
             var isArray = asTypeName.EndsWith("()");
+            var actualAsTypeName = isArray && asTypeName.EndsWith("()")
+                ? asTypeName.Substring(0, asTypeName.Length - 2)
+                : asTypeName;
             var declaration = CreateDeclaration(
                 name,
-                asTypeName,
+                actualAsTypeName,
                 accessibility,
                 DeclarationType.Function,
                 context,
@@ -417,9 +421,12 @@ public override void EnterPropertyGetStmt(VBAParser.PropertyGetStmtContext conte
                     : asTypeClause.type().GetText()
                 : SymbolList.TypeHintToTypeName[typeHint];
             var isArray = asTypeClause != null && asTypeClause.type().LPAREN() != null;
+            var actualAsTypeName = isArray && asTypeName.EndsWith("()")
+                ? asTypeName.Substring(0, asTypeName.Length - 2)
+                : asTypeName;
             var declaration = CreateDeclaration(
                 name,
-                asTypeName,
+                actualAsTypeName,
                 accessibility,
                 DeclarationType.PropertyGet,
                 context,

diff --git a/RubberduckTests/Grammar/ResolverTests.cs b/RubberduckTests/Grammar/ResolverTests.cs
@@ -2730,6 +2730,52 @@ End Sub
             }
         }
 
+        [Category("Grammar")]
+        [Category("Resolver")]
+        [Test]
+        public void FunctionTypeNameIsElementTypeName()
+        {
+            var code = @"
+Public Function bar() As Long()
+End Function
+";
+            using (var state = Resolve(code))
+            {
+
+                var declaration = state.DeclarationFinder
+                    .UserDeclarations(DeclarationType.Function)
+                    .Single();
+
+                var expectedTypeName = "Long";
+                var actualAsTypeName = declaration.AsTypeName;
+
+                Assert.AreEqual(expectedTypeName, actualAsTypeName);
+            }
+        }
+
+        [Category("Grammar")]
+        [Category("Resolver")]
+        [Test]
+        public void PropertyGetTypeNameIsElementTypeName()
+        {
+            var code = @"
+Public Property Get bar() As Long()
+End Property
+";
+            using (var state = Resolve(code))
+            {
+
+                var declaration = state.DeclarationFinder
+                    .UserDeclarations(DeclarationType.PropertyGet)
+                    .Single();
+
+                var expectedTypeName = "Long";
+                var actualAsTypeName = declaration.AsTypeName;
+
+                Assert.AreEqual(expectedTypeName, actualAsTypeName);
+            }
+        }
+
         [Category("Grammar")]
         [Category("Resolver")]
         [Test]