Adding Cloning tests for ClientWorker + not cloning Instant and Duration

framework/tst-self/dslabs/framework/testing/utils/CloningTest.java

@@ -38,6 +38,7 @@
 import dslabs.framework.Message;
 import dslabs.framework.Node;
 import dslabs.framework.Result;
+import dslabs.framework.testing.ClientWorker;
 import dslabs.framework.testing.Event;
 import dslabs.framework.testing.LocalAddress;
 import dslabs.framework.testing.MessageEnvelope;
@@ -126,6 +127,11 @@ public void cloning() {
     clonesEqual(new MessageExample("foo", false));
     clonesEqual(Address.subAddress(new AddressExample("foo"), "bar"));
 
+    clonesEqual(
+        new ClientWorker(
+            new ClientExample(new AddressExample("foo")),
+            Workload.builder().commands(new CommandExample("c")).build()));
+
     clonesEqual(new AlsoNotFastSerializable());
     clonesEqual(new ShouldFastSerialize());
   }
@@ -139,6 +145,11 @@ public void serializationDeserialization() {
     serializeDeserializeEqual(new MessageExample("foo", false));
     serializeDeserializeEqual(Address.subAddress(new AddressExample("foo"), "bar"));
 
+    serializeDeserializeEqual(
+        new ClientWorker(
+            new ClientExample(new AddressExample("foo")),
+            Workload.builder().commands(new CommandExample("c")).build()));
+
     serializeDeserializeEqual(new AlsoNotFastSerializable());
   }
 
@@ -151,6 +162,11 @@ public void fastCloneWorks() throws NoSuchFieldException, IllegalAccessException
     Cloning.clone(new MessageExample("foo", false));
     Cloning.clone(Address.subAddress(new AddressExample("foo"), "bar"));
 
+    Cloning.clone(
+        new ClientWorker(
+            new ClientExample(new AddressExample("foo")),
+            Workload.builder().commands(new CommandExample("c")).build()));
+
     assertFastCloned();
 
     Cloning.clone(new ShouldFastSerialize());
@@ -495,6 +511,16 @@ public Result execute(Command command) {
   }
 }
 
+@Data
+class CommandExample implements Command {
+  private final String s;
+}
+
+@Data
+class ResultExample implements Result {
+  private final int i;
+}
+
 @Data
 class MessageExample implements Message {
   private final String s;

framework/tst/dslabs/framework/testing/utils/Cloning.java

@@ -27,6 +27,8 @@
 import java.io.Serializable;
 import java.lang.reflect.Field;
 import java.lang.reflect.Modifier;
+import java.time.Duration;
+import java.time.Instant;
 import java.util.Collections;
 import java.util.Objects;
 import java.util.Set;
@@ -82,6 +84,10 @@ public Strategy strategyFor(Object toBeCloned, Field field) {
             return Strategy.IGNORE;
           }
         });
+    // Instant is a final class and is immutable and thread-safe.
+    jdclCloner.registerImmutable(Instant.class);
+    // Duration is a final class and is immutable and thread-safe.
+    jdclCloner.registerImmutable(Duration.class);
   }
 
   private static <T extends Serializable> T jdclClone(T object) {