Add a decimal field in record

raw-labs · Feb 26, 2025 · 9281f60 · 9281f60
1 parent 93753d6
commit 9281f60
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/src/test/scala/com/rawlabs/das/mock/DASMock.scala b/src/test/scala/com/rawlabs/das/mock/DASMock.scala
@@ -239,6 +239,10 @@ class DASMock(options: Map[String, String]) extends DASSdk with StrictLogging {
                     .setName("intField")
                     .setTipe(Type.newBuilder().setInt(IntType.newBuilder()).build())
                     .build())
+                  .addAtts(AttrType
+                    .newBuilder()
+                    .setName("decimalField")
+                    .setTipe(Type.newBuilder().setDecimal(DecimalType.newBuilder()).build()))
                   .addAtts(AttrType
                     .newBuilder()
                     .setName("binaryField")

diff --git a/src/test/scala/com/rawlabs/das/mock/DASMockAllTypesTable.scala b/src/test/scala/com/rawlabs/das/mock/DASMockAllTypesTable.scala
@@ -150,6 +150,11 @@ class DASMockAllTypesTable(maxRows: Int) extends DASTable {
                       .newBuilder()
                       .setName("intField")
                       .setValue(Value.newBuilder().setInt(ValueInt.newBuilder().setV(i).build()).build()))
+                  .addAtts(ValueRecordAttr
+                    .newBuilder()
+                    .setName("decimalField")
+                    .setValue(
+                      Value.newBuilder().setDecimal(ValueDecimal.newBuilder().setV(i.toString).build()).build()))
                   .addAtts(
                     ValueRecordAttr
                       .newBuilder()