add additional check for method/fetches in TestQueryable/ExpressionTr…

…eeModifier (#21) * add additional check for method/fetches in TestQueryable/ExpressionTreeModifier * up version
Luxoft · Sep 20, 2024 · 00eb725 · 00eb725
1 parent 8c86c61
commit 00eb725
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 11 deletions.
diff --git a/src/Bss.Platform.NHibernate.UnitTesting/Queryable/ExpressionTreeModifier.cs b/src/Bss.Platform.NHibernate.UnitTesting/Queryable/ExpressionTreeModifier.cs
@@ -27,15 +27,26 @@ protected override Expression VisitMethodCall(MethodCallExpression node)
 
         var fetchInput = AvoidFetch(node);
         return fetchInput.NodeType switch
-               {
-                   ExpressionType.Constant => this.VisitConstant((ConstantExpression)fetchInput),
-                   _ => fetchInput
-               };
+        {
+            ExpressionType.Constant => this.VisitConstant((ConstantExpression)fetchInput),
+            _ => fetchInput
+        };
     }
 
-    private static Expression AvoidFetch(MethodCallExpression node) =>
-        IsFetchMethod(node.Method) ? AvoidFetch((node.Arguments[0] as MethodCallExpression)!) : node;
+    private static Expression AvoidFetch(MethodCallExpression node)
+    {
+        if (!IsFetchMethod(node.Method))
+        {
+            return node;
+        }
+
+        return node.Arguments[0] switch
+        {
+            ConstantExpression constantExpression => constantExpression,
+            MethodCallExpression methodCallExpression => AvoidFetch(methodCallExpression),
+            _ => throw new ArgumentOutOfRangeException($"Not handled case - first argument is '{node.Arguments[0].GetType().Name}'")
+        };
+    }
 
-    private static bool IsFetchMethod(MethodInfo info) =>
-        info.IsGenericMethod && VisitedMethods.Contains(info.GetGenericMethodDefinition());
+    private static bool IsFetchMethod(MethodInfo info) => info.IsGenericMethod && VisitedMethods.Contains(info.GetGenericMethodDefinition());
 }
diff --git a/src/__SolutionItems/CommonAssemblyInfo.cs b/src/__SolutionItems/CommonAssemblyInfo.cs
@@ -4,9 +4,9 @@
 [assembly: AssemblyCompany("Luxoft")]
 [assembly: AssemblyCopyright("Copyright © Luxoft 2024")]
 
-[assembly: AssemblyVersion("1.5.5.0")]
-[assembly: AssemblyFileVersion("1.5.5.0")]
-[assembly: AssemblyInformationalVersion("1.5.5.0")]
+[assembly: AssemblyVersion("1.5.6.0")]
+[assembly: AssemblyFileVersion("1.5.6.0")]
+[assembly: AssemblyInformationalVersion("1.5.6.0")]
 
 #if DEBUG
 [assembly: AssemblyConfiguration("Debug")]