renderers/iovisor/src/test/java/org/opendaylight/groupbasedpolicy/renderer/iovisor/endpoint/IovisorEndpointAugCovrgTest.java

   1 package org.opendaylight.groupbasedpolicy.renderer.iovisor.endpoint;\r
   2 \r
   3 import static org.junit.Assert.assertNotNull;\r
   4 import static org.junit.Assert.assertNull;\r
   5 import static org.mockito.Matchers.any;\r
   6 import static org.mockito.Mockito.mock;\r
   7 import static org.mockito.Mockito.verify;\r
   8 import static org.mockito.Mockito.when;\r
   9 \r
  10 import org.junit.Before;\r
  11 import org.junit.Test;\r
  12 import org.opendaylight.groupbasedpolicy.api.EpRendererAugmentationRegistry;\r
  13 import org.opendaylight.groupbasedpolicy.renderer.iovisor.test.GbpIovisorDataBrokerTest;\r
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.endpoint.rev140421.RegisterEndpointInput;\r
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.iovisor.rev151030.IovisorModuleAugmentationInput;\r
  16 \r
  17 public class IovisorEndpointAugCovrgTest extends GbpIovisorDataBrokerTest {\r
  18 \r
  19     private EpRendererAugmentationRegistry epRAR;\r
  20     private IovisorEndpointAug aug;\r
  21 \r
  22     @Before\r
  23     public void init() {\r
  24         epRAR = mock(EpRendererAugmentationRegistry.class);\r
  25 \r
  26         aug = new IovisorEndpointAug(epRAR);\r
  27     }\r
  28 \r
  29     @Test\r
  30     public void testClose() throws Exception {\r
  31         IovisorEndpointAug other = new IovisorEndpointAug(epRAR);\r
  32         other.close();\r
  33 \r
  34         verify(epRAR).unregister(any(IovisorEndpointAug.class));\r
  35     }\r
  36 \r
  37     @Test\r
  38     public void testBuildEndpointAugmentation() {\r
  39         assertNull(aug.buildEndpointAugmentation(null));\r
  40     }\r
  41 \r
  42     @Test\r
  43     public void testBuildEndpointL3Augmentation() {\r
  44         RegisterEndpointInput input = mock(RegisterEndpointInput.class);\r
  45         IovisorModuleAugmentationInput iomAugInput = mock(IovisorModuleAugmentationInput.class);\r
  46         when(input.getAugmentation(IovisorModuleAugmentationInput.class)).thenReturn(iomAugInput);\r
  47         assertNotNull(aug.buildEndpointL3Augmentation(input));\r
  48     }\r
  49 \r
  50     @Test\r
  51     public void testBuildEndpointL3Augmentation_Null() {\r
  52         RegisterEndpointInput input = mock(RegisterEndpointInput.class);\r
  53         when(input.getAugmentation(IovisorModuleAugmentationInput.class)).thenReturn(null);\r
  54         assertNull(aug.buildEndpointL3Augmentation(input));\r
  55     }\r
  56 \r
  57     @Test\r
  58     public void testBuildL3PrefixEndpointAugmentation() {\r
  59         assertNull(aug.buildL3PrefixEndpointAugmentation(null));\r
  60     }\r
  61 \r
  62 }\r