[CYB-171] test fix

cloudera · Feb 12, 2024 · f550b5d · f550b5d
1 parent 835fed0
commit f550b5d
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/...hive/src/test/java/com/cloudera/cyber/indexing/hive/tableapi/TableApiAbstractJobTest.java b/...hive/src/test/java/com/cloudera/cyber/indexing/hive/tableapi/TableApiAbstractJobTest.java
@@ -33,11 +33,25 @@ public static Stream<Arguments> mappingsData() {
 
         Arguments.of(Collections.singletonMap(GIVEN_TABLE_NAME, ResolvedSchema.of(
                 Column.physical("column1", DataTypes.STRING()),
-                Column.physical("column2", DataTypes.STRING()))),
+                Column.physical("column2", DataTypes.STRING()),
+                Column.physical("column3", DataTypes.INT()),
+                Column.physical("column4", DataTypes.BOOLEAN()),
+                Column.physical("column5", DataTypes.BIGINT()),
+                Column.physical("column6", DataTypes.DOUBLE()),
+                Column.physical("column7", DataTypes.FLOAT()),
+                Column.physical("column8", DataTypes.SMALLINT()),
+                Column.physical("column9", DataTypes.TINYINT()))),
             Collections.singletonMap(GIVEN_SOURCE,
                 new MappingDto(GIVEN_TABLE_NAME, new ArrayList<>(), Arrays.asList(
                     new MappingColumnDto("column1", null, null, null, false),
-                    new MappingColumnDto("column2", null, null, null, false))))));
+                    new MappingColumnDto("column2", null, null, null, false),
+                    new MappingColumnDto("column3", null, null, null, false),
+                    new MappingColumnDto("column4", null, null, null, false),
+                    new MappingColumnDto("column5", null, null, null, false),
+                    new MappingColumnDto("column6", null, null, null, false),
+                    new MappingColumnDto("column7", null, null, null, false),
+                    new MappingColumnDto("column8", null, null, null, false),
+                    new MappingColumnDto("column9", null, null, null, false))))));
   }
 
   public static Stream<Arguments> mappingsExceptionData() {
@@ -65,8 +79,8 @@ public static Stream<Arguments> mappingsExceptionData() {
                 GIVEN_SOURCE, "[ , someName]")),
 
         Arguments.of(Collections.singletonMap(GIVEN_TABLE_NAME, ResolvedSchema.of(
-                Column.physical("column1", DataTypes.INT()),
-                Column.physical("column2", DataTypes.INT()),
+                Column.physical("column1", DataTypes.DATE()),
+                Column.physical("column2", DataTypes.DATE()),
                 Column.physical("column3", DataTypes.STRING()))),
             Collections.singletonMap(GIVEN_SOURCE,
                 new MappingDto(GIVEN_TABLE_NAME, new ArrayList<>(), Arrays.asList(
@@ -78,7 +92,6 @@ public static Stream<Arguments> mappingsExceptionData() {
                 "Found column mappings of non-string type without transformations for source [%s]: %s",
                 GIVEN_SOURCE, "[column1]")));
   }
-
   @ParameterizedTest
   @MethodSource("mappingsData")
   void shouldValidateMappings(Map<String, ResolvedSchema> givenTableSchemaMap,