sxp-mapper/src/test/java/org/opendaylight/groupbasedpolicy/sxp/mapper/impl/SxpEndpointAugmentorImplTest.java

   1 /*
   2  * Copyright (c) 2016 Cisco Systems, Inc. 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
   9 package org.opendaylight.groupbasedpolicy.sxp.mapper.impl;
  10
  11 import java.util.ArrayList;
  12 import java.util.Collections;
  13 import java.util.List;
  14 import java.util.Map;
  15 import org.junit.Assert;
  16 import org.junit.Before;
  17 import org.junit.Test;
  18 import org.junit.runner.RunWith;
  19 import org.mockito.ArgumentCaptor;
  20 import org.mockito.Captor;
  21 import org.mockito.Matchers;
  22 import org.mockito.Mock;
  23 import org.mockito.Mockito;
  24 import org.mockito.Spy;
  25 import org.mockito.runners.MockitoJUnitRunner;
  26 import org.opendaylight.groupbasedpolicy.sxp.mapper.api.ReadableByKey;
  27 import org.opendaylight.groupbasedpolicy.sxp.mapper.impl.dao.EpPolicyTemplateValueKey;
  28 import org.opendaylight.groupbasedpolicy.sxp.mapper.impl.dao.EpPolicyTemplateValueKeyFactory;
  29 import org.opendaylight.groupbasedpolicy.sxp.mapper.impl.util.EPTemplateUtil;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.endpoints.address.endpoints.AddressEndpoint;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.endpoints.address.endpoints.AddressEndpointBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.endpoints.containment.endpoints.ContainmentEndpoint;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.endpoints.containment.endpoints.ContainmentEndpointBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.AddressEndpointReg;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.AddressEndpointRegBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.ContainmentEndpointReg;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.base_endpoint.rev160427.register.endpoint.input.ContainmentEndpointRegBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.ConditionName;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.EndpointGroupId;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.renderer.policy.configuration.endpoints.AddressEndpointWithLocation;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.AddressEndpointWithLocationAug;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointPolicyTemplateBySgt;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.mapper.model.rev160302.sxp.mapper.EndpointPolicyTemplateBySgtBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
  46 import org.opendaylight.yangtools.yang.binding.Augmentation;
  47 import org.opendaylight.yangtools.yang.binding.DataObject;
  48
  49 /**
  50  * Test for {@link SxpEndpointAugmentorImpl}.
  51  */
  52 @RunWith(MockitoJUnitRunner.class)
  53 public class SxpEndpointAugmentorImplTest {
  54
  55     @Mock
  56     private ReadableByKey<EpPolicyTemplateValueKey, EndpointPolicyTemplateBySgt> epPolicyDao;
  57     @Spy
  58     private EpPolicyTemplateValueKeyFactory keyFactory = new EpPolicyTemplateValueKeyFactory(
  59             EPTemplateUtil.createEndpointGroupIdOrdering(), EPTemplateUtil.createConditionNameOrdering());
  60     @Captor
  61     private ArgumentCaptor<EpPolicyTemplateValueKey> keyCapt;
  62
  63     private SxpEndpointAugmentorImpl augmetor;
  64
  65     @Before
  66     public void setUp() throws Exception {
  67         augmetor = new SxpEndpointAugmentorImpl(epPolicyDao, keyFactory);
  68     }
  69
  70     @Test
  71     public void testBuildAddressEndpointWithLocationAugmentation() throws Exception {
  72         final TenantId tenantId = new TenantId("tn1");
  73         final AddressEndpoint endpoint = new AddressEndpointBuilder()
  74                 .setTenant(tenantId)
  75                 .setCondition(buildConditions(new String[]{"cn2", "cn1"}))
  76                 .setEndpointGroup(buildEndpointGroupIds(new String[]{"epg2", "epg1"}))
  77                 .build();
  78
  79         Mockito.doCallRealMethod().when(keyFactory).sortValueKeyLists(Matchers.<EndpointPolicyTemplateBySgt>any());
  80         final List<ConditionName> conditions = buildConditions(new String[]{"cn1", "cn2"});
  81         final List<EndpointGroupId> endpointGroupIds = buildEndpointGroupIds(new String[]{"epg1", "epg2"});
  82         final EndpointPolicyTemplateBySgt epPolicyTemplate = new EndpointPolicyTemplateBySgtBuilder()
  83                 .setTenant(tenantId)
  84                 .setEndpointGroups(endpointGroupIds)
  85                 .setConditions(conditions)
  86                 .setSgt(new Sgt(42))
  87                 .build();
  88
  89         Mockito.when(epPolicyDao.readBy(keyCapt.capture())).thenReturn(Collections.singletonList(epPolicyTemplate));
  90
  91         final Map.Entry<Class<? extends Augmentation<AddressEndpointWithLocation>>, Augmentation<AddressEndpointWithLocation>>
  92                 augmentationEntry = augmetor.buildAddressEndpointWithLocationAugmentation(endpoint);
  93
  94         Assert.assertEquals(AddressEndpointWithLocationAug.class, augmentationEntry.getKey());
  95         Assert.assertTrue(DataObject.class.isAssignableFrom(augmentationEntry.getValue().getClass()));
  96         Assert.assertEquals(AddressEndpointWithLocationAug.class, ((DataObject) augmentationEntry.getValue()).getImplementedInterface());
  97         Assert.assertEquals(42, ((AddressEndpointWithLocationAug) augmentationEntry.getValue()).getSgt().getValue().intValue());
  98
  99         final EpPolicyTemplateValueKey keyValue = keyCapt.getValue();
 100         Assert.assertEquals(tenantId, keyValue.getTenantId());
 101         Assert.assertEquals(endpointGroupIds, keyValue.getEpgId());
 102         Assert.assertEquals(conditions, keyValue.getConditionName());
 103     }
 104
 105     private static List<EndpointGroupId> buildEndpointGroupIds(final String[] names) {
 106         final List<EndpointGroupId> endpointGroupIds = new ArrayList<>();
 107         for (String epgId : names) {
 108             endpointGroupIds.add(new EndpointGroupId(epgId));
 109         }
 110         return endpointGroupIds;
 111     }
 112
 113     private static List<ConditionName> buildConditions(final String[] names) {
 114         final List<ConditionName> conditions = new ArrayList<>();
 115         for (String condition : names) {
 116             conditions.add(new ConditionName(condition));
 117         }
 118         return conditions;
 119     }
 120
 121     @Test
 122     public void testBuildAddressEndpointAugmentation() throws Exception {
 123         final AddressEndpointReg endpoint = new AddressEndpointRegBuilder().build();
 124         Assert.assertNull(augmetor.buildAddressEndpointAugmentation(endpoint));
 125     }
 126
 127     @Test
 128     public void testBuildContainmentEndpointAugmentation() throws Exception {
 129         final ContainmentEndpointReg endpoint = new ContainmentEndpointRegBuilder().build();
 130         Assert.assertNull(augmetor.buildContainmentEndpointAugmentation(endpoint));
 131     }
 132
 133     @Test
 134     public void testBuildContainmentEndpointWithLocationAugmentation() throws Exception {
 135         final ContainmentEndpoint endpoint = new ContainmentEndpointBuilder().build();
 136         Assert.assertNull(augmetor.buildContainmentEndpointWithLocationAugmentation(endpoint));
 137     }
 138 }