3 * This program and the accompanying materials are made available under the
4 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
5 * and is available at http://www.eclipse.org/legal/epl-v10.html
7 package org.opendaylight.controller.cluster.databroker;
9 import static org.junit.Assert.assertTrue;
10 import static org.mockito.Matchers.any;
11 import static org.mockito.Mockito.doThrow;
13 import com.google.common.util.concurrent.FluentFuture;
14 import java.util.Collection;
15 import java.util.Collections;
16 import java.util.concurrent.ExecutionException;
17 import org.junit.Assert;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.mockito.Mock;
21 import org.mockito.MockitoAnnotations;
22 import org.opendaylight.mdsal.common.api.CommitInfo;
23 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
24 import org.opendaylight.mdsal.common.api.TransactionCommitFailedException;
25 import org.opendaylight.mdsal.dom.spi.store.DOMStoreWriteTransaction;
27 public class AbstractDOMBrokerWriteTransactionTest {
30 private AbstractDOMTransactionFactory abstractDOMTransactionFactory;
33 private DOMStoreWriteTransaction domStoreWriteTransaction;
35 private class AbstractDOMBrokerWriteTransactionTestImpl
36 extends AbstractDOMBrokerWriteTransaction<DOMStoreWriteTransaction> {
38 AbstractDOMBrokerWriteTransactionTestImpl() {
39 super(new Object(), Collections.emptyMap(), abstractDOMTransactionFactory);
43 protected DOMStoreWriteTransaction createTransaction(LogicalDatastoreType key) {
48 protected Collection<DOMStoreWriteTransaction> getSubtransactions() {
49 return Collections.singletonList(domStoreWriteTransaction);
55 MockitoAnnotations.initMocks(this);
59 public void readyRuntimeExceptionAndCancel() throws InterruptedException {
60 RuntimeException thrown = new RuntimeException();
61 doThrow(thrown).when(domStoreWriteTransaction).ready();
62 AbstractDOMBrokerWriteTransactionTestImpl abstractDOMBrokerWriteTransactionTestImpl =
63 new AbstractDOMBrokerWriteTransactionTestImpl();
65 FluentFuture<? extends CommitInfo> submitFuture = abstractDOMBrokerWriteTransactionTestImpl.commit();
68 Assert.fail("TransactionCommitFailedException expected");
69 } catch (ExecutionException e) {
70 assertTrue(e.getCause() instanceof TransactionCommitFailedException);
71 assertTrue(e.getCause().getCause() == thrown);
72 abstractDOMBrokerWriteTransactionTestImpl.cancel();
77 public void submitRuntimeExceptionAndCancel() throws InterruptedException {
78 RuntimeException thrown = new RuntimeException();
79 doThrow(thrown).when(abstractDOMTransactionFactory).commit(any(), any());
80 AbstractDOMBrokerWriteTransactionTestImpl abstractDOMBrokerWriteTransactionTestImpl
81 = new AbstractDOMBrokerWriteTransactionTestImpl();
83 FluentFuture<? extends CommitInfo> submitFuture = abstractDOMBrokerWriteTransactionTestImpl.commit();
86 Assert.fail("TransactionCommitFailedException expected");
87 } catch (ExecutionException e) {
88 assertTrue(e.getCause() instanceof TransactionCommitFailedException);
89 assertTrue(e.getCause().getCause() == thrown);
90 abstractDOMBrokerWriteTransactionTestImpl.cancel();