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 org.junit.Before;
16 import org.mockito.Mock;
17 import org.opendaylight.mdsal.binding.api.RpcProviderService;
18 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractConcurrentDataBrokerTest;
19 import org.opendaylight.mdsal.singleton.api.ClusterSingletonService;
20 import org.opendaylight.mdsal.singleton.api.ClusterSingletonServiceProvider;
21 import org.opendaylight.yangtools.concepts.Registration;
23 abstract class AbstractProgrammingTest extends AbstractConcurrentDataBrokerTest {
25 RpcProviderService rpcRegistry;
27 ClusterSingletonServiceProvider cssp;
29 Registration singletonServiceRegistration;
31 private Registration registration;
32 ClusterSingletonService singletonService;
34 AbstractProgrammingTest() {
39 public void setUp() throws Exception {
40 doAnswer(invocationOnMock -> {
41 singletonService = invocationOnMock.getArgument(0);
42 return singletonServiceRegistration;
43 }).when(cssp).registerClusterSingletonService(any(ClusterSingletonService.class));
44 doAnswer(invocationOnMock -> {
45 singletonService.closeServiceInstance().get();
47 }).when(singletonServiceRegistration).close();
49 doReturn(registration).when(rpcRegistry).registerRpcImplementations(any());
51 doNothing().when(registration).close();