SQLite Query: Fix more DateOnly/TimeOnly cases

Fixes #28051
dotnet · Aug 30, 2022 · 0131b32 · 0131b32
1 parent 45d0a4a
commit 0131b32
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 6 deletions.
diff --git a/src/EFCore.Sqlite.Core/Query/Internal/SqliteObjectToStringTranslator.cs b/src/EFCore.Sqlite.Core/Query/Internal/SqliteObjectToStringTranslator.cs
@@ -19,6 +19,7 @@ public class SqliteObjectToStringTranslator : IMethodCallTranslator
         typeof(byte),
         typeof(byte[]),
         typeof(char),
+        typeof(DateOnly),
         typeof(DateTime),
         typeof(DateTimeOffset),
         typeof(decimal),
@@ -29,6 +30,7 @@ public class SqliteObjectToStringTranslator : IMethodCallTranslator
         typeof(long),
         typeof(sbyte),
         typeof(short),
+        typeof(TimeOnly),
         typeof(TimeSpan),
         typeof(uint),
         typeof(ushort)

diff --git a/src/EFCore.Sqlite.Core/Query/Internal/SqliteSqlTranslatingExpressionVisitor.cs b/src/EFCore.Sqlite.Core/Query/Internal/SqliteSqlTranslatingExpressionVisitor.cs
@@ -19,11 +19,18 @@ private static readonly IReadOnlyDictionary<ExpressionType, IReadOnlyCollection<
         {
             [ExpressionType.Add] = new HashSet<Type>
             {
+                typeof(DateOnly),
                 typeof(DateTime),
                 typeof(DateTimeOffset),
+                typeof(TimeOnly),
                 typeof(TimeSpan)
             },
-            [ExpressionType.Divide] = new HashSet<Type> { typeof(TimeSpan), typeof(ulong) },
+            [ExpressionType.Divide] = new HashSet<Type>
+            {
+                typeof(TimeOnly),
+                typeof(TimeSpan),
+                typeof(ulong)
+            },
             [ExpressionType.GreaterThan] = new HashSet<Type>
             {
                 typeof(DateTimeOffset),
@@ -49,11 +56,18 @@ private static readonly IReadOnlyDictionary<ExpressionType, IReadOnlyCollection<
                 typeof(ulong)
             },
             [ExpressionType.Modulo] = new HashSet<Type> { typeof(ulong) },
-            [ExpressionType.Multiply] = new HashSet<Type> { typeof(TimeSpan), typeof(ulong) },
+            [ExpressionType.Multiply] = new HashSet<Type>
+            {
+                typeof(TimeOnly),
+                typeof(TimeSpan),
+                typeof(ulong)
+            },
             [ExpressionType.Subtract] = new HashSet<Type>
             {
+                typeof(DateOnly),
                 typeof(DateTime),
                 typeof(DateTimeOffset),
+                typeof(TimeOnly),
                 typeof(TimeSpan)
             }
         };
@@ -120,7 +134,8 @@ protected override Expression VisitUnary(UnaryExpression unaryExpression)
                     visitedExpression.Type);
             }
 
-            if (operandType == typeof(TimeSpan))
+            if (operandType == typeof(TimeOnly)
+                || operandType == typeof(TimeSpan))
             {
                 return QueryCompilationContext.NotTranslatedExpression;
             }

diff --git a/test/EFCore.Sqlite.FunctionalTests/Query/GearsOfWarQuerySqliteTest.cs b/test/EFCore.Sqlite.FunctionalTests/Query/GearsOfWarQuerySqliteTest.cs
@@ -646,7 +646,7 @@ public override async Task Where_TimeOnly_IsBetween(bool async)
     public override async Task Where_TimeOnly_subtract_TimeOnly(bool async)
     {
         // TimeSpan. Issue #18844.
-        await Assert.ThrowsAsync<InvalidCastException>(() => base.Where_TimeOnly_subtract_TimeOnly(async));
+        await AssertTranslationFailed(() => base.Where_TimeOnly_subtract_TimeOnly(async));
 
         AssertSql();
     }

diff --git a/test/EFCore.Sqlite.FunctionalTests/Query/TPCGearsOfWarQuerySqliteTest.cs b/test/EFCore.Sqlite.FunctionalTests/Query/TPCGearsOfWarQuerySqliteTest.cs
@@ -368,7 +368,7 @@ public override Task Where_TimeOnly_IsBetween(bool async)
 
     public override Task Where_TimeOnly_subtract_TimeOnly(bool async)
         // TimeSpan. Issue #18844.
-        => Assert.ThrowsAsync<InvalidCastException>(() => base.Where_TimeOnly_subtract_TimeOnly(async));
+        => AssertTranslationFailed(() => base.Where_TimeOnly_subtract_TimeOnly(async));
 
     private void AssertSql(params string[] expected)
         => Fixture.TestSqlLoggerFactory.AssertBaseline(expected);

diff --git a/test/EFCore.Sqlite.FunctionalTests/Query/TPTGearsOfWarQuerySqliteTest.cs b/test/EFCore.Sqlite.FunctionalTests/Query/TPTGearsOfWarQuerySqliteTest.cs
@@ -368,7 +368,7 @@ public override Task Where_TimeOnly_IsBetween(bool async)
 
     public override Task Where_TimeOnly_subtract_TimeOnly(bool async)
         // TimeSpan. Issue #18844.
-        => Assert.ThrowsAsync<InvalidCastException>(() => base.Where_TimeOnly_subtract_TimeOnly(async));
+        => AssertTranslationFailed(() => base.Where_TimeOnly_subtract_TimeOnly(async));
 
     private void AssertSql(params string[] expected)
         => Fixture.TestSqlLoggerFactory.AssertBaseline(expected);