impl/src/test/java/org/opendaylight/unimgr/command/EvcDeleteCommandTest.java

   1 /*
   2  * Copyright (c) 2016 CableLabs 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.unimgr.command;
   9
  10 import static org.mockito.Matchers.any;
  11 import static org.mockito.Mockito.mock;
  12 import static org.mockito.Mockito.times;
  13 import static org.mockito.Mockito.when;
  14
  15 import java.util.ArrayList;
  16 import java.util.HashSet;
  17 import java.util.List;
  18 import java.util.Set;
  19
  20 import org.junit.Before;
  21 import org.junit.Test;
  22 import org.junit.runner.RunWith;
  23 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  24 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  25 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  26 import org.opendaylight.unimgr.impl.UnimgrMapper;
  27 import org.opendaylight.unimgr.utils.EvcUtils;
  28 import org.opendaylight.unimgr.utils.MdsalUtils;
  29 import org.opendaylight.unimgr.utils.OvsdbUtils;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.EvcAugmentation;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.evc.UniDest;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.unimgr.rev151012.evc.UniSource;
  33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  34 import org.opendaylight.yangtools.yang.binding.DataObject;
  35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  36 import org.powermock.api.mockito.PowerMockito;
  37 import org.powermock.core.classloader.annotations.PrepareForTest;
  38 import org.powermock.modules.junit4.PowerMockRunner;
  39
  40 import com.google.common.base.Optional;
  41
  42 @RunWith(PowerMockRunner.class)
  43 @PrepareForTest({EvcUtils.class, MdsalUtils.class, OvsdbUtils.class, UnimgrMapper.class})
  44 public class EvcDeleteCommandTest {
  45
  46     private EvcDeleteCommand evcDeleteCommand;
  47     private AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changes;
  48     private DataBroker dataBroker;
  49
  50     @SuppressWarnings("unchecked")
  51     @Before
  52     public void setUp(){
  53         PowerMockito.mockStatic(MdsalUtils.class);
  54         PowerMockito.mockStatic(EvcUtils.class);
  55         PowerMockito.mockStatic(OvsdbUtils.class);
  56         PowerMockito.mockStatic(UnimgrMapper.class);
  57         changes = mock(AsyncDataChangeEvent.class);
  58         dataBroker = mock(DataBroker.class);
  59         evcDeleteCommand = new EvcDeleteCommand(dataBroker, changes);
  60     }
  61     /**
  62      * Test method for {@link org.opendaylight.unimgr.command.EvcDeleteCommand#EvcDeleteCommand(org.opendaylight.controller.md.sal.binding.api.DataBroker, org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent)}.
  63      * @throws Exception
  64      */
  65     @SuppressWarnings({ "unchecked", "rawtypes" })
  66     @Test
  67     public void testEvcDeleteCommand() throws Exception {
  68         final Set<InstanceIdentifier<EvcAugmentation>> removedEvcs =
  69                 new HashSet<InstanceIdentifier<EvcAugmentation>>();
  70         final List<UniSource> unisSource = new ArrayList<UniSource>();
  71         final UniSource uniSource = mock(UniSource.class);
  72         final List<UniDest> unisDest = new ArrayList<UniDest>();
  73         final UniDest uniDest = mock(UniDest.class);
  74         final EvcAugmentation evcAugmentation = mock(EvcAugmentation.class);
  75         final Optional<Node> optionalNode = mock(Optional.class);
  76         final InstanceIdentifier instanceOfNode = mock(InstanceIdentifier.class);
  77         removedEvcs.add(mock(InstanceIdentifier.class));
  78         unisSource.add(uniSource);
  79         unisDest.add(uniDest);
  80
  81         when(evcAugmentation.getUniSource()).thenReturn(unisSource);
  82         when(evcAugmentation.getUniDest()).thenReturn(unisDest);
  83         when(uniSource.getUni()).thenReturn(instanceOfNode);
  84         when(uniDest.getUni()).thenReturn(instanceOfNode);
  85         when(OvsdbUtils.extractRemoved(any(AsyncDataChangeEvent.class), any(Class.class)))
  86                 .thenReturn(removedEvcs);
  87         when(MdsalUtils.read(any(DataBroker.class), any(LogicalDatastoreType.class),
  88                 any(InstanceIdentifier.class))).thenReturn(evcAugmentation);
  89         when(MdsalUtils.readNode(any(DataBroker.class), any(LogicalDatastoreType.class),
  90                 any(InstanceIdentifier.class))).thenReturn(optionalNode);
  91         PowerMockito.doNothing().when(EvcUtils.class, "deleteEvcData",
  92                 any(DataBroker.class), any(Optional.class));
  93         when(MdsalUtils.deleteNode(any(DataBroker.class), any(InstanceIdentifier.class),
  94                 any(LogicalDatastoreType.class))).thenReturn(true);
  95
  96         evcDeleteCommand.execute();
  97         PowerMockito.verifyStatic(times(2));
  98         EvcUtils.deleteEvcData(any(DataBroker.class), any(Optional.class));
  99         PowerMockito.verifyStatic(times(1));
 100         MdsalUtils.deleteNode(any(DataBroker.class), any(InstanceIdentifier.class),
 101                 any(LogicalDatastoreType.class));
 102     }
 103
 104 }