Adding test case when scale is null

data-integrations · Oct 6, 2023 · 540dba4 · 540dba4
1 parent ad3f91b
commit 540dba4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/wrangler-core/src/main/java/io/cdap/wrangler/utils/ColumnConverter.java b/wrangler-core/src/main/java/io/cdap/wrangler/utils/ColumnConverter.java
@@ -318,7 +318,7 @@ public static Schema getSchemaForType(String type, Integer scale) throws Directi
     if (type.equals(ColumnTypeNames.DECIMAL)) {
       // TODO make set-type support setting decimal precision
       scale = scale != null ? scale : 38;
-      typeSchema = Schema.nullableOf(Schema.decimalOf(76, scale));
+      typeSchema = Schema.nullableOf(Schema.decimalOf(77, scale));
     } else {
       if (!SCHEMA_TYPE_MAP.containsKey(type)) {
         throw new DirectiveParseException(String.format("'%s' is an unsupported type. " +

diff --git a/wrangler-core/src/test/java/io/cdap/directives/column/SetTypeTest.java b/wrangler-core/src/test/java/io/cdap/directives/column/SetTypeTest.java
@@ -213,6 +213,17 @@ public void testToDecimalInvalidRoundingMode() throws Exception {
     TestingRig.execute(directives, rows);
   }
 
+  @Test
+  public void testToDecimalScaleIsNull() throws Exception {
+    List<Row> rows = Collections.singletonList(new Row("scale_2", "125.45"));
+    String[] directives = new String[] {"set-type scale_2 decimal"};
+    List<Row> results = TestingRig.execute(directives, rows);
+    Row row = results.get(0);
+
+    Assert.assertTrue(row.getValue(0) instanceof BigDecimal);
+    Assert.assertEquals(row.getValue(0), new BigDecimal("125.45"));
+  }
+
   @Test
   public void testToBoolean() throws Exception {
     List<Row> trueRows = Collections.singletonList(