Add PushPredicateIntoTableScan test where pushed domain is narrower

trinodb · Sep 30, 2019 · 32dcec6 · 32dcec6
1 parent 78771fa
commit 32dcec6
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/...src/test/java/io/prestosql/sql/planner/iterative/rule/TestPushPredicateIntoTableScan.java b/...src/test/java/io/prestosql/sql/planner/iterative/rule/TestPushPredicateIntoTableScan.java
@@ -15,6 +15,7 @@
 
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
+import io.airlift.slice.Slices;
 import io.prestosql.connector.CatalogName;
 import io.prestosql.metadata.TableHandle;
 import io.prestosql.plugin.tpch.TpchColumnHandle;
@@ -144,6 +145,26 @@ public void consumesDeterministicPredicateIfNewDomainIsWider()
                         ImmutableMap.of("nationkey", "nationkey")));
     }
 
+    @Test
+    public void consumesDeterministicPredicateIfNewDomainIsNarrower()
+    {
+        Type orderStatusType = createVarcharType(1);
+        ColumnHandle columnHandle = new TpchColumnHandle("orderstatus", orderStatusType);
+        Map<String, Domain> filterConstraint = ImmutableMap.<String, Domain>builder()
+                .put("orderstatus", singleValue(orderStatusType, utf8Slice("O")))
+                .build();
+        tester().assertThat(pushPredicateIntoTableScan)
+                .on(p -> p.filter(expression("orderstatus = 'O' OR orderstatus = 'F'"),
+                        p.tableScan(
+                                ordersTableHandle,
+                                ImmutableList.of(p.symbol("orderstatus", orderStatusType)),
+                                ImmutableMap.of(p.symbol("orderstatus", orderStatusType), new TpchColumnHandle("orderstatus", orderStatusType)),
+                                TupleDomain.withColumnDomains(ImmutableMap.of(
+                                        columnHandle, Domain.multipleValues(orderStatusType, ImmutableList.of(Slices.utf8Slice("O"), Slices.utf8Slice("P"))))))))
+                .matches(
+                        constrainedTableScanWithTableLayout("orders", filterConstraint, ImmutableMap.of("orderstatus", "orderstatus")));
+    }
+
     @Test
     public void doesNotConsumeRemainingPredicateIfNewDomainIsWider()
     {