2 * Copyright (c) 2014 Cisco 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.openflowplugin.openflow.md.core.session;
10 import com.google.common.util.concurrent.Futures;
11 import java.math.BigInteger;
12 import java.util.Collections;
13 import java.util.List;
14 import org.junit.Assert;
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.junit.runner.RunWith;
18 import org.mockito.ArgumentCaptor;
19 import org.mockito.Matchers;
20 import org.mockito.Mock;
21 import org.mockito.Mockito;
22 import org.mockito.runners.MockitoJUnitRunner;
23 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
24 import org.opendaylight.openflowplugin.api.openflow.md.core.ConnectionConductor;
25 import org.opendaylight.openflowplugin.api.openflow.md.core.session.SessionContext;
26 import org.opendaylight.openflowplugin.api.openflow.md.core.session.SessionManager;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ControllerRole;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierInput;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutput;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutputBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestInput;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestOutput;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestOutputBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.common.config.impl.rev140326.OfpRole;
36 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
41 @RunWith(MockitoJUnitRunner.class)
42 public class OFRoleManagerTest {
45 private SessionManager sessionManager;
47 private SessionContext session;
49 private GetFeaturesOutput features;
51 private ConnectionConductor primaryConductor;
53 private ConnectionAdapter connectionAdapter;
55 private OFRoleManager manager;
56 private RoleRequestOutput roleRequestOutput;
57 private BarrierOutput barrierOutput;
58 private BigInteger generationId = BigInteger.TEN;
65 Mockito.when(session.getFeatures()).thenReturn(features);
66 Mockito.when(features.getVersion()).thenReturn(Short.valueOf((short) 42));
67 Mockito.when(session.getNextXid()).thenReturn(84L);
68 Mockito.when(session.getPrimaryConductor()).thenReturn(primaryConductor);
69 Mockito.when(primaryConductor.getConnectionAdapter()).thenReturn(connectionAdapter);
70 roleRequestOutput = new RoleRequestOutputBuilder()
71 .setGenerationId(generationId)
72 .setRole(ControllerRole.OFPCRROLESLAVE)
73 .setVersion((short) 42)
76 barrierOutput = new BarrierOutputBuilder()
77 .setVersion((short) 42)
81 manager = new OFRoleManager(sessionManager);
85 * Test method for {@link org.opendaylight.openflowplugin.openflow.md.core.session.OFRoleManager#manageRoleChange(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.common.config.impl.rev140326.OfpRole)}.
87 * session lot is empty is invalid
90 public void testManageRoleChangeFail1() {
91 manager.manageRoleChange(OfpRole.BECOMESLAVE);
92 Mockito.verify(connectionAdapter, Mockito.never()).roleRequest(Matchers.any(RoleRequestInput.class));
96 * Test method for {@link org.opendaylight.openflowplugin.openflow.md.core.session.OFRoleManager#manageRoleChange(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.common.config.impl.rev140326.OfpRole)}.
98 * available session is invalid
101 public void testManageRoleChangeFail2() {
102 Mockito.when(sessionManager.getAllSessions()).thenReturn(Collections.singleton(session));
103 manager.manageRoleChange(OfpRole.BECOMESLAVE);
104 Mockito.verify(connectionAdapter, Mockito.never()).roleRequest(Matchers.any(RoleRequestInput.class));
108 * Test method for {@link org.opendaylight.openflowplugin.openflow.md.core.session.OFRoleManager#manageRoleChange(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.common.config.impl.rev140326.OfpRole)}.
110 * device bound to valid session is not answering
113 public void testManageRoleChangeFail3() {
114 Mockito.when(session.isValid()).thenReturn(true);
115 Mockito.when(sessionManager.getAllSessions()).thenReturn(Collections.singleton(session));
116 // manager.manageRoleChange(OfpRole.BECOMESLAVE);
117 // Mockito.verify(connectionAdapter, Mockito.times(1)).roleRequest(Matchers.any(RoleRequestInput.class));
121 * Test method for {@link org.opendaylight.openflowplugin.openflow.md.core.session.OFRoleManager#manageRoleChange(org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.common.config.impl.rev140326.OfpRole)}.
124 public void testManageRoleChangeOK() {
125 Mockito.when(session.isValid()).thenReturn(true);
126 Mockito.when(sessionManager.getAllSessions()).thenReturn(Collections.singleton(session));
127 Mockito.when(connectionAdapter.roleRequest(Matchers.any(RoleRequestInput.class)))
128 .thenReturn(Futures.immediateFuture(RpcResultBuilder.success(roleRequestOutput).build()));
129 Mockito.when(connectionAdapter.barrier(Matchers.any(BarrierInput.class)))
130 .thenReturn(Futures.immediateFuture(RpcResultBuilder.success(barrierOutput).build()));
132 //manager.manageRoleChange(OfpRole.BECOMESLAVE);
134 ArgumentCaptor<RoleRequestInput> roleRequestCaptor = ArgumentCaptor.forClass(RoleRequestInput.class);
135 //Mockito.verify(connectionAdapter, Mockito.times(2)).roleRequest(roleRequestCaptor.capture());
137 // List<RoleRequestInput> values = roleRequestCaptor.getAllValues();
138 // Assert.assertEquals(ControllerRole.OFPCRROLENOCHANGE, values.get(0).getRole());
139 // Assert.assertEquals(0L, values.get(0).getGenerationId().longValue());
140 // Assert.assertEquals(ControllerRole.OFPCRROLESLAVE, values.get(1).getRole());
141 // Assert.assertEquals(11L, values.get(1).getGenerationId().longValue());