Merge pull request #58 from ppi-ag/bug-util-date

Fixes #56:
ppi-ag · Apr 6, 2021 · 109f637 · 109f637
2 parents b77c47d + 09137fd
commit 109f637
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/...-persistence/src/main/java/de/ppi/deepsampler/persistence/bean/ext/JavaTimeExtension.java b/...-persistence/src/main/java/de/ppi/deepsampler/persistence/bean/ext/JavaTimeExtension.java
@@ -7,12 +7,14 @@
 
 import java.time.LocalDate;
 import java.time.LocalDateTime;
+import java.util.Date;
 
 public class JavaTimeExtension extends StandardBeanFactoryExtension {
 
     @Override
     public boolean isProcessable(Class<?> beanCls) {
-        return LocalDateTime.class.isAssignableFrom(beanCls) || LocalDate.class.isAssignableFrom(beanCls);
+        return LocalDateTime.class.isAssignableFrom(beanCls) || LocalDate.class.isAssignableFrom(beanCls)
+                || Date.class.isAssignableFrom(beanCls);
     }
 
     @Override

diff --git a/...pler-persistence/src/test/java/de/ppi/deepsampler/persistence/SamplerBeanFactoryTest.java b/...pler-persistence/src/test/java/de/ppi/deepsampler/persistence/SamplerBeanFactoryTest.java
@@ -3,6 +3,7 @@
 import de.ppi.deepsampler.persistence.model.PersistentBean;
 import org.junit.jupiter.api.Test;
 
+import java.sql.Timestamp;
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Collections;
@@ -28,6 +29,24 @@ void testImmutableCollectionBean() {
         assertEquals(bean.collectionOfStrings, persistentBean.getValue("0$collectionOfStrings"));
     }
 
+    @Test
+    void testTimestampSql() {
+        // GIVEN
+        Timestamp ts =  new Timestamp(1L);
+        final TimestampBean timestampBean = new TimestampBean();
+        timestampBean.timestamp = ts;
+
+        // WHEN
+        PersistentBean persistentBean = SamplerBeanFactory.create().toBean(timestampBean);
+
+        // THEN
+        assertEquals(ts, persistentBean.getValue("0$timestamp"));
+    }
+
+    private static class TimestampBean {
+        Timestamp timestamp;
+    }
+
     private static class CollectionBean {
         Collection<String> collectionOfStrings;
     }