X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fmd-sal%2Fsal-distributed-datastore%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fcluster%2Fdatabroker%2FConcurrentDOMDataBrokerTest.java;h=346578f949df5edc9a6f41eeb56ee24d17bc20da;hb=18991f44b807ab6f06fcec76216b7f70b900b0f4;hp=ca255c82501da2a9ed4f1dfd4fc0802f253ea68e;hpb=f41c5e6e6f6e10b36b1e4b1992877e38e718c8fb;p=controller.git diff --git a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/ConcurrentDOMDataBrokerTest.java b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/ConcurrentDOMDataBrokerTest.java index ca255c8250..346578f949 100644 --- a/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/ConcurrentDOMDataBrokerTest.java +++ b/opendaylight/md-sal/sal-distributed-datastore/src/test/java/org/opendaylight/controller/cluster/databroker/ConcurrentDOMDataBrokerTest.java @@ -19,10 +19,14 @@ import static org.mockito.Mockito.mock; import static org.mockito.Mockito.never; import static org.mockito.Mockito.times; import static org.mockito.Mockito.verify; +import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateFalseFluentFuture; +import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateNullFluentFuture; +import static org.opendaylight.yangtools.util.concurrent.FluentFutures.immediateTrueFluentFuture; import com.google.common.base.Throwables; +import com.google.common.collect.ClassToInstanceMap; import com.google.common.collect.ImmutableMap; -import com.google.common.util.concurrent.CheckedFuture; +import com.google.common.util.concurrent.FluentFuture; import com.google.common.util.concurrent.FutureCallback; import com.google.common.util.concurrent.Futures; import com.google.common.util.concurrent.ListenableFuture; @@ -30,10 +34,7 @@ import com.google.common.util.concurrent.MoreExecutors; import com.google.common.util.concurrent.SettableFuture; import com.google.common.util.concurrent.Uninterruptibles; import java.util.ArrayList; -import java.util.Arrays; -import java.util.Collection; import java.util.List; -import java.util.Map; import java.util.concurrent.CountDownLatch; import java.util.concurrent.ExecutionException; import java.util.concurrent.SynchronousQueue; @@ -47,10 +48,8 @@ import org.junit.Test; import org.mockito.InOrder; import org.mockito.stubbing.Answer; import org.opendaylight.controller.cluster.datastore.DistributedDataStore; -import org.opendaylight.controller.cluster.datastore.exceptions.NoShardLeaderException; -import org.opendaylight.mdsal.common.api.DataStoreUnavailableException; +import org.opendaylight.mdsal.common.api.CommitInfo; import org.opendaylight.mdsal.common.api.LogicalDatastoreType; -import org.opendaylight.mdsal.common.api.TransactionChainListener; import org.opendaylight.mdsal.common.api.TransactionCommitFailedException; import org.opendaylight.mdsal.dom.api.DOMDataBrokerExtension; import org.opendaylight.mdsal.dom.api.DOMDataTreeChangeService; @@ -61,6 +60,8 @@ import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction; import org.opendaylight.mdsal.dom.api.DOMDataTreeReadWriteTransaction; import org.opendaylight.mdsal.dom.api.DOMDataTreeWriteTransaction; import org.opendaylight.mdsal.dom.api.DOMTransactionChain; +import org.opendaylight.mdsal.dom.api.DOMTransactionChainListener; +import org.opendaylight.mdsal.dom.broker.TransactionCommitFailedExceptionMapper; import org.opendaylight.mdsal.dom.spi.store.DOMStore; import org.opendaylight.mdsal.dom.spi.store.DOMStoreReadTransaction; import org.opendaylight.mdsal.dom.spi.store.DOMStoreReadWriteTransaction; @@ -79,8 +80,7 @@ import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode; public class ConcurrentDOMDataBrokerTest { private final DOMDataTreeWriteTransaction transaction = mock(DOMDataTreeWriteTransaction.class); - private final DOMStoreThreePhaseCommitCohort mockCohort1 = mock(DOMStoreThreePhaseCommitCohort.class); - private final DOMStoreThreePhaseCommitCohort mockCohort2 = mock(DOMStoreThreePhaseCommitCohort.class); + private final DOMStoreThreePhaseCommitCohort mockCohort = mock(DOMStoreThreePhaseCommitCohort.class); private final ThreadPoolExecutor futureExecutor = new ThreadPoolExecutor(0, 1, 5, TimeUnit.SECONDS, new SynchronousQueue<>()); private ConcurrentDOMDataBroker coordinator; @@ -116,32 +116,27 @@ public class ConcurrentDOMDataBrokerTest { final SettableFuture future = SettableFuture.create(); if (doAsync) { new Thread(() -> { - Uninterruptibles.awaitUninterruptibly(asyncCanCommitContinue, - 10, TimeUnit.SECONDS); - future.set(true); + Uninterruptibles.awaitUninterruptibly(asyncCanCommitContinue, 10, TimeUnit.SECONDS); + future.set(Boolean.TRUE); }).start(); } else { - future.set(true); + future.set(Boolean.TRUE); } return future; }; - doAnswer(asyncCanCommit).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).preCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).commit(); + doAnswer(asyncCanCommit).when(mockCohort).canCommit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).preCommit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).commit(); - doReturn(Futures.immediateFuture(true)).when(mockCohort2).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).preCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).commit(); - - ListenableFuture future = coordinator.submit(transaction, Arrays.asList(mockCohort1, mockCohort2)); + ListenableFuture future = coordinator.commit(transaction, mockCohort); final CountDownLatch doneLatch = new CountDownLatch(1); final AtomicReference caughtEx = new AtomicReference<>(); - Futures.addCallback(future, new FutureCallback() { + Futures.addCallback(future, new FutureCallback() { @Override - public void onSuccess(final Void result) { + public void onSuccess(final CommitInfo result) { doneLatch.countDown(); } @@ -154,7 +149,7 @@ public class ConcurrentDOMDataBrokerTest { asyncCanCommitContinue.countDown(); - assertEquals("Submit complete", true, doneLatch.await(5, TimeUnit.SECONDS)); + assertTrue("Submit complete", doneLatch.await(5, TimeUnit.SECONDS)); if (caughtEx.get() != null) { Throwables.throwIfUnchecked(caughtEx.get()); @@ -163,48 +158,31 @@ public class ConcurrentDOMDataBrokerTest { assertEquals("Task count", doAsync ? 1 : 0, futureExecutor.getTaskCount()); - InOrder inOrder = inOrder(mockCohort1, mockCohort2); - inOrder.verify(mockCohort1).canCommit(); - inOrder.verify(mockCohort2).canCommit(); - inOrder.verify(mockCohort1).preCommit(); - inOrder.verify(mockCohort2).preCommit(); - inOrder.verify(mockCohort1).commit(); - inOrder.verify(mockCohort2).commit(); + InOrder inOrder = inOrder(mockCohort); + inOrder.verify(mockCohort, times(1)).canCommit(); + inOrder.verify(mockCohort, times(1)).preCommit(); + inOrder.verify(mockCohort, times(1)).commit(); } @Test public void testSubmitWithNegativeCanCommitResponse() throws Exception { - doReturn(Futures.immediateFuture(true)).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).abort(); - - doReturn(Futures.immediateFuture(false)).when(mockCohort2).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).abort(); - - DOMStoreThreePhaseCommitCohort mockCohort3 = mock(DOMStoreThreePhaseCommitCohort.class); - doReturn(Futures.immediateFuture(false)).when(mockCohort3).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort3).abort(); - - CheckedFuture future = coordinator.submit( - transaction, Arrays.asList(mockCohort1, mockCohort2, mockCohort3)); + doReturn(Futures.immediateFuture(Boolean.FALSE)).when(mockCohort).canCommit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).abort(); - assertFailure(future, null, mockCohort1, mockCohort2, mockCohort3); + assertFailure(coordinator.commit(transaction, mockCohort), null, mockCohort); } - private static void assertFailure(final CheckedFuture future, - final Exception expCause, final DOMStoreThreePhaseCommitCohort... mockCohorts) - throws Exception { + private static void assertFailure(final ListenableFuture future, final Exception expCause, + final DOMStoreThreePhaseCommitCohort mockCohort) throws Exception { try { - future.checkedGet(5, TimeUnit.SECONDS); + future.get(5, TimeUnit.SECONDS); fail("Expected TransactionCommitFailedException"); - } catch (TransactionCommitFailedException e) { + } catch (ExecutionException e) { + TransactionCommitFailedException tcf = TransactionCommitFailedExceptionMapper.COMMIT_ERROR_MAPPER.apply(e); if (expCause != null) { - assertSame("Expected cause", expCause.getClass(), e.getCause().getClass()); - } - - InOrder inOrder = inOrder((Object[])mockCohorts); - for (DOMStoreThreePhaseCommitCohort c: mockCohorts) { - inOrder.verify(c).abort(); + assertSame("Expected cause", expCause.getClass(), tcf.getCause().getClass()); } + verify(mockCohort, times(1)).abort(); } catch (TimeoutException e) { throw e; } @@ -212,97 +190,42 @@ public class ConcurrentDOMDataBrokerTest { @Test public void testSubmitWithCanCommitException() throws Exception { - doReturn(Futures.immediateFuture(true)).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).abort(); - - IllegalStateException cause = new IllegalStateException("mock"); - doReturn(Futures.immediateFailedFuture(cause)).when(mockCohort2).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).abort(); + final Exception cause = new IllegalStateException("mock"); + doReturn(Futures.immediateFailedFuture(cause)).when(mockCohort).canCommit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).abort(); - CheckedFuture future = coordinator.submit( - transaction, Arrays.asList(mockCohort1, mockCohort2)); - - assertFailure(future, cause, mockCohort1, mockCohort2); - } - - @Test - public void testSubmitWithCanCommitDataStoreUnavailableException() throws Exception { - doReturn(Futures.immediateFuture(true)).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).abort(); - NoShardLeaderException rootCause = new NoShardLeaderException("mock"); - DataStoreUnavailableException cause = new DataStoreUnavailableException(rootCause.getMessage(), rootCause); - doReturn(Futures.immediateFailedFuture(rootCause)).when(mockCohort2).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).abort(); - - CheckedFuture future = coordinator.submit( - transaction, Arrays.asList(mockCohort1, mockCohort2)); - - assertFailure(future, cause, mockCohort1, mockCohort2); + assertFailure(coordinator.commit(transaction, mockCohort), cause, mockCohort); } @Test public void testSubmitWithPreCommitException() throws Exception { - doReturn(Futures.immediateFuture(true)).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).preCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).abort(); - - doReturn(Futures.immediateFuture(true)).when(mockCohort2).canCommit(); - IllegalStateException cause = new IllegalStateException("mock"); - doReturn(Futures.immediateFailedFuture(cause)).when(mockCohort2).preCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).abort(); - - DOMStoreThreePhaseCommitCohort mockCohort3 = mock(DOMStoreThreePhaseCommitCohort.class); - doReturn(Futures.immediateFuture(true)).when(mockCohort3).canCommit(); - doReturn(Futures.immediateFailedFuture(new IllegalStateException("mock2"))) - .when(mockCohort3).preCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort3).abort(); - - CheckedFuture future = coordinator.submit( - transaction, Arrays.asList(mockCohort1, mockCohort2, mockCohort3)); + doReturn(immediateTrueFluentFuture()).when(mockCohort).canCommit(); + final IllegalStateException cause = new IllegalStateException("mock"); + doReturn(Futures.immediateFailedFuture(cause)).when(mockCohort).preCommit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).abort(); - assertFailure(future, cause, mockCohort1, mockCohort2, mockCohort3); + assertFailure(coordinator.commit(transaction, mockCohort), cause, mockCohort); } @Test public void testSubmitWithCommitException() throws Exception { - doReturn(Futures.immediateFuture(true)).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).preCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).commit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort1).abort(); - - doReturn(Futures.immediateFuture(true)).when(mockCohort2).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).preCommit(); - IllegalStateException cause = new IllegalStateException("mock"); - doReturn(Futures.immediateFailedFuture(cause)).when(mockCohort2).commit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).abort(); - - DOMStoreThreePhaseCommitCohort mockCohort3 = mock(DOMStoreThreePhaseCommitCohort.class); - doReturn(Futures.immediateFuture(true)).when(mockCohort3).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort3).preCommit(); - doReturn(Futures.immediateFailedFuture(new IllegalStateException("mock2"))) - .when(mockCohort3).commit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort3).abort(); - - CheckedFuture future = coordinator.submit( - transaction, Arrays.asList(mockCohort1, mockCohort2, mockCohort3)); - - assertFailure(future, cause, mockCohort1, mockCohort2, mockCohort3); + doReturn(immediateTrueFluentFuture()).when(mockCohort).canCommit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).preCommit(); + final IllegalStateException cause = new IllegalStateException("mock"); + doReturn(Futures.immediateFailedFuture(cause)).when(mockCohort).commit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).abort(); + + assertFailure(coordinator.commit(transaction, mockCohort), cause, mockCohort); } @Test public void testSubmitWithAbortException() throws Exception { - doReturn(Futures.immediateFuture(true)).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFailedFuture(new IllegalStateException("mock abort error"))) - .when(mockCohort1).abort(); - - IllegalStateException cause = new IllegalStateException("mock canCommit error"); - doReturn(Futures.immediateFailedFuture(cause)).when(mockCohort2).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort2).abort(); + final Exception canCommitCause = new IllegalStateException("canCommit error"); + doReturn(Futures.immediateFailedFuture(canCommitCause)).when(mockCohort).canCommit(); + final Exception abortCause = new IllegalStateException("abort error"); + doReturn(Futures.immediateFailedFuture(abortCause)).when(mockCohort).abort(); - CheckedFuture future = coordinator.submit( - transaction, Arrays.asList(mockCohort1, mockCohort2)); - - assertFailure(future, cause, mockCohort1, mockCohort2); + assertFailure(coordinator.commit(transaction, mockCohort), canCommitCause, mockCohort); } @Test @@ -355,17 +278,12 @@ public class ConcurrentDOMDataBrokerTest { configDomStore), futureExecutor)) { DOMDataTreeReadWriteTransaction dataTxn = dataBroker.newReadWriteTransaction(); - dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); - dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); - dataTxn.read(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY); + dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty(), mock(NormalizedNode.class)); + dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty(), mock(NormalizedNode.class)); + dataTxn.read(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty()); verify(configDomStore, never()).newReadWriteTransaction(); verify(operationalDomStore, times(1)).newReadWriteTransaction(); - - dataTxn.put(LogicalDatastoreType.CONFIGURATION, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); - - verify(configDomStore, times(1)).newReadWriteTransaction(); - verify(operationalDomStore, times(1)).newReadWriteTransaction(); } } @@ -384,16 +302,11 @@ public class ConcurrentDOMDataBrokerTest { configDomStore), futureExecutor)) { DOMDataTreeWriteTransaction dataTxn = dataBroker.newWriteOnlyTransaction(); - dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); - dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); + dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty(), mock(NormalizedNode.class)); + dataTxn.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty(), mock(NormalizedNode.class)); verify(configDomStore, never()).newWriteOnlyTransaction(); verify(operationalDomStore, times(1)).newWriteOnlyTransaction(); - - dataTxn.put(LogicalDatastoreType.CONFIGURATION, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); - - verify(configDomStore, times(1)).newWriteOnlyTransaction(); - verify(operationalDomStore, times(1)).newWriteOnlyTransaction(); } } @@ -411,16 +324,11 @@ public class ConcurrentDOMDataBrokerTest { configDomStore), futureExecutor)) { DOMDataTreeReadTransaction dataTxn = dataBroker.newReadOnlyTransaction(); - dataTxn.read(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY); - dataTxn.read(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY); + dataTxn.read(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty()); + dataTxn.read(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty()); verify(configDomStore, never()).newReadOnlyTransaction(); verify(operationalDomStore, times(1)).newReadOnlyTransaction(); - - dataTxn.read(LogicalDatastoreType.CONFIGURATION, YangInstanceIdentifier.EMPTY); - - verify(configDomStore, times(1)).newReadOnlyTransaction(); - verify(operationalDomStore, times(1)).newReadOnlyTransaction(); } } @@ -429,12 +337,11 @@ public class ConcurrentDOMDataBrokerTest { DOMStore configDomStore = mock(DOMStore.class); DOMStore operationalDomStore = mock(DOMStore.class); DOMStoreReadWriteTransaction mockStoreReadWriteTransaction = mock(DOMStoreReadWriteTransaction.class); - DOMStoreThreePhaseCommitCohort mockCohort = mock(DOMStoreThreePhaseCommitCohort.class); doReturn(mockStoreReadWriteTransaction).when(operationalDomStore).newReadWriteTransaction(); doReturn(mockCohort).when(mockStoreReadWriteTransaction).ready(); - doReturn(Futures.immediateFuture(false)).when(mockCohort).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohort).abort(); + doReturn(immediateFalseFluentFuture()).when(mockCohort).canCommit(); + doReturn(immediateNullFluentFuture()).when(mockCohort).abort(); final CountDownLatch latch = new CountDownLatch(1); final List commitCohorts = new ArrayList<>(); @@ -443,18 +350,18 @@ public class ConcurrentDOMDataBrokerTest { LogicalDatastoreType.OPERATIONAL, operationalDomStore, LogicalDatastoreType.CONFIGURATION, configDomStore), futureExecutor) { @Override - public CheckedFuture submit(DOMDataTreeWriteTransaction writeTx, - Collection cohorts) { - commitCohorts.addAll(cohorts); + public FluentFuture commit(DOMDataTreeWriteTransaction writeTx, + DOMStoreThreePhaseCommitCohort cohort) { + commitCohorts.add(cohort); latch.countDown(); - return super.submit(writeTx, cohorts); + return super.commit(writeTx, cohort); } }) { DOMDataTreeReadWriteTransaction domDataReadWriteTransaction = dataBroker.newReadWriteTransaction(); - domDataReadWriteTransaction.delete(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY); + domDataReadWriteTransaction.delete(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty()); - domDataReadWriteTransaction.submit(); + domDataReadWriteTransaction.commit(); assertTrue(latch.await(10, TimeUnit.SECONDS)); @@ -462,56 +369,6 @@ public class ConcurrentDOMDataBrokerTest { } } - @Test - public void testSubmitWithOnlyTwoSubTransactions() throws InterruptedException { - DOMStore configDomStore = mock(DOMStore.class); - DOMStore operationalDomStore = mock(DOMStore.class); - DOMStoreReadWriteTransaction operationalTransaction = mock(DOMStoreReadWriteTransaction.class); - DOMStoreReadWriteTransaction configTransaction = mock(DOMStoreReadWriteTransaction.class); - DOMStoreThreePhaseCommitCohort mockCohortOperational = mock(DOMStoreThreePhaseCommitCohort.class); - DOMStoreThreePhaseCommitCohort mockCohortConfig = mock(DOMStoreThreePhaseCommitCohort.class); - - doReturn(operationalTransaction).when(operationalDomStore).newReadWriteTransaction(); - doReturn(configTransaction).when(configDomStore).newReadWriteTransaction(); - - doReturn(mockCohortOperational).when(operationalTransaction).ready(); - doReturn(Futures.immediateFuture(false)).when(mockCohortOperational).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohortOperational).abort(); - - doReturn(mockCohortConfig).when(configTransaction).ready(); - doReturn(Futures.immediateFuture(false)).when(mockCohortConfig).canCommit(); - doReturn(Futures.immediateFuture(null)).when(mockCohortConfig).abort(); - - final CountDownLatch latch = new CountDownLatch(1); - final List commitCohorts = new ArrayList<>(); - - try (ConcurrentDOMDataBroker dataBroker = new ConcurrentDOMDataBroker(ImmutableMap.of( - LogicalDatastoreType.OPERATIONAL, operationalDomStore, LogicalDatastoreType.CONFIGURATION, - configDomStore), futureExecutor) { - @Override - @SuppressWarnings("checkstyle:hiddenField") - public CheckedFuture submit(DOMDataTreeWriteTransaction writeTx, - Collection cohorts) { - commitCohorts.addAll(cohorts); - latch.countDown(); - return super.submit(writeTx, cohorts); - } - }) { - DOMDataTreeReadWriteTransaction domDataReadWriteTransaction = dataBroker.newReadWriteTransaction(); - - domDataReadWriteTransaction.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, - mock(NormalizedNode.class)); - domDataReadWriteTransaction.merge(LogicalDatastoreType.CONFIGURATION, YangInstanceIdentifier.EMPTY, - mock(NormalizedNode.class)); - - domDataReadWriteTransaction.submit(); - - assertTrue(latch.await(10, TimeUnit.SECONDS)); - - assertTrue(commitCohorts.size() == 2); - } - } - @Test public void testCreateTransactionChain() { DOMStore domStore = mock(DOMStore.class); @@ -519,7 +376,7 @@ public class ConcurrentDOMDataBrokerTest { LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), futureExecutor)) { - dataBroker.createTransactionChain(mock(TransactionChainListener.class)); + dataBroker.createTransactionChain(mock(DOMTransactionChainListener.class)); verify(domStore, times(2)).createTransactionChain(); } @@ -540,13 +397,13 @@ public class ConcurrentDOMDataBrokerTest { doReturn(operationalTransaction).when(mockChain).newWriteOnlyTransaction(); DOMTransactionChain transactionChain = dataBroker.createTransactionChain( - mock(TransactionChainListener.class)); + mock(DOMTransactionChainListener.class)); DOMDataTreeWriteTransaction domDataWriteTransaction = transactionChain.newWriteOnlyTransaction(); verify(mockChain, never()).newWriteOnlyTransaction(); - domDataWriteTransaction.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, + domDataWriteTransaction.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.empty(), mock(NormalizedNode.class)); } } @@ -558,15 +415,13 @@ public class ConcurrentDOMDataBrokerTest { LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), futureExecutor)) { - CheckedFuture submit1 = - dataBroker.newWriteOnlyTransaction().submit(); + FluentFuture submit1 = dataBroker.newWriteOnlyTransaction().commit(); assertNotNull(submit1); submit1.get(); - CheckedFuture submit2 = - dataBroker.newReadWriteTransaction().submit(); + FluentFuture submit2 = dataBroker.newReadWriteTransaction().commit(); assertNotNull(submit2); @@ -582,21 +437,20 @@ public class ConcurrentDOMDataBrokerTest { LogicalDatastoreType.OPERATIONAL, mockOperStore, LogicalDatastoreType.CONFIGURATION, mockConfigStore), futureExecutor)) { - Map, DOMDataBrokerExtension> supportedExtensions = - dataBroker.getSupportedExtensions(); - assertNotNull(supportedExtensions.get(DOMDataTreeChangeService.class)); + ClassToInstanceMap supportedExtensions = dataBroker.getExtensions(); + assertNotNull(supportedExtensions.getInstance(DOMDataTreeChangeService.class)); - DOMDataTreeCommitCohortRegistry cohortRegistry = - (DOMDataTreeCommitCohortRegistry) supportedExtensions.get(DOMDataTreeCommitCohortRegistry.class); + DOMDataTreeCommitCohortRegistry cohortRegistry = supportedExtensions.getInstance( + DOMDataTreeCommitCohortRegistry.class); assertNotNull(cohortRegistry); - DOMDataTreeCommitCohort mockCohort = mock(DOMDataTreeCommitCohort.class); + DOMDataTreeCommitCohort cohort = mock(DOMDataTreeCommitCohort.class); DOMDataTreeIdentifier path = new DOMDataTreeIdentifier( org.opendaylight.mdsal.common.api.LogicalDatastoreType.CONFIGURATION, - YangInstanceIdentifier.EMPTY); - cohortRegistry.registerCommitCohort(path, mockCohort); + YangInstanceIdentifier.empty()); + cohortRegistry.registerCommitCohort(path, cohort); - verify(mockConfigStore).registerCommitCohort(path, mockCohort); + verify(mockConfigStore).registerCommitCohort(path, cohort); } } }