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=bef18ffe200216502fb7fc42fe10de657b6f7799;hb=28551609a31799a43d3017ba0681e198f5136d70;hp=8686cc49815f9773f9f72e0f2c363ac39d95c1e9;hpb=dea3effede98cfb561c44d66b24c2d71a44b10a3;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 8686cc4981..bef18ffe20 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,6 +19,8 @@ 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 com.google.common.base.Throwables; import com.google.common.collect.ImmutableMap; import com.google.common.util.concurrent.CheckedFuture; import com.google.common.util.concurrent.FutureCallback; @@ -31,6 +33,7 @@ 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; @@ -42,15 +45,18 @@ import org.junit.After; import org.junit.Before; import org.junit.Test; import org.mockito.InOrder; -import org.mockito.invocation.InvocationOnMock; import org.mockito.stubbing.Answer; +import org.opendaylight.controller.cluster.datastore.DistributedDataStore; import org.opendaylight.controller.cluster.datastore.exceptions.NoShardLeaderException; import org.opendaylight.controller.md.sal.common.api.data.DataStoreUnavailableException; import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType; import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener; import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException; +import org.opendaylight.controller.md.sal.dom.api.DOMDataBrokerExtension; import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction; import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction; +import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeChangeService; +import org.opendaylight.controller.md.sal.dom.api.DOMDataTreeCommitCohortRegistry; import org.opendaylight.controller.md.sal.dom.api.DOMDataWriteTransaction; import org.opendaylight.controller.md.sal.dom.api.DOMTransactionChain; import org.opendaylight.controller.md.sal.dom.store.impl.InMemoryDOMDataStore; @@ -60,6 +66,8 @@ import org.opendaylight.controller.sal.core.spi.data.DOMStoreReadWriteTransactio import org.opendaylight.controller.sal.core.spi.data.DOMStoreThreePhaseCommitCohort; import org.opendaylight.controller.sal.core.spi.data.DOMStoreTransactionChain; import org.opendaylight.controller.sal.core.spi.data.DOMStoreWriteTransaction; +import org.opendaylight.mdsal.dom.api.DOMDataTreeCommitCohort; +import org.opendaylight.mdsal.dom.api.DOMDataTreeIdentifier; import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier; import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode; @@ -84,7 +92,8 @@ public class ConcurrentDOMDataBrokerTest { DOMStore store = new InMemoryDOMDataStore("OPER", MoreExecutors.newDirectExecutorService()); - coordinator = new ConcurrentDOMDataBroker(ImmutableMap.of(LogicalDatastoreType.OPERATIONAL, store), futureExecutor); + coordinator = new ConcurrentDOMDataBroker(ImmutableMap.of(LogicalDatastoreType.OPERATIONAL, store), + futureExecutor); } @After @@ -93,36 +102,30 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testSuccessfulSubmitAsync() throws Throwable { + public void testSuccessfulSubmitAsync() throws Exception { testSuccessfulSubmit(true); } @Test - public void testSuccessfulSubmitSync() throws Throwable { + public void testSuccessfulSubmitSync() throws Exception { testSuccessfulSubmit(false); } - private void testSuccessfulSubmit(final boolean doAsync) throws Throwable { + private void testSuccessfulSubmit(final boolean doAsync) throws InterruptedException { final CountDownLatch asyncCanCommitContinue = new CountDownLatch(1); - Answer> asyncCanCommit = new Answer>() { - @Override - public ListenableFuture answer(final InvocationOnMock invocation) { - final SettableFuture future = SettableFuture.create(); - if(doAsync) { - new Thread() { - @Override - public void run() { - Uninterruptibles.awaitUninterruptibly(asyncCanCommitContinue, - 10, TimeUnit.SECONDS); - future.set(true); - } - }.start(); - } else { + Answer> asyncCanCommit = invocation -> { + final SettableFuture future = SettableFuture.create(); + if (doAsync) { + new Thread(() -> { + Uninterruptibles.awaitUninterruptibly(asyncCanCommitContinue, + 10, TimeUnit.SECONDS); future.set(true); - } - - return future; + }).start(); + } else { + future.set(true); } + + return future; }; doAnswer(asyncCanCommit).when(mockCohort1).canCommit(); @@ -145,8 +148,8 @@ public class ConcurrentDOMDataBrokerTest { } @Override - public void onFailure(final Throwable t) { - caughtEx.set(t); + public void onFailure(final Throwable failure) { + caughtEx.set(failure); doneLatch.countDown(); } }); @@ -155,8 +158,8 @@ public class ConcurrentDOMDataBrokerTest { assertEquals("Submit complete", true, doneLatch.await(5, TimeUnit.SECONDS)); - if(caughtEx.get() != null) { - throw caughtEx.get(); + if (caughtEx.get() != null) { + Throwables.propagate(caughtEx.get()); } assertEquals("Task count", doAsync ? 1 : 0, futureExecutor.getTaskCount()); @@ -195,12 +198,12 @@ public class ConcurrentDOMDataBrokerTest { future.checkedGet(5, TimeUnit.SECONDS); fail("Expected TransactionCommitFailedException"); } catch (TransactionCommitFailedException e) { - if(expCause != null) { + if (expCause != null) { assertSame("Expected cause", expCause.getClass(), e.getCause().getClass()); } InOrder inOrder = inOrder((Object[])mockCohorts); - for(DOMStoreThreePhaseCommitCohort c: mockCohorts) { + for (DOMStoreThreePhaseCommitCohort c: mockCohorts) { inOrder.verify(c).abort(); } } catch (TimeoutException e) { @@ -251,8 +254,8 @@ public class ConcurrentDOMDataBrokerTest { DOMStoreThreePhaseCommitCohort mockCohort3 = mock(DOMStoreThreePhaseCommitCohort.class); doReturn(Futures.immediateFuture(true)).when(mockCohort3).canCommit(); - doReturn(Futures.immediateFailedFuture(new IllegalStateException("mock2"))). - when(mockCohort3).preCommit(); + doReturn(Futures.immediateFailedFuture(new IllegalStateException("mock2"))) + .when(mockCohort3).preCommit(); doReturn(Futures.immediateFuture(null)).when(mockCohort3).abort(); CheckedFuture future = coordinator.submit( @@ -277,8 +280,8 @@ public class ConcurrentDOMDataBrokerTest { 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.immediateFailedFuture(new IllegalStateException("mock2"))) + .when(mockCohort3).commit(); doReturn(Futures.immediateFuture(null)).when(mockCohort3).abort(); CheckedFuture future = coordinator.submit( @@ -290,8 +293,8 @@ public class ConcurrentDOMDataBrokerTest { @Test public void testSubmitWithAbortException() throws Exception { doReturn(Futures.immediateFuture(true)).when(mockCohort1).canCommit(); - doReturn(Futures.immediateFailedFuture(new IllegalStateException("mock abort error"))). - when(mockCohort1).abort(); + 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(); @@ -304,7 +307,7 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testCreateReadWriteTransaction(){ + public void testCreateReadWriteTransaction() { DOMStore domStore = mock(DOMStore.class); try (ConcurrentDOMDataBroker dataBroker = new ConcurrentDOMDataBroker(ImmutableMap.of( LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), @@ -316,7 +319,7 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testCreateWriteOnlyTransaction(){ + public void testCreateWriteOnlyTransaction() { DOMStore domStore = mock(DOMStore.class); try (ConcurrentDOMDataBroker dataBroker = new ConcurrentDOMDataBroker(ImmutableMap.of( LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), @@ -328,7 +331,7 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testCreateReadOnlyTransaction(){ + public void testCreateReadOnlyTransaction() { DOMStore domStore = mock(DOMStore.class); try (ConcurrentDOMDataBroker dataBroker = new ConcurrentDOMDataBroker(ImmutableMap.of( LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), @@ -340,7 +343,7 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testLazySubTransactionCreationForReadWriteTransactions(){ + public void testLazySubTransactionCreationForReadWriteTransactions() { DOMStore configDomStore = mock(DOMStore.class); DOMStore operationalDomStore = mock(DOMStore.class); DOMStoreReadWriteTransaction storeTxn = mock(DOMStoreReadWriteTransaction.class); @@ -369,7 +372,7 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testLazySubTransactionCreationForWriteOnlyTransactions(){ + public void testLazySubTransactionCreationForWriteOnlyTransactions() { DOMStore configDomStore = mock(DOMStore.class); DOMStore operationalDomStore = mock(DOMStore.class); DOMStoreWriteTransaction storeTxn = mock(DOMStoreWriteTransaction.class); @@ -396,7 +399,7 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testLazySubTransactionCreationForReadOnlyTransactions(){ + public void testLazySubTransactionCreationForReadOnlyTransactions() { DOMStore configDomStore = mock(DOMStore.class); DOMStore operationalDomStore = mock(DOMStore.class); DOMStoreReadTransaction storeTxn = mock(DOMStoreReadTransaction.class); @@ -441,10 +444,11 @@ public class ConcurrentDOMDataBrokerTest { LogicalDatastoreType.OPERATIONAL, operationalDomStore, LogicalDatastoreType.CONFIGURATION, configDomStore), futureExecutor) { @Override - public CheckedFuture submit(DOMDataWriteTransaction transaction, Collection cohorts) { + public CheckedFuture submit(DOMDataWriteTransaction writeTx, + Collection cohorts) { commitCohorts.addAll(cohorts); latch.countDown(); - return super.submit(transaction, cohorts); + return super.submit(writeTx, cohorts); } }) { DOMDataReadWriteTransaction domDataReadWriteTransaction = dataBroker.newReadWriteTransaction(); @@ -453,7 +457,7 @@ public class ConcurrentDOMDataBrokerTest { domDataReadWriteTransaction.submit(); - latch.await(10, TimeUnit.SECONDS); + assertTrue(latch.await(10, TimeUnit.SECONDS)); assertTrue(commitCohorts.size() == 1); } @@ -486,7 +490,8 @@ public class ConcurrentDOMDataBrokerTest { LogicalDatastoreType.OPERATIONAL, operationalDomStore, LogicalDatastoreType.CONFIGURATION, configDomStore), futureExecutor) { @Override - public CheckedFuture submit(DOMDataWriteTransaction transaction, Collection cohorts) { + public CheckedFuture submit(DOMDataWriteTransaction transaction, + Collection cohorts) { commitCohorts.addAll(cohorts); latch.countDown(); return super.submit(transaction, cohorts); @@ -494,19 +499,21 @@ public class ConcurrentDOMDataBrokerTest { }) { DOMDataReadWriteTransaction domDataReadWriteTransaction = dataBroker.newReadWriteTransaction(); - domDataReadWriteTransaction.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); - domDataReadWriteTransaction.merge(LogicalDatastoreType.CONFIGURATION, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); + domDataReadWriteTransaction.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, + mock(NormalizedNode.class)); + domDataReadWriteTransaction.merge(LogicalDatastoreType.CONFIGURATION, YangInstanceIdentifier.EMPTY, + mock(NormalizedNode.class)); domDataReadWriteTransaction.submit(); - latch.await(10, TimeUnit.SECONDS); + assertTrue(latch.await(10, TimeUnit.SECONDS)); assertTrue(commitCohorts.size() == 2); } } @Test - public void testCreateTransactionChain(){ + public void testCreateTransactionChain() { DOMStore domStore = mock(DOMStore.class); try (ConcurrentDOMDataBroker dataBroker = new ConcurrentDOMDataBroker(ImmutableMap.of( LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), @@ -520,7 +527,7 @@ public class ConcurrentDOMDataBrokerTest { } @Test - public void testCreateTransactionOnChain(){ + public void testCreateTransactionOnChain() { DOMStore domStore = mock(DOMStore.class); try (ConcurrentDOMDataBroker dataBroker = new ConcurrentDOMDataBroker(ImmutableMap.of( LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), @@ -532,13 +539,15 @@ public class ConcurrentDOMDataBrokerTest { doReturn(mockChain).when(domStore).createTransactionChain(); doReturn(operationalTransaction).when(mockChain).newWriteOnlyTransaction(); - DOMTransactionChain transactionChain = dataBroker.createTransactionChain(mock(TransactionChainListener.class)); + DOMTransactionChain transactionChain = dataBroker.createTransactionChain( + mock(TransactionChainListener.class)); DOMDataWriteTransaction domDataWriteTransaction = transactionChain.newWriteOnlyTransaction(); verify(mockChain, never()).newWriteOnlyTransaction(); - domDataWriteTransaction.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, mock(NormalizedNode.class)); + domDataWriteTransaction.put(LogicalDatastoreType.OPERATIONAL, YangInstanceIdentifier.EMPTY, + mock(NormalizedNode.class)); } } @@ -549,13 +558,15 @@ public class ConcurrentDOMDataBrokerTest { LogicalDatastoreType.OPERATIONAL, domStore, LogicalDatastoreType.CONFIGURATION, domStore), futureExecutor)) { - CheckedFuture submit1 = dataBroker.newWriteOnlyTransaction().submit(); + CheckedFuture submit1 = + dataBroker.newWriteOnlyTransaction().submit(); assertNotNull(submit1); submit1.get(); - CheckedFuture submit2 = dataBroker.newReadWriteTransaction().submit(); + CheckedFuture submit2 = + dataBroker.newReadWriteTransaction().submit(); assertNotNull(submit2); @@ -563,4 +574,29 @@ public class ConcurrentDOMDataBrokerTest { } } + @Test + public void testExtensions() { + DistributedDataStore mockConfigStore = mock(DistributedDataStore.class); + DistributedDataStore mockOperStore = mock(DistributedDataStore.class); + try (ConcurrentDOMDataBroker dataBroker = new ConcurrentDOMDataBroker(ImmutableMap.of( + LogicalDatastoreType.OPERATIONAL, mockOperStore, + LogicalDatastoreType.CONFIGURATION, mockConfigStore), futureExecutor)) { + + Map, DOMDataBrokerExtension> supportedExtensions = + dataBroker.getSupportedExtensions(); + assertNotNull(supportedExtensions.get(DOMDataTreeChangeService.class)); + + DOMDataTreeCommitCohortRegistry cohortRegistry = + (DOMDataTreeCommitCohortRegistry) supportedExtensions.get(DOMDataTreeCommitCohortRegistry.class); + assertNotNull(cohortRegistry); + + DOMDataTreeCommitCohort mockCohort = mock(DOMDataTreeCommitCohort.class); + DOMDataTreeIdentifier path = new DOMDataTreeIdentifier( + org.opendaylight.mdsal.common.api.LogicalDatastoreType.CONFIGURATION, + YangInstanceIdentifier.EMPTY); + cohortRegistry.registerCommitCohort(path, mockCohort); + + verify(mockConfigStore).registerCommitCohort(path, mockCohort); + } + } }