sxp-integration/sxp-ise-adapter/src/test/java/org/opendaylight/groupbasedpolicy/sxp_ise_adapter/impl/EPPolicyTemplateProviderIseImplTest.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_ise_adapter.impl;
  10
  11 import static org.powermock.api.support.membermodification.MemberMatcher.method;
  12 import static org.powermock.api.support.membermodification.MemberModifier.stub;
  13
  14 import com.google.common.util.concurrent.Futures;
  15 import com.google.common.util.concurrent.ListenableFuture;
  16 import com.sun.jersey.api.client.Client;
  17 import com.sun.jersey.api.client.WebResource;
  18 import java.util.Collections;
  19 import java.util.List;
  20 import java.util.Optional;
  21 import javax.xml.xpath.XPath;
  22 import javax.xml.xpath.XPathConstants;
  23 import org.junit.Assert;
  24 import org.junit.Before;
  25 import org.junit.Test;
  26 import org.junit.runner.RunWith;
  27 import org.mockito.Matchers;
  28 import org.mockito.Mock;
  29 import org.mockito.Mockito;
  30 import org.opendaylight.groupbasedpolicy.sxp_ise_adapter.impl.util.IseReplyUtil;
  31 import org.opendaylight.groupbasedpolicy.sxp_ise_adapter.impl.util.RestClientFactory;
  32 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.TemplateGenerated;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ep.provider.model.rev160302.sxp.ep.mapper.EndpointPolicyTemplateBySgt;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ise.adapter.model.rev160630.gbp.sxp.ise.adapter.IseSourceConfig;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ise.adapter.model.rev160630.gbp.sxp.ise.adapter.IseSourceConfigBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ise.adapter.model.rev160630.gbp.sxp.ise.adapter.ise.source.config.ConnectionConfig;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ise.adapter.model.rev160630.gbp.sxp.ise.adapter.ise.source.config.ConnectionConfigBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.sxp.integration.sxp.ise.adapter.model.rev160630.gbp.sxp.ise.adapter.ise.source.config.connection.config.HeaderBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.sxp.database.rev160308.Sgt;
  42 import org.powermock.core.classloader.annotations.PrepareForTest;
  43 import org.powermock.modules.junit4.PowerMockRunner;
  44 import org.w3c.dom.Node;
  45 import org.xml.sax.InputSource;
  46
  47 /**
  48  * Test for {@link EPPolicyTemplateProviderIseImpl}.
  49  */
  50 @RunWith(PowerMockRunner.class)
  51 @PrepareForTest({IseReplyUtil.class, RestClientFactory.class})
  52 public class EPPolicyTemplateProviderIseImplTest {
  53
  54     private static final Sgt SGT = new Sgt(42);
  55     private static final Sgt SGT_LOW = new Sgt(1);
  56     private static final Sgt SGT_HI = new Sgt(100);
  57     private static final TenantId TENANT_ID = new TenantId("tenant-01");
  58     private static final SgtInfo SGT_INFO = new SgtInfo(SGT, "ultimate_group", "uuidOf42");
  59
  60     @Mock
  61     private XPath xpathWalker;
  62     @Mock
  63     private Client iseClient;
  64     @Mock
  65     private WebResource.Builder wrBuilder;
  66     @Mock
  67     private GbpIseSgtHarvester iseSgtHarvester;
  68     @Mock
  69     private IseContext iseContext;
  70
  71     private EPPolicyTemplateProviderIseImpl templateProvider;
  72
  73     @Before
  74     public void setUp() throws Exception {
  75         templateProvider = new EPPolicyTemplateProviderIseImpl();
  76         templateProvider.setIseSgtHarvester(iseSgtHarvester);
  77     }
  78
  79     @Test
  80     public void testProvideTemplate_noConfig() throws Exception {
  81         final ListenableFuture<Optional<EndpointPolicyTemplateBySgt>> endpointPolicyTemplateBySgt =
  82                 templateProvider.provideTemplate(SGT);
  83         Assert.assertTrue(endpointPolicyTemplateBySgt.isDone());
  84         Assert.assertFalse(endpointPolicyTemplateBySgt.get().isPresent());
  85     }
  86
  87     @Test
  88     public void testProvideTemplate_config() throws Exception {
  89         final ConnectionConfig connectionConfig = new ConnectionConfigBuilder()
  90                 .setIseRestUrl(new Uri("http://example.org"))
  91                 .setConnectionTimeout(10)
  92                 .setReadTimeout(10)
  93                 .setHeader(Collections.singletonList(new HeaderBuilder()
  94                         .setName("headerName")
  95                         .setValue("headerValue")
  96                         .build()))
  97                 .build();
  98         final IseSourceConfig sourceConfig = new IseSourceConfigBuilder()
  99                 .setTenant(TENANT_ID)
 100                 .setConnectionConfig(connectionConfig)
 101                 .setSgtRangeMin(SGT_LOW)
 102                 .setSgtRangeMax(SGT_HI)
 103                 .build();
 104
 105         final String rawResponse = "";
 106         stub(method(IseReplyUtil.class, "deliverResponse")).toReturn(rawResponse);
 107         stub(method(IseReplyUtil.class, "setupXpath")).toReturn(xpathWalker);
 108         stub(method(RestClientFactory.class, "createIseClient", ConnectionConfig.class)).toReturn(iseClient);
 109         stub(method(RestClientFactory.class, "createRequestBuilder", WebResource.class, List.class, String.class)).toReturn(wrBuilder);
 110
 111
 112         final String epgName = SGT_INFO.getName();
 113         final Node sgtNameNode = Mockito.mock(Node.class);
 114         Mockito.when(sgtNameNode.getNodeValue()).thenReturn(epgName);
 115         // matching value of IseReplyUtil.EXPRESSION_SGT_NAME_ATTR
 116         Mockito.when(xpathWalker.evaluate(Matchers.same("./@name"),
 117                 Matchers.<InputSource>any(), Matchers.same(XPathConstants.NODE)))
 118                 .thenReturn(sgtNameNode);
 119
 120         Mockito.when(iseContext.getIseSourceConfig()).thenReturn(sourceConfig);
 121         Mockito.when(iseSgtHarvester.harvestAll(iseContext))
 122                 .thenReturn(Futures.immediateFuture(Collections.singletonList(SGT_INFO)));
 123
 124         templateProvider.assignIseContext(iseContext);
 125         final ListenableFuture<Optional<EndpointPolicyTemplateBySgt>> templateWrap = templateProvider.provideTemplate(SGT);
 126
 127         Assert.assertTrue(templateWrap.isDone());
 128         Assert.assertTrue(templateWrap.get().isPresent());
 129         final EndpointPolicyTemplateBySgt template = templateWrap.get().get();
 130         Assert.assertEquals(SGT, template.getSgt());
 131         Assert.assertNull(template.getConditions());
 132         Assert.assertEquals(TENANT_ID, template.getTenant());
 133         Assert.assertEquals(1, template.getEndpointGroups().size());
 134         Assert.assertEquals(epgName, template.getEndpointGroups().get(0).getValue());
 135         Assert.assertEquals(TemplateGenerated.class, template.getOrigin());
 136     }
 137 }