2 * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.groupbasedpolicy.sxp.mapper.impl;
11 import java.util.ArrayList;
12 import java.util.Collections;
13 import java.util.List;
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;
50 * Test for {@link SxpEndpointAugmentorImpl}.
52 @RunWith(MockitoJUnitRunner.class)
53 public class SxpEndpointAugmentorImplTest {
56 private ReadableByKey<EpPolicyTemplateValueKey, EndpointPolicyTemplateBySgt> epPolicyDao;
58 private EpPolicyTemplateValueKeyFactory keyFactory = new EpPolicyTemplateValueKeyFactory(
59 EPTemplateUtil.createEndpointGroupIdOrdering(), EPTemplateUtil.createConditionNameOrdering());
61 private ArgumentCaptor<EpPolicyTemplateValueKey> keyCapt;
63 private SxpEndpointAugmentorImpl augmetor;
66 public void setUp() throws Exception {
67 augmetor = new SxpEndpointAugmentorImpl(epPolicyDao, keyFactory);
71 public void testBuildAddressEndpointWithLocationAugmentation() throws Exception {
72 final TenantId tenantId = new TenantId("tn1");
73 final AddressEndpoint endpoint = new AddressEndpointBuilder()
75 .setCondition(buildConditions(new String[]{"cn2", "cn1"}))
76 .setEndpointGroup(buildEndpointGroupIds(new String[]{"epg2", "epg1"}))
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()
84 .setEndpointGroups(endpointGroupIds)
85 .setConditions(conditions)
89 Mockito.when(epPolicyDao.readBy(keyCapt.capture())).thenReturn(Collections.singletonList(epPolicyTemplate));
91 final Map.Entry<Class<? extends Augmentation<AddressEndpointWithLocation>>, Augmentation<AddressEndpointWithLocation>>
92 augmentationEntry = augmetor.buildAddressEndpointWithLocationAugmentation(endpoint);
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());
99 final EpPolicyTemplateValueKey keyValue = keyCapt.getValue();
100 Assert.assertEquals(tenantId, keyValue.getTenantId());
101 Assert.assertEquals(endpointGroupIds, keyValue.getEpgId());
102 Assert.assertEquals(conditions, keyValue.getConditionName());
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));
110 return endpointGroupIds;
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));
122 public void testBuildAddressEndpointAugmentation() throws Exception {
123 final AddressEndpointReg endpoint = new AddressEndpointRegBuilder().build();
124 Assert.assertNull(augmetor.buildAddressEndpointAugmentation(endpoint));
128 public void testBuildContainmentEndpointAugmentation() throws Exception {
129 final ContainmentEndpointReg endpoint = new ContainmentEndpointRegBuilder().build();
130 Assert.assertNull(augmetor.buildContainmentEndpointAugmentation(endpoint));
134 public void testBuildContainmentEndpointWithLocationAugmentation() throws Exception {
135 final ContainmentEndpoint endpoint = new ContainmentEndpointBuilder().build();
136 Assert.assertNull(augmetor.buildContainmentEndpointWithLocationAugmentation(endpoint));