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.common.api.ClusterSingletonService;
20 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
21 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceRegistration;
22 import org.opendaylight.yangtools.concepts.Registration;
24 abstract class AbstractProgrammingTest extends AbstractConcurrentDataBrokerTest {
26 RpcProviderService rpcRegistry;
28 ClusterSingletonServiceProvider cssp;
30 ClusterSingletonServiceRegistration singletonServiceRegistration;
32 private Registration registration;
33 ClusterSingletonService singletonService;
35 AbstractProgrammingTest() {
40 public void setUp() throws Exception {
41 doAnswer(invocationOnMock -> {
42 this.singletonService = (ClusterSingletonService) invocationOnMock.getArguments()[0];
43 return this.singletonServiceRegistration;
44 }).when(this.cssp).registerClusterSingletonService(any(ClusterSingletonService.class));
45 doAnswer(invocationOnMock -> {
46 this.singletonService.closeServiceInstance().get();
48 }).when(this.singletonServiceRegistration).close();
50 doReturn(this.registration).when(this.rpcRegistry).registerRpcImplementations(any());
52 doNothing().when(this.registration).close();