diff --git a/maven-projects/info/src/main/java/org/apache/graphar/info/type/DataType.java b/maven-projects/info/src/main/java/org/apache/graphar/info/type/DataType.java index 83c31aec1..d5329338b 100644 --- a/maven-projects/info/src/main/java/org/apache/graphar/info/type/DataType.java +++ b/maven-projects/info/src/main/java/org/apache/graphar/info/type/DataType.java @@ -39,7 +39,13 @@ public enum DataType { STRING, /** List of same type */ - LIST; + LIST, + + /** Date value */ + DATE, + + /** Timestamp value */ + TIMESTAMP; public static DataType fromString(String s) { switch (s) { @@ -57,6 +63,10 @@ public static DataType fromString(String s) { return STRING; case "list": return LIST; + case "date": + return DATE; + case "timestamp": + return TIMESTAMP; default: throw new IllegalArgumentException("Unknown data type: " + s); } diff --git a/maven-projects/info/src/test/java/org/apache/graphar/info/PropertyTest.java b/maven-projects/info/src/test/java/org/apache/graphar/info/PropertyTest.java index 7326ad3c7..78fa86d98 100644 --- a/maven-projects/info/src/test/java/org/apache/graphar/info/PropertyTest.java +++ b/maven-projects/info/src/test/java/org/apache/graphar/info/PropertyTest.java @@ -88,6 +88,17 @@ public void testPropertyWithAllDataTypes() { Property listProp = TestDataFactory.createProperty("items", DataType.LIST, false, true); Assert.assertEquals(DataType.LIST, listProp.getDataType()); TestVerificationUtils.verifyProperty(listProp, "items", false, true); + + // Newly added data types + Property dateProp = + TestDataFactory.createProperty("creationDate", DataType.DATE, false, true); + Assert.assertEquals(DataType.DATE, dateProp.getDataType()); + TestVerificationUtils.verifyProperty(dateProp, "creationDate", false, true); + + Property timestampProp = + TestDataFactory.createProperty("createdAt", DataType.TIMESTAMP, false, true); + Assert.assertEquals(DataType.TIMESTAMP, timestampProp.getDataType()); + TestVerificationUtils.verifyProperty(timestampProp, "createdAt", false, true); } @Test