bgp/bmp-impl/src/test/java/org/opendaylight/controller/config/yang/bmp/impl/AbstractBmpModuleTest.java

   1 /*
   2  * Copyright (c) 2016 Brocade Communications 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.controller.config.yang.bmp.impl;
   9
  10 import static org.mockito.Matchers.any;
  11 import static org.mockito.Matchers.anyString;
  12 import static org.mockito.Matchers.contains;
  13 import static org.mockito.Mockito.doAnswer;
  14 import static org.mockito.Mockito.doReturn;
  15 import static org.mockito.Mockito.mock;
  16 import org.junit.Before;
  17 import org.mockito.Mockito;
  18 import org.mockito.invocation.InvocationOnMock;
  19 import org.mockito.stubbing.Answer;
  20 import org.opendaylight.controller.config.manager.impl.AbstractConfigTest;
  21 import org.opendaylight.protocol.bgp.parser.spi.AddressFamilyRegistry;
  22 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
  23 import org.opendaylight.protocol.bgp.parser.spi.MessageRegistry;
  24 import org.opendaylight.protocol.bgp.parser.spi.SubsequentAddressFamilyRegistry;
  25 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionProviderContext;
  26 import org.opendaylight.protocol.bgp.rib.spi.SimpleRIBExtensionProviderContext;
  27 import org.opendaylight.protocol.bmp.spi.registry.BmpExtensionProviderContext;
  28 import org.opendaylight.protocol.bmp.spi.registry.SimpleBmpExtensionProviderContext;
  29 import org.osgi.framework.Filter;
  30 import org.osgi.framework.ServiceListener;
  31 import org.osgi.framework.ServiceReference;
  32
  33 /**
  34  * Abstract base class for bmp module tests.
  35  *
  36  * @author Thomas Pantelis
  37  */
  38 public class AbstractBmpModuleTest extends AbstractConfigTest {
  39     @Before
  40     public void setUp() throws Exception {
  41         doAnswer(invocation -> {
  42             final String str = invocation.getArgumentAt(0, String.class);
  43             final Filter mockFilter = mock(Filter.class);
  44             doReturn(str).when(mockFilter).toString();
  45             return mockFilter;
  46         }).when(this.mockedContext).createFilter(anyString());
  47
  48         Mockito.doNothing().when(this.mockedContext).addServiceListener(any(ServiceListener.class), Mockito.anyString());
  49         Mockito.doNothing().when(this.mockedContext).removeServiceListener(any(ServiceListener.class));
  50
  51         BGPExtensionProviderContext mockContext = mock(BGPExtensionProviderContext.class);
  52         doReturn(mock(AddressFamilyRegistry.class)).when(mockContext).getAddressFamilyRegistry();
  53         doReturn(mock(MessageRegistry.class)).when(mockContext).getMessageRegistry();
  54         doReturn(mock(SubsequentAddressFamilyRegistry.class)).when(mockContext).getSubsequentAddressFamilyRegistry();
  55         setupMockService(BGPExtensionProviderContext.class, mockContext);
  56
  57         setupMockService(BmpExtensionProviderContext.class, new SimpleBmpExtensionProviderContext());
  58
  59         setupMockService(RIBExtensionProviderContext.class, new SimpleRIBExtensionProviderContext());
  60     }
  61
  62     void setupMockService(final Class<?> serviceInterface, final Object instance) throws Exception {
  63         final ServiceReference<?> mockServiceRef = mock(ServiceReference.class);
  64         doReturn(new ServiceReference[]{mockServiceRef}).when(this.mockedContext).
  65                 getServiceReferences(anyString(), contains(serviceInterface.getName()));
  66         doReturn(new ServiceReference[]{mockServiceRef}).when(this.mockedContext).
  67                 getServiceReferences(serviceInterface.getName(), null);
  68         doReturn(instance).when(this.mockedContext).getService(mockServiceRef);
  69     }
  70 }