2 * Copyright (c) 2017 Red Hat, Inc. and others. All rights reserved.
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
8 package org.opendaylight.mdsal.binding.testutils;
10 import static org.junit.Assert.assertTrue;
11 import static org.junit.Assert.fail;
12 import static org.junit.runners.MethodSorters.NAME_ASCENDING;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
16 import java.util.concurrent.ExecutionException;
17 import java.util.concurrent.TimeUnit;
18 import java.util.concurrent.TimeoutException;
19 import org.junit.Before;
20 import org.junit.FixMethodOrder;
21 import org.junit.Test;
22 import org.opendaylight.mdsal.binding.api.DataBroker;
23 import org.opendaylight.mdsal.binding.api.ReadWriteTransaction;
24 import org.opendaylight.mdsal.binding.api.WriteTransaction;
25 import org.opendaylight.mdsal.common.api.CommitInfo;
26 import org.opendaylight.mdsal.common.api.OptimisticLockFailedException;
27 import org.opendaylight.mdsal.common.api.TransactionCommitFailedException;
30 * Unit test for DataBrokerFailuresImpl.
32 * @author Michael Vorburger.ch
34 @FixMethodOrder(NAME_ASCENDING)
35 public class DataBrokerFailuresTest {
37 private final DataBrokerFailures dbFailures;
38 private final DataBroker dataBroker;
40 public DataBrokerFailuresTest() {
41 DataBroker mockDataBroker = mock(DataBroker.class);
42 ReadWriteTransaction readWriteTransaction = mock(ReadWriteTransaction.class);
43 doReturn(CommitInfo.emptyFluentFuture()).when(readWriteTransaction).commit();
44 doReturn(readWriteTransaction).when(mockDataBroker).newReadWriteTransaction();
45 WriteTransaction writeTransaction = mock(WriteTransaction.class);
46 doReturn(CommitInfo.emptyFluentFuture()).when(writeTransaction).commit();
47 doReturn(writeTransaction).when(mockDataBroker).newWriteOnlyTransaction();
48 dbFailures = new DataBrokerFailuresImpl(mockDataBroker);
49 dataBroker = (DataBroker) dbFailures;
58 public void testFailReadWriteTransactionCommit() throws TimeoutException, InterruptedException {
59 dbFailures.failCommits(new OptimisticLockFailedException("bada boum bam!"));
61 // Now make sure that it still fails, and not just once:
67 private void checkCommitFails() throws TimeoutException, InterruptedException {
69 dataBroker.newReadWriteTransaction().commit().get(5, TimeUnit.SECONDS);
70 fail("This should have led to a TransactionCommitFailedException!");
71 } catch (ExecutionException e) {
72 assertTrue("Expected TransactionCommitFailedException",
73 e.getCause() instanceof TransactionCommitFailedException);
78 public void testFailReadWriteTransactionCommitNext()
79 throws TimeoutException, InterruptedException, ExecutionException {
80 // This must pass (the failCommits from previous test cannot affect this)
81 // (It's a completely new instance of DataBroker & DataBrokerFailures anyways, but just to be to sure.)
82 dataBroker.newReadWriteTransaction().commit().get(5, TimeUnit.SECONDS);
86 public void testFailTwoReadWriteTransactionCommit()
87 throws TimeoutException, InterruptedException, ExecutionException {
88 dbFailures.failCommits(2, new OptimisticLockFailedException("bada boum bam!"));
90 // Now make sure that it still fails again a 2nd time, and not just once:
92 // But now it should pass.. because we specified howManyTimes = 2 above
93 dataBroker.newReadWriteTransaction().commit().get(5, TimeUnit.SECONDS);
94 dataBroker.newWriteOnlyTransaction().commit().get(5, TimeUnit.SECONDS);
95 dataBroker.newReadWriteTransaction().commit().get(5, TimeUnit.SECONDS);
98 @Test(expected = OptimisticLockFailedException.class)
99 @SuppressWarnings("checkstyle:AvoidHidingCauseException")
100 public void testFailWriteTransactionCommit()
101 throws TimeoutException, InterruptedException, TransactionCommitFailedException {
102 dbFailures.failCommits(new OptimisticLockFailedException("bada boum bam!"));
104 dataBroker.newWriteOnlyTransaction().commit().get(5, TimeUnit.SECONDS);
105 } catch (ExecutionException e) {
106 assertTrue("Expected TransactionCommitFailedException",
107 e.getCause() instanceof TransactionCommitFailedException);
108 throw (TransactionCommitFailedException)e.getCause();
113 public void testUnfailCommits() throws TimeoutException, InterruptedException, ExecutionException {
114 dbFailures.failCommits(new OptimisticLockFailedException("bada boum bam!"));
116 dbFailures.unfailCommits();
117 dataBroker.newReadWriteTransaction().commit().get(5, TimeUnit.SECONDS);
118 dataBroker.newWriteOnlyTransaction().commit().get(5, TimeUnit.SECONDS);
119 dataBroker.newReadWriteTransaction().commit().get(5, TimeUnit.SECONDS);
123 public void testFailButCommitAnywayReadWriteTransaction() throws TimeoutException, InterruptedException {
124 dbFailures.failButCommitAnyway();
128 // TODO make this work for TransactionChain as well ...