TestKit backend: add full support for temporal types

testkit-backend/src/main/java/neo4j/org/testkit/backend/TestkitState.java

@@ -40,7 +40,6 @@
 import neo4j.org.testkit.backend.holder.TransactionHolder;
 import neo4j.org.testkit.backend.messages.requests.TestkitCallbackResult;
 import neo4j.org.testkit.backend.messages.responses.TestkitResponse;
-import org.neo4j.driver.exceptions.Neo4jException;
 import org.neo4j.driver.internal.cluster.RoutingTableRegistry;
 import reactor.core.publisher.Mono;
 
@@ -69,7 +68,7 @@ public class TestkitState {
     private final Map<String, ReactiveTransactionHolder> transactionIdToReactiveTransactionHolder = new HashMap<>();
 
     @Getter
-    private final Map<String, Neo4jException> errors = new HashMap<>();
+    private final Map<String, Exception> errors = new HashMap<>();
 
     private final AtomicInteger idGenerator = new AtomicInteger(0);
 

testkit-backend/src/main/java/neo4j/org/testkit/backend/channel/handler/TestkitRequestProcessorHandler.java

@@ -21,6 +21,7 @@
 import io.netty.channel.Channel;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelInboundHandlerAdapter;
+import java.time.zone.ZoneRulesException;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.CompletionException;
 import java.util.concurrent.CompletionStage;
@@ -120,8 +121,10 @@ private TestkitResponse createErrorResponse(Throwable throwable) {
                     .build();
         } else if (isConnectionPoolClosedException(throwable)
                 || throwable instanceof UntrustedServerException
-                || throwable instanceof NoSuchRecordException) {
+                || throwable instanceof NoSuchRecordException
+                || throwable instanceof ZoneRulesException) {
             String id = testkitState.newId();
+            testkitState.getErrors().put(id, (Exception) throwable);
             return DriverError.builder()
                     .data(DriverError.DriverErrorBody.builder()
                             .id(id)

testkit-backend/src/main/java/neo4j/org/testkit/backend/channel/handler/TestkitRequestResponseMapperHandler.java

@@ -18,28 +18,24 @@
  */
 package neo4j.org.testkit.backend.channel.handler;
 
-import com.fasterxml.jackson.core.JsonProcessingException;
 import com.fasterxml.jackson.databind.DeserializationFeature;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import io.netty.channel.ChannelDuplexHandler;
 import io.netty.channel.ChannelHandlerContext;
 import io.netty.channel.ChannelPromise;
 import neo4j.org.testkit.backend.messages.TestkitModule;
 import neo4j.org.testkit.backend.messages.requests.TestkitRequest;
+import neo4j.org.testkit.backend.messages.responses.BackendError;
 import neo4j.org.testkit.backend.messages.responses.TestkitResponse;
 
 public class TestkitRequestResponseMapperHandler extends ChannelDuplexHandler {
     private final ObjectMapper objectMapper = newObjectMapper();
 
     @Override
-    public void channelRead(ChannelHandlerContext ctx, Object msg) {
+    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
         String testkitMessage = (String) msg;
         TestkitRequest testkitRequest;
-        try {
-            testkitRequest = objectMapper.readValue(testkitMessage, TestkitRequest.class);
-        } catch (JsonProcessingException e) {
-            throw new RuntimeException("Failed to deserialize Testkit message", e);
-        }
+        testkitRequest = objectMapper.readValue(testkitMessage, TestkitRequest.class);
         ctx.fireChannelRead(testkitRequest);
     }
 
@@ -50,6 +46,16 @@ public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise)
         ctx.writeAndFlush(responseStr, promise);
     }
 
+    @Override
+    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
+        TestkitResponse response = BackendError.builder()
+                .data(BackendError.BackendErrorBody.builder()
+                        .msg(cause.toString())
+                        .build())
+                .build();
+        ctx.writeAndFlush(objectMapper.writeValueAsString(response));
+    }
+
     public static ObjectMapper newObjectMapper() {
         ObjectMapper objectMapper = new ObjectMapper();
         TestkitModule testkitModule = new TestkitModule();

testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/TestkitModule.java

@@ -19,33 +19,60 @@
 package neo4j.org.testkit.backend.messages;
 
 import com.fasterxml.jackson.databind.module.SimpleModule;
-import java.time.ZonedDateTime;
+import java.time.LocalDate;
 import java.util.List;
+import neo4j.org.testkit.backend.messages.requests.deserializer.TestkitCypherDateDeserializer;
 import neo4j.org.testkit.backend.messages.requests.deserializer.TestkitCypherDateTimeDeserializer;
+import neo4j.org.testkit.backend.messages.requests.deserializer.TestkitCypherDurationDeserializer;
+import neo4j.org.testkit.backend.messages.requests.deserializer.TestkitCypherTimeDeserializer;
 import neo4j.org.testkit.backend.messages.requests.deserializer.TestkitListDeserializer;
+import neo4j.org.testkit.backend.messages.requests.deserializer.types.CypherDateTime;
+import neo4j.org.testkit.backend.messages.requests.deserializer.types.CypherTime;
+import neo4j.org.testkit.backend.messages.responses.serializer.TestkitDateTimeValueSerializer;
+import neo4j.org.testkit.backend.messages.responses.serializer.TestkitDateValueSerializer;
+import neo4j.org.testkit.backend.messages.responses.serializer.TestkitDurationValueSerializer;
 import neo4j.org.testkit.backend.messages.responses.serializer.TestkitListValueSerializer;
+import neo4j.org.testkit.backend.messages.responses.serializer.TestkitLocalDateTimeValueSerializer;
+import neo4j.org.testkit.backend.messages.responses.serializer.TestkitLocalTimeValueSerializer;
 import neo4j.org.testkit.backend.messages.responses.serializer.TestkitMapValueSerializer;
 import neo4j.org.testkit.backend.messages.responses.serializer.TestkitNodeValueSerializer;
 import neo4j.org.testkit.backend.messages.responses.serializer.TestkitPathValueSerializer;
 import neo4j.org.testkit.backend.messages.responses.serializer.TestkitRecordSerializer;
 import neo4j.org.testkit.backend.messages.responses.serializer.TestkitRelationshipValueSerializer;
+import neo4j.org.testkit.backend.messages.responses.serializer.TestkitTimeValueSerializer;
 import neo4j.org.testkit.backend.messages.responses.serializer.TestkitValueSerializer;
 import org.neo4j.driver.Record;
 import org.neo4j.driver.Value;
+import org.neo4j.driver.internal.value.DateTimeValue;
+import org.neo4j.driver.internal.value.DateValue;
+import org.neo4j.driver.internal.value.DurationValue;
 import org.neo4j.driver.internal.value.ListValue;
+import org.neo4j.driver.internal.value.LocalDateTimeValue;
+import org.neo4j.driver.internal.value.LocalTimeValue;
 import org.neo4j.driver.internal.value.MapValue;
 import org.neo4j.driver.internal.value.NodeValue;
 import org.neo4j.driver.internal.value.PathValue;
 import org.neo4j.driver.internal.value.RelationshipValue;
+import org.neo4j.driver.internal.value.TimeValue;
+import org.neo4j.driver.types.IsoDuration;
 
 public class TestkitModule extends SimpleModule {
     public TestkitModule() {
         this.addDeserializer(List.class, new TestkitListDeserializer());
-        this.addDeserializer(ZonedDateTime.class, new TestkitCypherDateTimeDeserializer());
+        this.addDeserializer(CypherDateTime.class, new TestkitCypherDateTimeDeserializer());
+        this.addDeserializer(CypherTime.class, new TestkitCypherTimeDeserializer());
+        this.addDeserializer(IsoDuration.class, new TestkitCypherDurationDeserializer());
+        this.addDeserializer(LocalDate.class, new TestkitCypherDateDeserializer());
 
         this.addSerializer(Value.class, new TestkitValueSerializer());
         this.addSerializer(NodeValue.class, new TestkitNodeValueSerializer());
         this.addSerializer(ListValue.class, new TestkitListValueSerializer());
+        this.addSerializer(DateTimeValue.class, new TestkitDateTimeValueSerializer());
+        this.addSerializer(DateValue.class, new TestkitDateValueSerializer());
+        this.addSerializer(DurationValue.class, new TestkitDurationValueSerializer());
+        this.addSerializer(LocalDateTimeValue.class, new TestkitLocalDateTimeValueSerializer());
+        this.addSerializer(LocalTimeValue.class, new TestkitLocalTimeValueSerializer());
+        this.addSerializer(TimeValue.class, new TestkitTimeValueSerializer());
         this.addSerializer(Record.class, new TestkitRecordSerializer());
         this.addSerializer(MapValue.class, new TestkitMapValueSerializer());
         this.addSerializer(PathValue.class, new TestkitPathValueSerializer());

testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionReadTransaction.java

@@ -23,7 +23,6 @@
 import java.util.concurrent.ExecutionException;
 import lombok.Getter;
 import lombok.Setter;
-import neo4j.org.testkit.backend.FrontendError;
 import neo4j.org.testkit.backend.ReactiveTransactionContextAdapter;
 import neo4j.org.testkit.backend.TestkitState;
 import neo4j.org.testkit.backend.holder.AsyncTransactionHolder;
@@ -38,7 +37,6 @@
 import org.neo4j.driver.TransactionWork;
 import org.neo4j.driver.async.AsyncSession;
 import org.neo4j.driver.async.AsyncTransactionWork;
-import org.neo4j.driver.exceptions.Neo4jException;
 import org.neo4j.driver.reactive.ReactiveTransactionCallback;
 import org.neo4j.driver.reactive.RxTransactionWork;
 import org.reactivestreams.Publisher;
@@ -129,11 +127,8 @@ private TransactionWork<Void> handle(TestkitState testkitState, SessionHolder se
                 if (workThrowable instanceof ExecutionException) {
                     workThrowable = workThrowable.getCause();
                 }
-                if (workThrowable instanceof Neo4jException) {
-                    throw (Neo4jException) workThrowable;
-                }
-                if (workThrowable instanceof FrontendError) {
-                    throw (FrontendError) workThrowable;
+                if (workThrowable instanceof RuntimeException) {
+                    throw (RuntimeException) workThrowable;
                 }
                 throw new RuntimeException("Unexpected exception occurred in transaction work function", workThrowable);
             }

testkit-backend/src/main/java/neo4j/org/testkit/backend/messages/requests/SessionWriteTransaction.java

@@ -24,7 +24,6 @@
 import java.util.concurrent.ExecutionException;
 import lombok.Getter;
 import lombok.Setter;
-import neo4j.org.testkit.backend.FrontendError;
 import neo4j.org.testkit.backend.ReactiveTransactionContextAdapter;
 import neo4j.org.testkit.backend.TestkitState;
 import neo4j.org.testkit.backend.holder.AsyncTransactionHolder;
@@ -39,7 +38,6 @@
 import org.neo4j.driver.TransactionWork;
 import org.neo4j.driver.async.AsyncSession;
 import org.neo4j.driver.async.AsyncTransactionWork;
-import org.neo4j.driver.exceptions.Neo4jException;
 import org.neo4j.driver.reactive.ReactiveTransactionCallback;
 import org.neo4j.driver.reactive.RxTransactionWork;
 import org.reactivestreams.Publisher;
@@ -130,11 +128,8 @@ private TransactionWork<Void> handle(TestkitState testkitState, SessionHolder se
                 if (workThrowable instanceof ExecutionException) {
                     workThrowable = workThrowable.getCause();
                 }
-                if (workThrowable instanceof Neo4jException) {
-                    throw (Neo4jException) workThrowable;
-                }
-                if (workThrowable instanceof FrontendError) {
-                    throw (FrontendError) workThrowable;
+                if (workThrowable instanceof RuntimeException) {
+                    throw (RuntimeException) workThrowable;
                 }
                 throw new RuntimeException("Unexpected exception occurred in transaction work function", workThrowable);
             }