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