Merge pull request #9773 from RyanSkraba/BEAM-7073-avro-sql-unit-test
[BEAM-7073]: Add unit test for Avro logical type datum.
diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/schemas/utils/AvroUtilsTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/schemas/utils/AvroUtilsTest.java
index cedeb77..3679e21 100644
--- a/sdks/java/core/src/test/java/org/apache/beam/sdk/schemas/utils/AvroUtilsTest.java
+++ b/sdks/java/core/src/test/java/org/apache/beam/sdk/schemas/utils/AvroUtilsTest.java
@@ -502,8 +502,14 @@
@Test
public void testGenericRecordToBeamRow() {
+ GenericRecord genericRecord = getGenericRecord();
Row row = AvroUtils.toBeamRowStrict(getGenericRecord(), null);
assertEquals(getBeamRow(), row);
+
+ // Alternatively, a timestamp-millis logical type can have a joda datum.
+ genericRecord.put("timestampMillis", new DateTime(genericRecord.get("timestampMillis")));
+ row = AvroUtils.toBeamRowStrict(getGenericRecord(), null);
+ assertEquals(getBeamRow(), row);
}
@Test
