/* * Copyright (c) 2014 Cisco Systems, Inc. and others. All rights reserved. * * This program and the accompanying materials are made available under the * terms of the Eclipse Public License v1.0 which accompanies this distribution, * and is available at http://www.eclipse.org/legal/epl-v10.html */ package test.mock; import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertTrue; import org.junit.After; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mock; import org.mockito.junit.MockitoJUnitRunner; import org.opendaylight.mdsal.singleton.common.api.ClusterSingletonServiceProvider; import org.opendaylight.openflowplugin.api.openflow.FlowGroupCacheManager; import org.opendaylight.openflowplugin.api.openflow.mastership.MastershipChangeServiceManager; import org.opendaylight.openflowplugin.applications.frm.impl.ForwardingRulesManagerImpl; import org.opendaylight.openflowplugin.applications.frm.recovery.OpenflowServiceRecoveryHandler; import org.opendaylight.openflowplugin.applications.reconciliation.ReconciliationManager; import org.opendaylight.serviceutils.srm.ServiceRecoveryRegistry; import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode; import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId; import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes; import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node; import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey; import org.opendaylight.yangtools.yang.binding.InstanceIdentifier; import test.mock.util.FRMTest; import test.mock.util.RpcProviderRegistryMock; @RunWith(MockitoJUnitRunner.class) public class NodeListenerTest extends FRMTest { private ForwardingRulesManagerImpl forwardingRulesManager; private static final NodeKey NODE_KEY = new NodeKey(new NodeId("testnode:1")); RpcProviderRegistryMock rpcProviderRegistryMock = new RpcProviderRegistryMock(); @Mock ClusterSingletonServiceProvider clusterSingletonService; @Mock private ReconciliationManager reconciliationManager; @Mock private OpenflowServiceRecoveryHandler openflowServiceRecoveryHandler; @Mock private ServiceRecoveryRegistry serviceRecoveryRegistry; @Mock private MastershipChangeServiceManager mastershipChangeServiceManager; @Mock private FlowGroupCacheManager flowGroupCacheManager; @Before public void setUp() { forwardingRulesManager = new ForwardingRulesManagerImpl( getDataBroker(), rpcProviderRegistryMock, rpcProviderRegistryMock, getConfig(), mastershipChangeServiceManager, clusterSingletonService, getConfigurationService(), reconciliationManager, openflowServiceRecoveryHandler, serviceRecoveryRegistry, flowGroupCacheManager, getRegistrationHelper()); forwardingRulesManager.start(); } @Test public void addRemoveNodeTest() throws Exception { addFlowCapableNode(NODE_KEY); InstanceIdentifier nodeII = InstanceIdentifier.create(Nodes.class).child(Node.class, NODE_KEY) .augmentation(FlowCapableNode.class); boolean nodeActive = forwardingRulesManager.isNodeActive(nodeII); assertTrue(nodeActive); removeNode(NODE_KEY); nodeActive = forwardingRulesManager.isNodeActive(nodeII); assertFalse(nodeActive); } @After public void tearDown() throws Exception { forwardingRulesManager.close(); } }