2 * Copyright (c) 2017 Pantheon Technologies s.r.o. 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.bgpcep.programming.impl;
10 import static org.mockito.ArgumentMatchers.any;
11 import static org.mockito.Mockito.doAnswer;
12 import static org.mockito.Mockito.doNothing;
13 import static org.mockito.Mockito.doReturn;
15 import com.google.common.collect.ClassToInstanceMap;
16 import org.junit.Before;
17 import org.mockito.Mock;
18 import org.opendaylight.mdsal.binding.api.RpcProviderService;
19 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractConcurrentDataBrokerTest;
20 import org.opendaylight.mdsal.singleton.api.ClusterSingletonService;
21 import org.opendaylight.mdsal.singleton.api.ClusterSingletonServiceProvider;
22 import org.opendaylight.yangtools.concepts.Registration;
24 abstract class AbstractProgrammingTest extends AbstractConcurrentDataBrokerTest {
26 RpcProviderService rpcRegistry;
28 ClusterSingletonServiceProvider cssp;
30 Registration singletonServiceRegistration;
32 private Registration registration;
33 ClusterSingletonService singletonService;
35 AbstractProgrammingTest() {
40 public void setUp() throws Exception {
41 doAnswer(invocationOnMock -> {
42 singletonService = invocationOnMock.getArgument(0);
43 return singletonServiceRegistration;
44 }).when(cssp).registerClusterSingletonService(any(ClusterSingletonService.class));
45 doAnswer(invocationOnMock -> {
46 singletonService.closeServiceInstance().get();
48 }).when(singletonServiceRegistration).close();
50 doReturn(registration).when(rpcRegistry).registerRpcImplementations(any(ClassToInstanceMap.class));
52 doNothing().when(registration).close();