2 * Copyright (c) 2016 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
9 package org.opendaylight.groupbasedpolicy.gbp_ise_adapter.impl;
11 import com.google.common.util.concurrent.Futures;
12 import java.util.Collections;
13 import org.junit.Before;
14 import org.junit.Test;
15 import org.junit.runner.RunWith;
16 import org.mockito.InOrder;
17 import org.mockito.Matchers;
18 import org.mockito.Mock;
19 import org.mockito.Mockito;
20 import org.mockito.runners.MockitoJUnitRunner;
21 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
22 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
23 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
24 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.gbp.ise.adapter.model.rev160630.gbp.ise.adapter.IseHarvestConfig;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.gbp.ise.adapter.model.rev160630.gbp.ise.adapter.IseHarvestStatus;
28 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
31 * Test for {@link GbpIseConfigListenerImpl}.
33 @RunWith(MockitoJUnitRunner.class)
34 public class GbpIseConfigListenerImplTest {
37 private DataBroker dataBroker;
39 private GbpIseSgtHarvester harvester;
41 private DataTreeModification<IseHarvestConfig> treeModification;
43 private DataObjectModification<IseHarvestConfig> dataModification;
45 private WriteTransaction wTx;
47 private IseHarvestConfig config;
49 private GbpIseConfigListenerImpl listener;
52 public void setUp() throws Exception {
53 listener = new GbpIseConfigListenerImpl(dataBroker, harvester);
57 public void testOnDataTreeChanged_noop() throws Exception {
58 Mockito.when(dataModification.getDataAfter()).thenReturn(null);
59 Mockito.when(treeModification.getRootNode()).thenReturn(dataModification);
61 listener.onDataTreeChanged(Collections.singleton(treeModification));
62 Mockito.verifyNoMoreInteractions(harvester, dataBroker);
66 public void testOnDataTreeChanged_succeeded() throws Exception {
67 Mockito.when(dataModification.getDataAfter()).thenReturn(config);
68 Mockito.when(treeModification.getRootNode()).thenReturn(dataModification);
70 Mockito.when(harvester.harvest(config)).thenReturn(Futures.immediateFuture(42));
72 Mockito.when(wTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
73 Mockito.when(dataBroker.newWriteOnlyTransaction()).thenReturn(wTx);
75 listener.onDataTreeChanged(Collections.singleton(treeModification));
78 final InOrder inOrder = Mockito.inOrder(harvester, dataBroker, wTx);
79 inOrder.verify(harvester).harvest(config);
80 inOrder.verify(dataBroker).newWriteOnlyTransaction();
81 inOrder.verify(wTx).put(Matchers.eq(LogicalDatastoreType.OPERATIONAL),
82 Matchers.<InstanceIdentifier<IseHarvestStatus>>any(),
83 Matchers.<IseHarvestStatus>any(),
85 inOrder.verify(wTx).submit();
86 inOrder.verifyNoMoreInteractions();
90 public void testOnDataTreeChanged_failed() throws Exception {
91 Mockito.when(dataModification.getDataAfter()).thenReturn(config);
92 Mockito.when(treeModification.getRootNode()).thenReturn(dataModification);
94 Mockito.when(harvester.harvest(config)).thenReturn(Futures.immediateFailedFuture(
95 new Exception("extremely poor harvest occurred")));
97 Mockito.when(wTx.submit()).thenReturn(Futures.immediateCheckedFuture(null));
98 Mockito.when(dataBroker.newWriteOnlyTransaction()).thenReturn(wTx);
100 listener.onDataTreeChanged(Collections.singleton(treeModification));
103 final InOrder inOrder = Mockito.inOrder(harvester, dataBroker, wTx);
104 inOrder.verify(harvester).harvest(config);
105 inOrder.verify(dataBroker).newWriteOnlyTransaction();
106 inOrder.verify(wTx).put(Matchers.eq(LogicalDatastoreType.OPERATIONAL),
107 Matchers.<InstanceIdentifier<IseHarvestStatus>>any(),
108 Matchers.<IseHarvestStatus>any(),
110 inOrder.verify(wTx).submit();
111 inOrder.verifyNoMoreInteractions();