fix: Ensure ConstantTypedExpr and variant types are equivalent

Author: pramodsatya

velox/core/Expressions.h

@@ -61,7 +61,11 @@ class ConstantTypedExpr : public ITypedExpr {
   // Variant::null() value is supported.
   ConstantTypedExpr(TypePtr type, Variant value)
       : ITypedExpr{ExprKind::kConstant, std::move(type)},
-        value_{std::move(value)} {}
+        value_{std::move(value)} {
+    VELOX_CHECK(
+        value_.isTypeCompatible(ITypedExpr::type()),
+        "Expression type does not match variant type");
+  }
 
   // Creates constant expression of scalar or complex type. The value comes from
   // index zero.

velox/core/tests/ConstantTypedExprTest.cpp

@@ -15,6 +15,7 @@
  */
 #include <gtest/gtest.h>
 
+#include "velox/common/base/tests/GTestUtils.h"
 #include "velox/common/memory/Memory.h"
 #include "velox/core/Expressions.h"
 #include "velox/functions/prestosql/types/HyperLogLogType.h"
@@ -579,4 +580,21 @@ TEST_F(ConstantTypedExprTest, toStringComplexTypes) {
       << "toString mismatch for OPAQUE variant vs vector";
 }
 
+TEST_F(ConstantTypedExprTest, variantTypeCheck) {
+  VELOX_ASSERT_THROW(
+      std::make_shared<core::ConstantTypedExpr>(INTEGER(), "abc"),
+      "Expression type does not match variant type");
+  VELOX_ASSERT_THROW(
+      std::make_shared<core::ConstantTypedExpr>(INTEGER(), variant(123LL)),
+      "Expression type does not match variant type");
+  VELOX_ASSERT_THROW(
+      std::make_shared<core::ConstantTypedExpr>(
+          ARRAY(VARCHAR()), variant::array({1, 2, 3})),
+      "Expression type does not match variant type");
+  VELOX_ASSERT_THROW(
+      std::make_shared<core::ConstantTypedExpr>(
+          MAP(INTEGER(), VARCHAR()), variant::map({{"abc", "xyz"}})),
+      "Expression type does not match variant type");
+}
+
 } // namespace facebook::velox::core::test

velox/expression/tests/ExprRewriteRegistryTest.cpp

@@ -31,8 +31,7 @@ TEST_F(ExprRewriteRegistryTest, basic) {
       };
   registry.registerRewrite(testRewrite);
 
-  auto input =
-      std::make_shared<core::ConstantTypedExpr>(BIGINT(), variant(123));
+  auto input = std::make_shared<core::ConstantTypedExpr>(BIGINT(), 123LL);
   const auto rewritten = registry.rewrite(input);
   ASSERT_TRUE(rewritten->isCallKind());
   ASSERT_TRUE(rewritten->type()->isBigint());

velox/expression/tests/ExprTest.cpp

@@ -2595,8 +2595,8 @@ TEST_P(ParameterizedExprTest, constantToString) {
 
 TEST_F(ExprTest, constantEqualsNullConsistency) {
   // Constant expr created using variant
-  auto nullVariantToExpr =
-      std::make_shared<core::ConstantTypedExpr>(VARCHAR(), Variant{});
+  auto nullVariantToExpr = std::make_shared<core::ConstantTypedExpr>(
+      VARCHAR(), variant::null(TypeKind::VARCHAR));
   auto nonNullVariantToExpr =
       std::make_shared<core::ConstantTypedExpr>(VARCHAR(), Variant{"test"});