2 * Copyright (c) 2015 Cisco Systems, 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.protocol.bgp.openconfig.impl.openconfig;
10 import static org.junit.Assert.assertTrue;
12 import com.google.common.util.concurrent.CheckedFuture;
13 import org.junit.Before;
14 import org.junit.Test;
15 import org.mockito.Mockito;
16 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
17 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
18 import org.opendaylight.controller.md.sal.binding.impl.BindingDOMDataBrokerAdapter;
19 import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
20 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
21 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
22 import org.opendaylight.protocol.bgp.openconfig.impl.spi.BGPConfigHolder;
23 import org.opendaylight.protocol.bgp.openconfig.impl.spi.BGPConfigStateStore;
24 import org.opendaylight.protocol.bgp.openconfig.spi.pojo.BGPAppPeerInstanceConfiguration;
25 import org.opendaylight.protocol.bgp.openconfig.spi.pojo.BGPPeerInstanceConfiguration;
26 import org.opendaylight.protocol.bgp.openconfig.spi.pojo.BGPRibInstanceConfiguration;
27 import org.opendaylight.yangtools.yang.binding.DataObject;
28 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
30 public class BGPConfigModuleMapperProviderTest {
32 private BGPConfigModuleMapperProvider mapperProvider;
34 private final BindingTransactionChain txChain = Mockito.mock(BindingTransactionChain.class);
35 private final AsyncTransaction<?, ?> asyncTx = Mockito.mock(AsyncTransaction.class);
36 private final WriteTransaction myTx = Mockito.mock(WriteTransaction.class);
38 @SuppressWarnings("unchecked")
40 public void setUp() throws Exception {
42 final BindingDOMDataBrokerAdapter dataBroker = Mockito.mock(BindingDOMDataBrokerAdapter.class);
43 final BGPConfigStateStore stateHolders = Mockito.mock(BGPConfigStateStore.class);
44 final CheckedFuture<?, ?> checkedFuture = Mockito.mock(CheckedFuture.class);
45 final BGPConfigHolder<?> confHolder = Mockito.mock(BGPConfigHolder.class);
47 Mockito.doReturn(this.txChain).when(dataBroker).createTransactionChain(Mockito.any(TransactionChainListener.class));
48 Mockito.doReturn(this.myTx).when(this.txChain).newWriteOnlyTransaction();
49 Mockito.doNothing().when(this.txChain).close();
50 Mockito.doNothing().when(this.myTx).put(Mockito.any(LogicalDatastoreType.class), Mockito.any(InstanceIdentifier.class), Mockito.any(DataObject.class));
51 Mockito.doNothing().when(this.myTx).delete(Mockito.any(LogicalDatastoreType.class), Mockito.any(InstanceIdentifier.class));
52 Mockito.doReturn(checkedFuture).when(this.myTx).submit();
53 Mockito.doReturn(null).when(checkedFuture).checkedGet();
54 Mockito.doReturn(confHolder).when(stateHolders).getBGPConfigHolder(Mockito.any(Class.class));
56 this.mapperProvider = new BGPConfigModuleMapperProvider(dataBroker, stateHolders);
60 public void testGetOpenConfigMapper() {
61 assertTrue(this.mapperProvider.getOpenConfigMapper(BGPRibInstanceConfiguration.class) instanceof BGPGlobalProviderImpl);
62 assertTrue(this.mapperProvider.getOpenConfigMapper(BGPPeerInstanceConfiguration.class) instanceof BGPNeighborProviderImpl);
63 assertTrue(this.mapperProvider.getOpenConfigMapper(BGPAppPeerInstanceConfiguration.class) instanceof BGPAppNeighborProviderImpl);
67 public void testOnTransactionChainFailed() {
68 this.mapperProvider.onTransactionChainFailed(this.txChain, this.asyncTx, new Throwable());
69 Mockito.verify(this.txChain).close();
73 public void testClose() throws Exception {
74 this.mapperProvider.close();
75 Mockito.verify(this.myTx).delete(Mockito.any(LogicalDatastoreType.class), Mockito.any(InstanceIdentifier.class));
76 Mockito.verify(this.myTx, Mockito.times(1)).submit();
77 Mockito.verify(this.txChain).close();