opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/actors/dds/RemoteProxyTransactionTest.java

   1 /*
   2  * Copyright (c) 2017 Pantheon Technologies s.r.o. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.cluster.databroker.actors.dds;
   9
  10 import static org.hamcrest.CoreMatchers.both;
  11 import static org.hamcrest.CoreMatchers.hasItem;
  12 import static org.hamcrest.CoreMatchers.isA;
  13 import static org.hamcrest.MatcherAssert.assertThat;
  14 import static org.junit.Assert.assertEquals;
  15 import static org.opendaylight.controller.cluster.databroker.actors.dds.TestUtils.assertFutureEquals;
  16
  17 import akka.testkit.TestProbe;
  18 import com.google.common.util.concurrent.FluentFuture;
  19 import com.google.common.util.concurrent.ListenableFuture;
  20 import java.util.List;
  21 import java.util.Optional;
  22 import org.junit.Test;
  23 import org.opendaylight.controller.cluster.access.commands.ExistsTransactionRequest;
  24 import org.opendaylight.controller.cluster.access.commands.ExistsTransactionSuccess;
  25 import org.opendaylight.controller.cluster.access.commands.ModifyTransactionRequest;
  26 import org.opendaylight.controller.cluster.access.commands.ModifyTransactionRequestBuilder;
  27 import org.opendaylight.controller.cluster.access.commands.PersistenceProtocol;
  28 import org.opendaylight.controller.cluster.access.commands.ReadTransactionRequest;
  29 import org.opendaylight.controller.cluster.access.commands.ReadTransactionSuccess;
  30 import org.opendaylight.controller.cluster.access.commands.TransactionAbortRequest;
  31 import org.opendaylight.controller.cluster.access.commands.TransactionCanCommitSuccess;
  32 import org.opendaylight.controller.cluster.access.commands.TransactionCommitSuccess;
  33 import org.opendaylight.controller.cluster.access.commands.TransactionDelete;
  34 import org.opendaylight.controller.cluster.access.commands.TransactionDoCommitRequest;
  35 import org.opendaylight.controller.cluster.access.commands.TransactionMerge;
  36 import org.opendaylight.controller.cluster.access.commands.TransactionModification;
  37 import org.opendaylight.controller.cluster.access.commands.TransactionPreCommitRequest;
  38 import org.opendaylight.controller.cluster.access.commands.TransactionPreCommitSuccess;
  39 import org.opendaylight.controller.cluster.access.commands.TransactionWrite;
  40 import org.opendaylight.controller.cluster.access.concepts.TransactionIdentifier;
  41 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  42 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  43 import org.opendaylight.yangtools.yang.data.tree.api.DataTreeSnapshot;
  44
  45 public class RemoteProxyTransactionTest extends AbstractProxyTransactionTest<RemoteProxyTransaction> {
  46     @Override
  47     protected RemoteProxyTransaction createTransaction(final ProxyHistory parent, final TransactionIdentifier id,
  48                                                        final DataTreeSnapshot snapshot) {
  49         mockForRemote();
  50         return new RemoteProxyTransaction(parent, TRANSACTION_ID, false, false, false);
  51     }
  52
  53     @Override
  54     @Test
  55     public void testExists() throws Exception {
  56         final TransactionTester<RemoteProxyTransaction> tester = getTester();
  57         final FluentFuture<Boolean> exists = transaction.exists(PATH_1);
  58         final ExistsTransactionRequest req = tester.expectTransactionRequest(ExistsTransactionRequest.class);
  59         final boolean existsResult = true;
  60         tester.replySuccess(new ExistsTransactionSuccess(TRANSACTION_ID, req.getSequence(), existsResult));
  61         assertFutureEquals(existsResult, exists);
  62     }
  63
  64     @Override
  65     @Test
  66     public void testRead() throws Exception {
  67         final TransactionTester<RemoteProxyTransaction> tester = getTester();
  68         final FluentFuture<Optional<NormalizedNode>> read = transaction.read(PATH_2);
  69         final ReadTransactionRequest req = tester.expectTransactionRequest(ReadTransactionRequest.class);
  70         final Optional<NormalizedNode> result = Optional.of(DATA_1);
  71         tester.replySuccess(new ReadTransactionSuccess(TRANSACTION_ID, req.getSequence(), result));
  72         assertFutureEquals(result, read);
  73     }
  74
  75     @Override
  76     @Test
  77     public void testWrite() {
  78         final YangInstanceIdentifier path = PATH_1;
  79         testModification(() -> transaction.write(path, DATA_1), TransactionWrite.class, path);
  80     }
  81
  82     @Override
  83     @Test
  84     public void testMerge() {
  85         final YangInstanceIdentifier path = PATH_2;
  86         testModification(() -> transaction.merge(path, DATA_2), TransactionMerge.class, path);
  87     }
  88
  89     @Override
  90     @Test
  91     public void testDelete() {
  92         final YangInstanceIdentifier path = PATH_3;
  93         testModification(() -> transaction.delete(path), TransactionDelete.class, path);
  94     }
  95
  96     @Override
  97     @Test
  98     public void testDirectCommit() throws Exception {
  99         transaction.seal();
 100         final ListenableFuture<Boolean> result = transaction.directCommit();
 101         final TransactionTester<RemoteProxyTransaction> tester = getTester();
 102         final ModifyTransactionRequest req = tester.expectTransactionRequest(ModifyTransactionRequest.class);
 103         assertEquals(Optional.of(PersistenceProtocol.SIMPLE), req.getPersistenceProtocol());
 104         tester.replySuccess(new TransactionCommitSuccess(TRANSACTION_ID, req.getSequence()));
 105         assertFutureEquals(true, result);
 106     }
 107
 108     @Override
 109     @Test
 110     public void testCanCommit() {
 111         testRequestResponse(transaction::canCommit, ModifyTransactionRequest.class,
 112                 TransactionCanCommitSuccess::new);
 113     }
 114
 115     @Override
 116     @Test
 117     public void testPreCommit() {
 118         testRequestResponse(transaction::preCommit, TransactionPreCommitRequest.class,
 119                 TransactionPreCommitSuccess::new);
 120     }
 121
 122     @Override
 123     @Test
 124     public void testDoCommit() {
 125         testRequestResponse(transaction::doCommit, TransactionDoCommitRequest.class, TransactionCommitSuccess::new);
 126     }
 127
 128     @Override
 129     @Test
 130     public void testForwardToRemoteAbort() {
 131         final TestProbe probe = createProbe();
 132         final TransactionAbortRequest request = new TransactionAbortRequest(TRANSACTION_ID, 0L, probe.ref());
 133         testForwardToRemote(request, TransactionAbortRequest.class);
 134
 135     }
 136
 137     @Override
 138     public void testForwardToRemoteCommit() {
 139         final TestProbe probe = createProbe();
 140         final TransactionAbortRequest request = new TransactionAbortRequest(TRANSACTION_ID, 0L, probe.ref());
 141         testForwardToRemote(request, TransactionAbortRequest.class);
 142     }
 143
 144     @Test
 145     public void testForwardToRemoteModifyCommitSimple() {
 146         final TestProbe probe = createProbe();
 147         final ModifyTransactionRequestBuilder builder =
 148                 new ModifyTransactionRequestBuilder(TRANSACTION_ID, probe.ref());
 149         builder.setSequence(0L);
 150         builder.setCommit(false);
 151         final ModifyTransactionRequest request = builder.build();
 152         final ModifyTransactionRequest received = testForwardToRemote(request, ModifyTransactionRequest.class);
 153         assertEquals(request.getPersistenceProtocol(), received.getPersistenceProtocol());
 154         assertEquals(request.getModifications(), received.getModifications());
 155         assertEquals(request.getTarget(), received.getTarget());
 156     }
 157
 158     @Test
 159     public void testForwardToRemoteModifyCommit3Phase() {
 160         final TestProbe probe = createProbe();
 161         final ModifyTransactionRequestBuilder builder =
 162                 new ModifyTransactionRequestBuilder(TRANSACTION_ID, probe.ref());
 163         builder.setSequence(0L);
 164         builder.setCommit(true);
 165         final ModifyTransactionRequest request = builder.build();
 166         final ModifyTransactionRequest received = testForwardToRemote(request, ModifyTransactionRequest.class);
 167         assertEquals(request.getPersistenceProtocol(), received.getPersistenceProtocol());
 168         assertEquals(request.getModifications(), received.getModifications());
 169         assertEquals(request.getTarget(), received.getTarget());
 170     }
 171
 172     @Test
 173     public void testForwardToRemoteModifyAbort() {
 174         final TestProbe probe = createProbe();
 175         final ModifyTransactionRequestBuilder builder =
 176                 new ModifyTransactionRequestBuilder(TRANSACTION_ID, probe.ref());
 177         builder.setSequence(0L);
 178         builder.setAbort();
 179         final ModifyTransactionRequest request = builder.build();
 180         final ModifyTransactionRequest received = testForwardToRemote(request, ModifyTransactionRequest.class);
 181         assertEquals(request.getTarget(), received.getTarget());
 182         assertEquals(Optional.of(PersistenceProtocol.ABORT), received.getPersistenceProtocol());
 183     }
 184
 185     @Test
 186     public void testForwardToRemoteModifyRead() {
 187         final TestProbe probe = createProbe();
 188         final ReadTransactionRequest request =
 189                 new ReadTransactionRequest(TRANSACTION_ID, 0L, probe.ref(), PATH_1, false);
 190         final ReadTransactionRequest received = testForwardToRemote(request, ReadTransactionRequest.class);
 191         assertEquals(request.getTarget(), received.getTarget());
 192         assertEquals(request.getPath(), received.getPath());
 193     }
 194
 195     @Test
 196     public void testForwardToRemoteModifyExists() {
 197         final TestProbe probe = createProbe();
 198         final ExistsTransactionRequest request =
 199                 new ExistsTransactionRequest(TRANSACTION_ID, 0L, probe.ref(), PATH_1, false);
 200         final ExistsTransactionRequest received = testForwardToRemote(request, ExistsTransactionRequest.class);
 201         assertEquals(request.getTarget(), received.getTarget());
 202         assertEquals(request.getPath(), received.getPath());
 203     }
 204
 205     @Test
 206     public void testForwardToRemoteModifyPreCommit() {
 207         final TestProbe probe = createProbe();
 208         final TransactionPreCommitRequest request =
 209                 new TransactionPreCommitRequest(TRANSACTION_ID, 0L, probe.ref());
 210         final TransactionPreCommitRequest received = testForwardToRemote(request, TransactionPreCommitRequest.class);
 211         assertEquals(request.getTarget(), received.getTarget());
 212     }
 213
 214     @Test
 215     public void testForwardToRemoteModifyDoCommit() {
 216         final TestProbe probe = createProbe();
 217         final TransactionDoCommitRequest request =
 218                 new TransactionDoCommitRequest(TRANSACTION_ID, 0L, probe.ref());
 219         final TransactionDoCommitRequest received = testForwardToRemote(request, TransactionDoCommitRequest.class);
 220         assertEquals(request.getTarget(), received.getTarget());
 221     }
 222
 223
 224     private <T extends TransactionModification> void testModification(final Runnable modification, final Class<T> cls,
 225             final YangInstanceIdentifier expectedPath) {
 226         modification.run();
 227         final ModifyTransactionRequest request = transaction.commitRequest(false);
 228         final List<TransactionModification> modifications = request.getModifications();
 229         assertEquals(1, modifications.size());
 230         assertThat(modifications, hasItem(both(isA(cls)).and(hasPath(expectedPath))));
 231     }
 232 }