2 * Copyright (c) 2016 Brocade Communications 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.controller.config.yang.bmp.impl;
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;
34 * Abstract base class for bmp module tests.
36 * @author Thomas Pantelis
38 public class AbstractBmpModuleTest extends AbstractConfigTest {
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();
46 }).when(this.mockedContext).createFilter(anyString());
48 Mockito.doNothing().when(this.mockedContext).addServiceListener(any(ServiceListener.class), Mockito.anyString());
49 Mockito.doNothing().when(this.mockedContext).removeServiceListener(any(ServiceListener.class));
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);
57 setupMockService(BmpExtensionProviderContext.class, new SimpleBmpExtensionProviderContext());
59 setupMockService(RIBExtensionProviderContext.class, new SimpleRIBExtensionProviderContext());
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);