protocol/netconf-server/src/test/java/org/opendaylight/netconf/server/osgi/AggregatedNetconfOperationServiceFactoryTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.netconf.server.osgi;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.mockito.Mockito.doNothing;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.times;
  14 import static org.mockito.Mockito.verify;
  15
  16 import com.google.common.collect.Sets;
  17 import java.util.HashSet;
  18 import java.util.Set;
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.junit.runner.RunWith;
  22 import org.mockito.Mock;
  23 import org.mockito.junit.MockitoJUnitRunner;
  24 import org.opendaylight.netconf.api.capability.BasicCapability;
  25 import org.opendaylight.netconf.api.capability.Capability;
  26 import org.opendaylight.netconf.api.monitoring.CapabilityListener;
  27 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory;
  28 import org.opendaylight.yangtools.concepts.Registration;
  29
  30 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  31 public class AggregatedNetconfOperationServiceFactoryTest {
  32
  33     private final Set<Capability> factory1Caps = new HashSet<>();
  34     private final Set<Capability> factory2Caps = new HashSet<>();
  35
  36     @Mock
  37     private CapabilityListener listener1;
  38     @Mock
  39     private CapabilityListener listener2;
  40     @Mock
  41     private CapabilityListener listener3;
  42     @Mock
  43     private NetconfOperationServiceFactory factory1;
  44     @Mock
  45     private NetconfOperationServiceFactory factory2;
  46     @Mock
  47     private Registration reg1;
  48     @Mock
  49     private Registration reg2;
  50     @Mock
  51     private Registration reg3;
  52
  53     private AggregatedNetconfOperationServiceFactory aggregatedFactory;
  54
  55     @Before
  56     public void setUp() throws Exception {
  57         factory1Caps.add(new BasicCapability("AAA"));
  58         factory1Caps.add(new BasicCapability("BBB"));
  59
  60         factory2Caps.add(new BasicCapability("CCC"));
  61         factory2Caps.add(new BasicCapability("DDD"));
  62
  63         aggregatedFactory = new AggregatedNetconfOperationServiceFactory();
  64
  65         aggregatedFactory.registerCapabilityListener(listener1);
  66         aggregatedFactory.registerCapabilityListener(listener2);
  67
  68         doReturn(reg1).when(factory1).registerCapabilityListener(listener1);
  69         doReturn(reg2).when(factory1).registerCapabilityListener(listener2);
  70         doReturn(factory1Caps).when(factory1).getCapabilities();
  71
  72         doReturn(reg1).when(factory2).registerCapabilityListener(listener1);
  73         doReturn(reg2).when(factory2).registerCapabilityListener(listener2);
  74         doReturn(factory2Caps).when(factory2).getCapabilities();
  75
  76         doNothing().when(reg1).close();
  77         doNothing().when(reg2).close();
  78
  79         doReturn(reg3).when(factory1).registerCapabilityListener(listener3);
  80         doReturn(reg3).when(factory2).registerCapabilityListener(listener3);
  81     }
  82
  83     @Test
  84     public void testOnAddAndOnRemove() throws Exception {
  85         aggregatedFactory.onAddNetconfOperationServiceFactory(factory1);
  86         aggregatedFactory.onAddNetconfOperationServiceFactory(factory2);
  87
  88         verify(factory1).registerCapabilityListener(listener1);
  89         verify(factory2).registerCapabilityListener(listener1);
  90         verify(factory1).registerCapabilityListener(listener2);
  91         verify(factory2).registerCapabilityListener(listener2);
  92
  93         aggregatedFactory.onRemoveNetconfOperationServiceFactory(factory1);
  94         aggregatedFactory.onRemoveNetconfOperationServiceFactory(factory2);
  95
  96         verify(reg1, times(2)).close();
  97         verify(reg2, times(2)).close();
  98     }
  99
 100     @Test
 101     public void testGetCapabilities() throws Exception {
 102         aggregatedFactory.onAddNetconfOperationServiceFactory(factory1);
 103         aggregatedFactory.onAddNetconfOperationServiceFactory(factory2);
 104         final Set<Capability> actual = aggregatedFactory.getCapabilities();
 105         Set<Capability> expected = Sets.union(factory1Caps, factory2Caps);
 106         assertEquals(expected, actual);
 107     }
 108
 109     @Test
 110     public void testRegisterCapabilityListener() throws Exception {
 111         aggregatedFactory.onAddNetconfOperationServiceFactory(factory1);
 112         aggregatedFactory.onAddNetconfOperationServiceFactory(factory2);
 113         aggregatedFactory.registerCapabilityListener(listener3);
 114
 115         verify(factory1).registerCapabilityListener(listener3);
 116         verify(factory2).registerCapabilityListener(listener3);
 117     }
 118
 119     @Test
 120     public void testClose() throws Exception {
 121         aggregatedFactory.onAddNetconfOperationServiceFactory(factory1);
 122         aggregatedFactory.onAddNetconfOperationServiceFactory(factory2);
 123         aggregatedFactory.close();
 124         verify(reg1, times(2)).close();
 125         verify(reg2, times(2)).close();
 126     }
 127 }