openflowplugin-impl/src/test/java/org/opendaylight/openflowplugin/impl/services/SalGroupServiceImplTest.java

   1 package org.opendaylight.openflowplugin.impl.services;
   2
   3 import static org.mockito.Matchers.eq;
   4 import static org.mockito.Mockito.mock;
   5 import static org.mockito.Mockito.verify;
   6 import static org.mockito.Mockito.when;
   7 import org.junit.Test;
   8 import org.mockito.Matchers;
   9 import org.mockito.Mock;
  10 import org.opendaylight.openflowplugin.api.openflow.registry.group.DeviceGroupRegistry;
  11 import org.opendaylight.openflowplugin.api.openflow.rpc.listener.ItemLifecycleListener;
  12 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.AddGroupInput;
  13 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.AddGroupInputBuilder;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.AddGroupOutput;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.RemoveGroupInput;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.RemoveGroupInputBuilder;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.RemoveGroupOutput;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.UpdateGroupInput;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.UpdateGroupInputBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.UpdateGroupOutput;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.group.update.OriginalGroup;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.group.update.OriginalGroupBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.group.update.UpdatedGroup;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.group.update.UpdatedGroupBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.Group;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.GroupKey;
  28 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
  29
  30 public class SalGroupServiceImplTest extends ServiceMocking {
  31
  32     private static final Long DUMMY_GROUP_ID = 15L;
  33
  34     @Mock
  35     DeviceGroupRegistry mockedDeviceGroupRegistry;
  36
  37     SalGroupServiceImpl salGroupService;
  38
  39     @Override
  40     public void initialization() {
  41         super.initialization();
  42         when(mockedDeviceContext.getDeviceGroupRegistry()).thenReturn(mockedDeviceGroupRegistry);
  43         salGroupService = new SalGroupServiceImpl(mockedRequestContextStack, mockedDeviceContext);
  44     }
  45
  46     @Test
  47     public void testAddGroup() throws Exception {
  48         addGroup(null);
  49     }
  50
  51     @Test
  52     public void testAddGroupWithItemLifecycle() throws Exception {
  53         addGroup(mock(ItemLifecycleListener.class));
  54     }
  55
  56     private void addGroup(final ItemLifecycleListener itemLifecycleListener) {
  57         final GroupId dummyGroupId = new GroupId(DUMMY_GROUP_ID);
  58         AddGroupInput addGroupInput = new AddGroupInputBuilder().setGroupId(dummyGroupId).build();
  59
  60         this.<AddGroupOutput>mockSuccessfulFuture();
  61
  62         salGroupService.setItemLifecycleListener(itemLifecycleListener);
  63
  64         salGroupService.addGroup(addGroupInput);
  65         verify(mockedRequestContextStack).createRequestContext();
  66         verify(mockedDeviceGroupRegistry).store(eq(dummyGroupId));
  67
  68         if (itemLifecycleListener != null) {
  69             verify(itemLifecycleListener).onAdded(Matchers.<KeyedInstanceIdentifier<Group, GroupKey>>any(),Matchers.<Group>any());
  70         }
  71     }
  72
  73     @Test
  74     public void testUpdateGroup() throws Exception {
  75         updateGroup(null);
  76     }
  77
  78     @Test
  79     public void testUpdateGroupWithItemLifecycle() throws Exception {
  80         updateGroup(mock(ItemLifecycleListener.class));
  81     }
  82
  83     private void updateGroup(final ItemLifecycleListener itemLifecycleListener) {
  84         final UpdatedGroup updatedGroup = new UpdatedGroupBuilder().setGroupId(new GroupId(DUMMY_GROUP_ID)).build();
  85         final OriginalGroup originalGroup = new OriginalGroupBuilder().setGroupId(new GroupId(DUMMY_GROUP_ID)).build();
  86         final UpdateGroupInput updateGroupInput = new UpdateGroupInputBuilder().setUpdatedGroup(updatedGroup).setOriginalGroup(originalGroup).build();
  87
  88         this.<UpdateGroupOutput>mockSuccessfulFuture();
  89
  90         salGroupService.setItemLifecycleListener(itemLifecycleListener);
  91
  92         salGroupService.updateGroup(updateGroupInput);
  93         verify(mockedRequestContextStack).createRequestContext();
  94
  95         if (itemLifecycleListener != null) {
  96             verify(itemLifecycleListener).onAdded(Matchers.<KeyedInstanceIdentifier<Group, GroupKey>>any(),Matchers.<Group>any());
  97             verify(itemLifecycleListener).onRemoved(Matchers.<KeyedInstanceIdentifier<Group, GroupKey>>any());
  98         }
  99     }
 100
 101     @Test
 102     public void testRemoveGroup() throws Exception {
 103         removeGroup(null);
 104     }
 105
 106     @Test
 107     public void testRemoveGroupWithItemLifecycle() throws Exception {
 108         removeGroup(mock(ItemLifecycleListener.class));
 109     }
 110
 111     private void removeGroup(final ItemLifecycleListener itemLifecycleListener) throws Exception {
 112         final GroupId dummyGroupId = new GroupId(DUMMY_GROUP_ID);
 113         RemoveGroupInput removeGroupInput = new RemoveGroupInputBuilder().setGroupId(dummyGroupId).build();
 114
 115         this.<RemoveGroupOutput>mockSuccessfulFuture();
 116
 117         salGroupService.setItemLifecycleListener(itemLifecycleListener);
 118
 119         salGroupService.removeGroup(removeGroupInput);
 120         verify(mockedRequestContextStack).createRequestContext();
 121         verify(mockedDeviceGroupRegistry).markToBeremoved(eq(dummyGroupId));
 122
 123         if (itemLifecycleListener != null) {
 124             verify(itemLifecycleListener).onRemoved(Matchers.<KeyedInstanceIdentifier<Group, GroupKey>>any());
 125         }
 126     }
 127 }