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 org.junit.Before;
  14 import org.junit.Test;
  15 import org.junit.runner.RunWith;
  16 import org.mockito.ArgumentCaptor;
  17 import org.mockito.Matchers;
  18 import org.mockito.Mock;
  19 import org.mockito.Mockito;
  20 import org.mockito.runners.MockitoJUnitRunner;
  21 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
  22 import org.opendaylight.openflowplugin.api.openflow.md.core.ConnectionConductor;
  23 import org.opendaylight.openflowplugin.api.openflow.md.core.session.SessionContext;
  24 import org.opendaylight.openflowplugin.api.openflow.md.core.session.SessionManager;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.ControllerRole;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierInput;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutput;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.BarrierOutputBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.GetFeaturesOutput;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestInput;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestOutput;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.RoleRequestOutputBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.openflow.common.config.impl.rev140326.OfpRole;
  34 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
  35
  36 /**
  37  *
  38  */
  39 @RunWith(MockitoJUnitRunner.class)
  40 public class OFRoleManagerTest {
  41
  42     @Mock
  43     private SessionManager sessionManager;
  44     @Mock
  45     private SessionContext session;
  46     @Mock
  47     private GetFeaturesOutput features;
  48     @Mock
  49     private ConnectionConductor primaryConductor;
  50     @Mock
  51     private ConnectionAdapter connectionAdapter;
  52
  53     private OFRoleManager manager;
  54     private RoleRequestOutput roleRequestOutput;
  55     private BarrierOutput barrierOutput;
  56     private BigInteger generationId = BigInteger.TEN;
  57
  58     /**
  59      * prepare values
  60      */
  61     @Before
  62     public void setUp() {
  63         Mockito.when(session.getFeatures()).thenReturn(features);
  64         Mockito.when(features.getVersion()).thenReturn(Short.valueOf((short) 42));
  65         Mockito.when(session.getNextXid()).thenReturn(84L);
  66         Mockito.when(session.getPrimaryConductor()).thenReturn(primaryConductor);
  67         Mockito.when(primaryConductor.getConnectionAdapter()).thenReturn(connectionAdapter);
  68         roleRequestOutput = new RoleRequestOutputBuilder()
  69                 .setGenerationId(generationId)
  70                 .setRole(ControllerRole.OFPCRROLESLAVE)
  71                 .setVersion((short) 42)
  72                 .setXid(21L)
  73                 .build();
  74         barrierOutput = new BarrierOutputBuilder()
  75                 .setVersion((short) 42)
  76                 .setXid(1L)
  77                 .build();
  78
  79         manager = new OFRoleManager(sessionManager);
  80     }
  81
  82     /**
  83      * 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)}.
  84      * <br/>
  85      * session lot is empty is invalid
  86      */
  87     @Test
  88     public void testManageRoleChangeFail1() {
  89         manager.manageRoleChange(OfpRole.BECOMESLAVE);
  90         Mockito.verify(connectionAdapter, Mockito.never()).roleRequest(Matchers.any(RoleRequestInput.class));
  91     }
  92
  93     /**
  94      * 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)}.
  95      * <br/>
  96      * available session is invalid
  97      */
  98     @Test
  99     public void testManageRoleChangeFail2() {
 100         Mockito.when(sessionManager.getAllSessions()).thenReturn(Collections.singleton(session));
 101         manager.manageRoleChange(OfpRole.BECOMESLAVE);
 102         Mockito.verify(connectionAdapter, Mockito.never()).roleRequest(Matchers.any(RoleRequestInput.class));
 103     }
 104
 105     /**
 106      * 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)}.
 107      * <br/>
 108      * device bound to valid session is not answering
 109      */
 110     @Test
 111     public void testManageRoleChangeFail3() {
 112         Mockito.when(session.isValid()).thenReturn(true);
 113         Mockito.when(sessionManager.getAllSessions()).thenReturn(Collections.singleton(session));
 114 //        manager.manageRoleChange(OfpRole.BECOMESLAVE);
 115 //        Mockito.verify(connectionAdapter, Mockito.times(1)).roleRequest(Matchers.any(RoleRequestInput.class));
 116     }
 117
 118     /**
 119      * 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)}.
 120      */
 121     @Test
 122     public void testManageRoleChangeOK() {
 123         Mockito.when(session.isValid()).thenReturn(true);
 124         Mockito.when(sessionManager.getAllSessions()).thenReturn(Collections.singleton(session));
 125         Mockito.when(connectionAdapter.roleRequest(Matchers.any(RoleRequestInput.class)))
 126                 .thenReturn(Futures.immediateFuture(RpcResultBuilder.success(roleRequestOutput).build()));
 127         Mockito.when(connectionAdapter.barrier(Matchers.any(BarrierInput.class)))
 128                 .thenReturn(Futures.immediateFuture(RpcResultBuilder.success(barrierOutput).build()));
 129
 130         //manager.manageRoleChange(OfpRole.BECOMESLAVE);
 131
 132         ArgumentCaptor<RoleRequestInput> roleRequestCaptor = ArgumentCaptor.forClass(RoleRequestInput.class);
 133         //Mockito.verify(connectionAdapter, Mockito.times(2)).roleRequest(roleRequestCaptor.capture());
 134
 135 //        List<RoleRequestInput> values = roleRequestCaptor.getAllValues();
 136 //        Assert.assertEquals(ControllerRole.OFPCRROLENOCHANGE, values.get(0).getRole());
 137 //        Assert.assertEquals(0L, values.get(0).getGenerationId().longValue());
 138 //        Assert.assertEquals(ControllerRole.OFPCRROLESLAVE, values.get(1).getRole());
 139 //        Assert.assertEquals(11L, values.get(1).getGenerationId().longValue());
 140     }
 141 }