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