Assert payload serialized size

[controller.git] / opendaylight / md-sal / cds-access-api / src / test / java / org / opendaylight / controller / cluster / access / concepts / RequestEnvelopeTest.java

diff --git a/opendaylight/md-sal/cds-access-api/src/test/java/org/opendaylight/controller/cluster/access/concepts/RequestEnvelopeTest.java b/opendaylight/md-sal/cds-access-api/src/test/java/org/opendaylight/controller/cluster/access/concepts/RequestEnvelopeTest.java
index d270a0c42387751b1d35d90781cea499ec1f7f3d..f5e1f28590f3e470680d101923435c1d26681daf 100644 (file)
--- a/opendaylight/md-sal/cds-access-api/src/test/java/org/opendaylight/controller/cluster/access/concepts/RequestEnvelopeTest.java
+++ b/opendaylight/md-sal/cds-access-api/src/test/java/org/opendaylight/controller/cluster/access/concepts/RequestEnvelopeTest.java
@@ -7,19 +7,21 @@
  */
 package org.opendaylight.controller.cluster.access.concepts;
 
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.Assert.assertEquals;
+
 import akka.actor.ActorRef;
 import akka.actor.ActorSystem;
 import akka.actor.ExtendedActorSystem;
 import akka.serialization.JavaSerializer;
 import akka.testkit.TestProbe;
 import org.junit.After;
-import org.junit.Assert;
 import org.junit.Before;
 import org.opendaylight.controller.cluster.access.commands.TransactionPurgeRequest;
 import org.opendaylight.controller.cluster.access.commands.TransactionPurgeResponse;
 
 public class RequestEnvelopeTest extends AbstractEnvelopeTest<RequestEnvelope> {
-
     private ActorSystem system;
     private ActorRef replyTo;
     private TestProbe replyToProbe;
@@ -33,31 +35,33 @@ public class RequestEnvelopeTest extends AbstractEnvelopeTest<RequestEnvelope> {
     }
 
     @Override
-    protected RequestEnvelope createEnvelope() {
+    protected EnvelopeDetails<RequestEnvelope> createEnvelope() {
         replyToProbe = new TestProbe(system);
         replyTo = replyToProbe.ref();
-        final TransactionPurgeRequest message = new TransactionPurgeRequest(OBJECT, 2L, replyTo);
-        return new RequestEnvelope(message, 1L, 2L);
+        final int refSize = replyTo.path().toSerializationFormat().length();
+
+        return new EnvelopeDetails<>(new RequestEnvelope(new TransactionPurgeRequest(OBJECT, 2L, replyTo), 1L, 2L),
+            refSize + 584);
     }
 
     @Override
     protected void doAdditionalAssertions(final RequestEnvelope envelope, final RequestEnvelope resolvedObject) {
         final Request<?, ?> actual = resolvedObject.getMessage();
-        Assert.assertTrue(actual instanceof TransactionPurgeRequest);
-        final TransactionPurgeRequest purgeRequest = (TransactionPurgeRequest) actual;
-        Assert.assertEquals(replyTo, purgeRequest.getReplyTo());
-        final TransactionPurgeResponse response = new TransactionPurgeResponse(OBJECT, 2L);
+        assertThat(actual, instanceOf(TransactionPurgeRequest.class));
+        final var purgeRequest = (TransactionPurgeRequest) actual;
+        assertEquals(replyTo, purgeRequest.getReplyTo());
+        final var response = new TransactionPurgeResponse(OBJECT, 2L);
         resolvedObject.sendSuccess(response, 11L);
-        final SuccessEnvelope successEnvelope = replyToProbe.expectMsgClass(SuccessEnvelope.class);
-        Assert.assertEquals(response, successEnvelope.getMessage());
-        final RuntimeRequestException failResponse = new RuntimeRequestException("fail", new RuntimeException());
+        final var successEnvelope = replyToProbe.expectMsgClass(SuccessEnvelope.class);
+        assertEquals(response, successEnvelope.getMessage());
+        final var failResponse = new RuntimeRequestException("fail", new RuntimeException());
         resolvedObject.sendFailure(failResponse, 11L);
-        final FailureEnvelope failureEnvelope = replyToProbe.expectMsgClass(FailureEnvelope.class);
-        Assert.assertEquals(failResponse, failureEnvelope.getMessage().getCause());
+        final var failureEnvelope = replyToProbe.expectMsgClass(FailureEnvelope.class);
+        assertEquals(failResponse, failureEnvelope.getMessage().getCause());
     }
 
     @After
-    public void tearDown() throws Exception {
+    public void tearDown() {
         system.terminate();
     }
 }
\ No newline at end of file