Add AbstractProxyTransaction derived classes tests

[controller.git] / opendaylight / md-sal / sal-distributed-datastore / src / test / java / org / opendaylight / controller / cluster / databroker / actors / dds / TransactionTester.java

diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/actors/dds/TransactionTester.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/actors/dds/TransactionTester.java
new file mode 100644 (file)
index 0000000..1bcf556
--- /dev/null
+++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/actors/dds/TransactionTester.java
@@ -0,0 +1,98 @@
+/*
+ * Copyright (c) 2017 Pantheon Technologies s.r.o. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.controller.cluster.databroker.actors.dds;
+
+import akka.testkit.TestProbe;
+import javax.annotation.Nonnull;
+import org.junit.Assert;
+import org.opendaylight.controller.cluster.access.ABIVersion;
+import org.opendaylight.controller.cluster.access.client.AbstractClientConnection;
+import org.opendaylight.controller.cluster.access.client.AccessClientUtil;
+import org.opendaylight.controller.cluster.access.commands.TransactionFailure;
+import org.opendaylight.controller.cluster.access.commands.TransactionRequest;
+import org.opendaylight.controller.cluster.access.concepts.AbstractRequestFailureProxy;
+import org.opendaylight.controller.cluster.access.concepts.FailureEnvelope;
+import org.opendaylight.controller.cluster.access.concepts.Request;
+import org.opendaylight.controller.cluster.access.concepts.RequestEnvelope;
+import org.opendaylight.controller.cluster.access.concepts.RequestException;
+import org.opendaylight.controller.cluster.access.concepts.RequestFailure;
+import org.opendaylight.controller.cluster.access.concepts.RequestSuccess;
+import org.opendaylight.controller.cluster.access.concepts.SuccessEnvelope;
+import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
+
+/**
+ * Helper class. Allows checking messages received by backend and respond to them.
+ */
+class TransactionTester<T extends AbstractProxyTransaction> {
+
+    private final T transaction;
+    private final AbstractClientConnection<ShardBackendInfo> connection;
+    private final TestProbe backendProbe;
+    private RequestEnvelope envelope;
+
+    TransactionTester(final T transaction,
+                      final AbstractClientConnection<ShardBackendInfo> connection,
+                      final TestProbe backendProbe) {
+        this.transaction = transaction;
+        this.connection = connection;
+        this.backendProbe = backendProbe;
+    }
+
+    T getTransaction() {
+        return transaction;
+    }
+
+    TransactionRequest getLastReceivedMessage() {
+        return (TransactionRequest) envelope.getMessage();
+    }
+
+    <R extends TransactionRequest> R expectTransactionRequest(final Class<R> expected) {
+        envelope = backendProbe.expectMsgClass(RequestEnvelope.class);
+        final Class<? extends Request> actual = envelope.getMessage().getClass();
+        final String errorMsg = String.format("Expected instance of %s, received %s", expected, actual);
+        Assert.assertTrue(errorMsg, expected.isAssignableFrom(actual));
+        return (R) envelope.getMessage();
+    }
+
+    void replySuccess(final RequestSuccess<?, ?> success) {
+        final long sessionId = envelope.getSessionId();
+        final long txSequence = envelope.getTxSequence();
+        final long executionTime = 0L;
+        final SuccessEnvelope responseEnvelope = new SuccessEnvelope(success, sessionId, txSequence, executionTime);
+        AccessClientUtil.completeRequest(connection, responseEnvelope);
+    }
+
+    void replyFailure(final RequestException cause) {
+        final long sessionId = envelope.getSessionId();
+        final long txSequence = envelope.getTxSequence();
+        final long executionTime = 0L;
+        final RequestFailure<?, ?> fail =
+                new MockFailure(transaction.getIdentifier(), envelope.getMessage().getSequence(), cause);
+        final FailureEnvelope responseEnvelope = new FailureEnvelope(fail, sessionId, txSequence, executionTime);
+        AccessClientUtil.completeRequest(connection, responseEnvelope);
+    }
+
+    private static class MockFailure extends RequestFailure<TransactionIdentifier, TransactionFailure> {
+        private MockFailure(@Nonnull final TransactionIdentifier target, final long sequence,
+                            @Nonnull final RequestException cause) {
+            super(target, sequence, cause);
+        }
+
+        @Nonnull
+        @Override
+        protected TransactionFailure cloneAsVersion(@Nonnull final ABIVersion targetVersion) {
+            throw new UnsupportedOperationException("Not implemented");
+        }
+
+        @Override
+        protected AbstractRequestFailureProxy<TransactionIdentifier, TransactionFailure> externalizableProxy(
+                @Nonnull final ABIVersion version) {
+            throw new UnsupportedOperationException("Not implemented");
+        }
+    }
+}