openflowplugin/src/test/java/org/opendaylight/openflowplugin/openflow/md/core/session/OFRoleManagerTest.java

   1 /**
   2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.openflowplugin.openflow.md.core.session;
   9
  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;
  37
  38 /**
  39  *
  40  */
  41 @RunWith(MockitoJUnitRunner.class)
  42 public class OFRoleManagerTest {
  43
  44     @Mock
  45     private SessionManager sessionManager;
  46     @Mock
  47     private SessionContext session;
  48     @Mock
  49     private GetFeaturesOutput features;
  50     @Mock
  51     private ConnectionConductor primaryConductor;
  52     @Mock
  53     private ConnectionAdapter connectionAdapter;
  54
  55     private OFRoleManager manager;
  56     private RoleRequestOutput roleRequestOutput;
  57     private BarrierOutput barrierOutput;
  58     private BigInteger generationId = BigInteger.TEN;
  59
  60     /**
  61      * prepare values
  62      */
  63     @Before
  64     public void setUp() {
  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)
  74                 .setXid(21L)
  75                 .build();
  76         barrierOutput = new BarrierOutputBuilder()
  77                 .setVersion((short) 42)
  78                 .setXid(1L)
  79                 .build();
  80
  81         manager = new OFRoleManager(sessionManager);
  82     }
  83
  84     /**
  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)}.
  86      * <br/>
  87      * session lot is empty is invalid
  88      */
  89     @Test
  90     public void testManageRoleChangeFail1() {
  91         manager.manageRoleChange(OfpRole.BECOMESLAVE);
  92         Mockito.verify(connectionAdapter, Mockito.never()).roleRequest(Matchers.any(RoleRequestInput.class));
  93     }
  94
  95     /**
  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)}.
  97      * <br/>
  98      * available session is invalid
  99      */
 100     @Test
 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));
 105     }
 106
 107     /**
 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)}.
 109      * <br/>
 110      * device bound to valid session is not answering
 111      */
 112     @Test
 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));
 118     }
 119
 120     /**
 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)}.
 122      */
 123     @Test
 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()));
 131
 132         manager.manageRoleChange(OfpRole.BECOMESLAVE);
 133
 134         ArgumentCaptor<RoleRequestInput> roleRequestCaptor = ArgumentCaptor.forClass(RoleRequestInput.class);
 135         Mockito.verify(connectionAdapter, Mockito.times(2)).roleRequest(roleRequestCaptor.capture());
 136
 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());
 142     }
 143 }