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

   1 package org.opendaylight.openflowplugin.impl.services;
   2
   3 import org.junit.Test;
   4 import org.mockito.Mock;
   5 import org.opendaylight.openflowplugin.api.openflow.registry.group.DeviceGroupRegistry;
   6 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.*;
   7 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.group.update.UpdatedGroup;
   8 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId;
   9
  10 import static org.mockito.Matchers.eq;
  11 import static org.mockito.Mockito.mock;
  12 import static org.mockito.Mockito.verify;
  13 import static org.mockito.Mockito.when;
  14
  15 public class SalGroupServiceImplTest extends ServiceMocking {
  16
  17     private static final Long DUMMY_GROUP_ID = 15L;
  18     public SalGroupService salGroupService;
  19
  20     @Mock
  21     DeviceGroupRegistry mockedDeviceGroupRegistry;
  22
  23     @Override
  24     public void initialization() {
  25         super.initialization();
  26         salGroupService = new SalGroupServiceImpl(mockedRequestContextStack, mockedDeviceContext);
  27         when(mockedDeviceContext.getDeviceGroupRegistry()).thenReturn(mockedDeviceGroupRegistry);
  28     }
  29
  30     @Test
  31     public void testAddGroup() throws Exception {
  32         final GroupId dummyGroupId = new GroupId(DUMMY_GROUP_ID);
  33         AddGroupInput addGroupInput = new AddGroupInputBuilder().setGroupId(dummyGroupId).build();
  34
  35         salGroupService.addGroup(addGroupInput);
  36         verify(mockedRequestContextStack).createRequestContext();
  37         verify(mockedDeviceGroupRegistry).store(eq(dummyGroupId));
  38     }
  39
  40     @Test
  41     public void testUpdateGroup() throws Exception {
  42         UpdatedGroup mockedUptatedGroup = mock(UpdatedGroup.class);
  43         final UpdateGroupInput updateGroupInput = new UpdateGroupInputBuilder().setUpdatedGroup(mockedUptatedGroup).build();
  44         salGroupService.updateGroup(updateGroupInput);
  45         verify(mockedRequestContextStack).createRequestContext();
  46     }
  47
  48     @Test
  49     public void testRemoveGroup() throws Exception {
  50         final GroupId dummyGroupId = new GroupId(DUMMY_GROUP_ID);
  51         RemoveGroupInput removeGroupInput = new RemoveGroupInputBuilder().setGroupId(dummyGroupId).build();
  52
  53         salGroupService.removeGroup(removeGroupInput);
  54         verify(mockedRequestContextStack).createRequestContext();
  55         verify(mockedDeviceGroupRegistry).markToBeremoved(eq(dummyGroupId));
  56     }
  57 }