commit 570c5987d6668a15c86d986ab8cce42785df2583
author James Turton <9107319+jnturton@users.noreply.github.com> Fri Aug 12 03:32:36 2022 +0200
committer GitHub <noreply@github.com> Thu Aug 11 18:32:36 2022 -0700
tree 525adab5a8ccbdfdd6e1fcbd5864c341ffafba7f
parent 6263f713891ddfbef547abf6dbd65bf12d1edf5d

More unit tests covering the remaining column types affected by DRILL-8273. (#2620)

exec/java-exec/src/test/java/org/apache/drill/exec/physical/resultSet/project/TestTupleProjection.java

diff --git a/exec/java-exec/src/test/java/org/apache/drill/exec/physical/resultSet/project/TestTupleProjection.java b/exec/java-exec/src/test/java/org/apache/drill/exec/physical/resultSet/project/TestTupleProjection.java
index 3fa27ff..157c785 100644
--- a/exec/java-exec/src/test/java/org/apache/drill/exec/physical/resultSet/project/TestTupleProjection.java
+++ b/exec/java-exec/src/test/java/org/apache/drill/exec/physical/resultSet/project/TestTupleProjection.java
@@ -29,6 +29,7 @@
 
 import org.apache.drill.categories.RowSetTest;
 import org.apache.drill.common.expression.SchemaPath;
+import org.apache.drill.common.types.TypeProtos.DataMode;
 import org.apache.drill.common.types.TypeProtos.MinorType;
 import org.apache.drill.common.types.Types;
 import org.apache.drill.exec.physical.resultSet.project.RequestedTuple.TupleProjectionType;
@@ -60,13 +61,25 @@
       MetadataUtils.newScalar("a", Types.required(MinorType.INT));
   private static final ColumnMetadata UNPROJECTED_SPECIAL_COLUMN =
       MetadataUtils.newScalar("bar", Types.required(MinorType.INT));
-  private static final ColumnMetadata COMPLEX_SPECIAL_COLUMN =
+  private static final ColumnMetadata SPECIAL_DICT =
+      MetadataUtils.newDict("a_dict");
+  private static final ColumnMetadata SPECIAL_MAP =
       MetadataUtils.newMap("a_map");
+  private static final ColumnMetadata SPECIAL_REP_LIST =
+      MetadataUtils.newRepeatedList(
+        "a_repeated_list",
+        MetadataUtils.newScalar("child", Types.repeated(MinorType.INT))
+      );
+  private static final ColumnMetadata SPECIAL_VARIANT =
+      MetadataUtils.newVariant("a_variant", DataMode.OPTIONAL);
 
   static {
     SPECIAL_COLUMN.setBooleanProperty(ColumnMetadata.EXCLUDE_FROM_WILDCARD, true);
     UNPROJECTED_SPECIAL_COLUMN.setBooleanProperty(ColumnMetadata.EXCLUDE_FROM_WILDCARD, true);
-    COMPLEX_SPECIAL_COLUMN.setBooleanProperty(ColumnMetadata.EXCLUDE_FROM_WILDCARD, true);
+    SPECIAL_DICT.setBooleanProperty(ColumnMetadata.EXCLUDE_FROM_WILDCARD, true);
+    SPECIAL_MAP.setBooleanProperty(ColumnMetadata.EXCLUDE_FROM_WILDCARD, true);
+    SPECIAL_REP_LIST.setBooleanProperty(ColumnMetadata.EXCLUDE_FROM_WILDCARD, true);
+    SPECIAL_VARIANT.setBooleanProperty(ColumnMetadata.EXCLUDE_FROM_WILDCARD, true);
   }
 
   /**
@@ -79,7 +92,10 @@
     assertTrue(projSet.isProjected("foo"));
     assertTrue(projSet.isProjected(NORMAL_COLUMN));
     assertFalse(projSet.isProjected(SPECIAL_COLUMN));
-    assertFalse(projSet.isProjected(COMPLEX_SPECIAL_COLUMN));
+    assertFalse(projSet.isProjected(SPECIAL_DICT));
+    assertFalse(projSet.isProjected(SPECIAL_MAP));
+    assertFalse(projSet.isProjected(SPECIAL_REP_LIST));
+    assertFalse(projSet.isProjected(SPECIAL_VARIANT));
     assertTrue(projSet.projections().isEmpty());
     assertFalse(projSet.isEmpty());
   }
@@ -95,7 +111,10 @@
     assertNull(projSet.get("foo"));
     assertTrue(projSet.isProjected(NORMAL_COLUMN));
     assertFalse(projSet.isProjected(SPECIAL_COLUMN));
-    assertFalse(projSet.isProjected(COMPLEX_SPECIAL_COLUMN));
+    assertFalse(projSet.isProjected(SPECIAL_DICT));
+    assertFalse(projSet.isProjected(SPECIAL_MAP));
+    assertFalse(projSet.isProjected(SPECIAL_REP_LIST));
+    assertFalse(projSet.isProjected(SPECIAL_VARIANT));
     assertEquals(1, projSet.projections().size());
     assertFalse(projSet.isEmpty());
   }
@@ -112,7 +131,10 @@
     assertFalse(projSet.isProjected("foo"));
     assertFalse(projSet.isProjected(NORMAL_COLUMN));
     assertFalse(projSet.isProjected(SPECIAL_COLUMN));
-    assertFalse(projSet.isProjected(COMPLEX_SPECIAL_COLUMN));
+    assertFalse(projSet.isProjected(SPECIAL_DICT));
+    assertFalse(projSet.isProjected(SPECIAL_MAP));
+    assertFalse(projSet.isProjected(SPECIAL_REP_LIST));
+    assertFalse(projSet.isProjected(SPECIAL_VARIANT));
     assertTrue(projSet.projections().isEmpty());
     assertTrue(projSet.isEmpty());
   }
@@ -134,7 +156,7 @@
     assertTrue(projSet.isProjected(SPECIAL_COLUMN));
     assertFalse(projSet.isProjected(UNPROJECTED_COLUMN));
     assertFalse(projSet.isProjected(UNPROJECTED_SPECIAL_COLUMN));
-    assertFalse(projSet.isProjected(COMPLEX_SPECIAL_COLUMN));
+    assertFalse(projSet.isProjected(SPECIAL_MAP));
 
     List<RequestedColumn> cols = projSet.projections();
     assertEquals(3, cols.size());