2 * Copyright (c) 2016 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.openflowplugin.impl.lifecycle;
10 import com.google.common.util.concurrent.Futures;
11 import org.junit.Assert;
12 import org.junit.Before;
13 import org.junit.Test;
14 import org.junit.runner.RunWith;
15 import org.mockito.Mock;
16 import org.mockito.Mockito;
17 import org.mockito.runners.MockitoJUnitRunner;
18 import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider;
19 import org.opendaylight.mdsal.singleton.common.api.ServiceGroupIdentifier;
20 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
21 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
22 import org.opendaylight.openflowplugin.api.openflow.device.DeviceInfo;
23 import org.opendaylight.openflowplugin.api.openflow.device.handlers.ClusterInitializationPhaseHandler;
24 import org.opendaylight.openflowplugin.api.openflow.lifecycle.LifecycleService;
25 import org.opendaylight.openflowplugin.api.openflow.registry.flow.DeviceFlowRegistry;
26 import org.opendaylight.openflowplugin.api.openflow.role.RoleContext;
27 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
28 import org.opendaylight.openflowplugin.api.openflow.statistics.StatisticsContext;
30 @RunWith(MockitoJUnitRunner.class)
31 public class LifecycleServiceImplTest {
33 private static final String TEST_NODE = "test node";
34 private static final ServiceGroupIdentifier SERVICE_GROUP_IDENTIFIER = ServiceGroupIdentifier.create(TEST_NODE);
37 private DeviceInfo deviceInfo;
39 private DeviceContext deviceContext;
41 private RpcContext rpcContext;
43 private RoleContext roleContext;
45 private StatisticsContext statContext;
47 private ConnectionContext connectionContext;
49 private DeviceFlowRegistry deviceFlowRegistry;
51 private ClusterSingletonServiceProvider clusterSingletonServiceProvider;
53 private LifecycleService lifecycleService;
57 Mockito.when(deviceContext.getDeviceInfo()).thenReturn(deviceInfo);
58 Mockito.when(deviceContext.getPrimaryConnectionContext()).thenReturn(connectionContext);
59 Mockito.when(deviceContext.getDeviceFlowRegistry()).thenReturn(deviceFlowRegistry);
60 Mockito.when(deviceContext.getServiceIdentifier()).thenReturn(SERVICE_GROUP_IDENTIFIER);
61 Mockito.when(deviceFlowRegistry.fill()).thenReturn(Futures.immediateFuture(null));
62 Mockito.when(connectionContext.getConnectionState()).thenReturn(ConnectionContext.CONNECTION_STATE.WORKING);
63 Mockito.when(deviceInfo.getLOGValue()).thenReturn(TEST_NODE);
65 lifecycleService = new LifecycleServiceImpl();
66 lifecycleService.setDeviceContext(deviceContext);
67 lifecycleService.setRpcContext(rpcContext);
68 lifecycleService.setRoleContext(roleContext);
69 lifecycleService.setStatContext(statContext);
70 lifecycleService.registerService(clusterSingletonServiceProvider);
74 public void instantiateServiceInstance() throws Exception {
75 lifecycleService.instantiateServiceInstance();
76 Mockito.verify(deviceContext).setLifecycleInitializationPhaseHandler(Mockito.<ClusterInitializationPhaseHandler>any());
77 Mockito.verify(statContext).setLifecycleInitializationPhaseHandler(Mockito.<ClusterInitializationPhaseHandler>any());
78 Mockito.verify(statContext).setInitialSubmitHandler(Mockito.<ClusterInitializationPhaseHandler>any());
79 Mockito.verify(rpcContext).setLifecycleInitializationPhaseHandler(Mockito.<ClusterInitializationPhaseHandler>any());
80 Mockito.verify(roleContext).setLifecycleInitializationPhaseHandler(Mockito.<ClusterInitializationPhaseHandler>any());
84 public void closeServiceInstance() throws Exception {
85 lifecycleService.closeServiceInstance();
86 Mockito.verify(statContext).stopClusterServices(false);
87 Mockito.verify(deviceContext).stopClusterServices(false);
88 Mockito.verify(rpcContext).stopClusterServices(false);
89 Mockito.verify(roleContext).stopClusterServices(false);
93 public void getIdentifier() throws Exception {
94 Assert.assertEquals(lifecycleService.getIdentifier(), SERVICE_GROUP_IDENTIFIER);
98 public void closeConnection() throws Exception {
99 lifecycleService.closeConnection();
100 Mockito.verify(deviceContext).shutdownConnection();