From ee64204ec7fcfb6de6cc0662fa82eebc9369a0dd Mon Sep 17 00:00:00 2001
From: Koen Bekkenutte <2912652+kbekkenutte@users.noreply.github.com>
Date: Thu, 3 Jun 2021 02:38:12 +0800
Subject: [PATCH] Fixed broken tests

---
 .../ExpressionSyntaxRewriter.cs               | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/EntityFrameworkCore.Projectables.Generator/ExpressionSyntaxRewriter.cs b/src/EntityFrameworkCore.Projectables.Generator/ExpressionSyntaxRewriter.cs
index 158b47c..dc21510 100644
--- a/src/EntityFrameworkCore.Projectables.Generator/ExpressionSyntaxRewriter.cs
+++ b/src/EntityFrameworkCore.Projectables.Generator/ExpressionSyntaxRewriter.cs
@@ -58,7 +58,7 @@ namespace EntityFrameworkCore.Projectables.Generator
                         node
                     );
                 }
-                else if (symbolInfo.Symbol.Kind is SymbolKind.NamedType)
+                else if (symbolInfo.Symbol.Kind is SymbolKind.NamedType && node.Parent.Kind() is not SyntaxKind.QualifiedName)
                 {
                     var typeInfo = _semanticModel.GetTypeInfo(node);
 
@@ -76,6 +76,23 @@ namespace EntityFrameworkCore.Projectables.Generator
 
         public override SyntaxNode? VisitQualifiedName(QualifiedNameSyntax node)
         {
+            var symbolInfo = _semanticModel.GetSymbolInfo(node);
+
+            if (symbolInfo.Symbol is not null)
+            {
+                if (symbolInfo.Symbol.Kind is SymbolKind.NamedType)
+                {
+                    var typeInfo = _semanticModel.GetTypeInfo(node);
+
+                    if (typeInfo.Type is not null)
+                    {
+                        return SyntaxFactory.ParseTypeName(
+                            typeInfo.Type.ToDisplayString(SymbolDisplayFormat.FullyQualifiedFormat)
+                        );
+                    }
+                }
+            }
+
             return base.VisitQualifiedName(node);
         }
     }