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

   1 package org.opendaylight.openflowplugin.impl.services;
   2
   3 import junit.framework.TestCase;
   4 import org.junit.Test;
   5 import org.mockito.Mock;
   6 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
   7 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
   8 import org.opendaylight.openflowplugin.api.openflow.registry.meter.DeviceMeterRegistry;
   9 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.*;
  10 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.service.rev130918.meter.update.UpdatedMeter;
  11 import org.opendaylight.yang.gen.v1.urn.opendaylight.meter.types.rev130918.MeterId;
  12
  13 import static org.mockito.Matchers.any;
  14 import static org.mockito.Matchers.eq;
  15 import static org.mockito.Mockito.mock;
  16 import static org.mockito.Mockito.verify;
  17 import static org.mockito.Mockito.when;
  18
  19 public class SalMeterServiceImplTest extends ServiceMocking {
  20
  21     private static final Long DUMMY_METER_ID = 15L;
  22     public SalMeterService salMeterService;
  23
  24     @Mock
  25     DeviceMeterRegistry mockedDeviceMeterRegistry;
  26
  27     @Override
  28     public void initialization() {
  29         super.initialization();
  30         salMeterService = new SalMeterServiceImpl(mockedRequestContextStack, mockedDeviceContext);
  31         when(mockedDeviceContext.getDeviceMeterRegistry()).thenReturn(mockedDeviceMeterRegistry);
  32     }
  33
  34     @Test
  35     public void testAddMeter() throws Exception {
  36         final MeterId dummyMeterId = new MeterId(DUMMY_METER_ID);
  37         AddMeterInput addMeterInput = new AddMeterInputBuilder().setMeterId(dummyMeterId).build();
  38
  39         salMeterService.addMeter(addMeterInput);
  40         verify(mockedRequestContextStack).createRequestContext();
  41         verify(mockedDeviceMeterRegistry).store(eq(dummyMeterId));
  42     }
  43
  44     @Test
  45     public void testUpdateMeter() throws Exception {
  46         UpdatedMeter mockedUptatedMeter = mock(UpdatedMeter.class);
  47         final UpdateMeterInput updateMeterInput = new UpdateMeterInputBuilder().setUpdatedMeter(mockedUptatedMeter).build();
  48         salMeterService.updateMeter(updateMeterInput);
  49         verify(mockedRequestContextStack).createRequestContext();
  50     }
  51
  52     @Test
  53     public void testRemoveMeter() throws Exception {
  54         final MeterId dummyMeterId = new MeterId(DUMMY_METER_ID);
  55         RemoveMeterInput removeMeterInput = new RemoveMeterInputBuilder().setMeterId(dummyMeterId).build();
  56
  57         salMeterService.removeMeter(removeMeterInput);
  58         verify(mockedRequestContextStack).createRequestContext();
  59         verify(mockedDeviceMeterRegistry).markToBeremoved(eq(dummyMeterId));
  60     }
  61 }