--- /dev/null
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.groupbasedpolicy.gbp_ise_adapter.impl;
+
+import static org.powermock.api.support.membermodification.MemberMatcher.method;
+import static org.powermock.api.support.membermodification.MemberModifier.stub;
+
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
+import com.sun.jersey.api.client.Client;
+import com.sun.jersey.api.client.ClientResponse;
+import com.sun.jersey.api.client.WebResource;
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStream;
+import java.io.InputStreamReader;
+import java.util.Collections;
+import java.util.List;
+import java.util.concurrent.TimeUnit;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InOrder;
+import org.mockito.Matchers;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.opendaylight.groupbasedpolicy.gbp_ise_adapter.impl.util.RestClientFactory;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.gbp.ise.adapter.model.rev160630.gbp.ise.adapter.IseHarvestConfig;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.gbp.ise.adapter.model.rev160630.gbp.ise.adapter.IseHarvestConfigBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.gbp.ise.adapter.model.rev160630.gbp.ise.adapter.ise.harvest.config.ConnectionConfigBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.config.groupbasedpolicy.gbp.ise.adapter.model.rev160630.gbp.ise.adapter.ise.harvest.config.connection.config.HeaderBuilder;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * Test for {@link GbpIseSgtHarvesterImpl}.
+ */
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({RestClientFactory.class})
+public class GbpIseSgtHarvesterImplTest {
+
+ private static final Logger LOG = LoggerFactory.getLogger(GbpIseSgtHarvesterImplTest.class);
+
+ public static final TenantId TENANT_ID = new TenantId("unit-tenant-id-1");
+ public static final Uri ISE_REST_URL = new Uri("https://example.org:9060");
+ public final String iseReplyAllSgts;
+ public final String iseReplySgtDetail;
+
+ @Mock
+ private SgtInfoProcessor processor;
+ @Mock
+ private Client client;
+ @Mock
+ private WebResource webResource;
+ @Mock
+ private WebResource.Builder builder;
+ @Mock
+ private ClientResponse response;
+
+ private IseHarvestConfig config;
+
+ private GbpIseSgtHarvesterImpl harvester;
+
+ public GbpIseSgtHarvesterImplTest() throws IOException {
+ iseReplyAllSgts = readLocalResource("./rawIse-allSgts.xml");
+ iseReplySgtDetail = readLocalResource("./rawIse-sgtDetail.xml");
+ }
+
+ private static String readLocalResource(final String resourcePath) throws IOException {
+ final StringBuilder collector = new StringBuilder();
+ try (
+ final InputStream iseReplySource = GbpIseSgtHarvesterImplTest.class.getResourceAsStream(resourcePath);
+ final BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(iseReplySource))
+ ) {
+ String line;
+ while ((line = bufferedReader.readLine()) != null) {
+ collector.append(line);
+ }
+ }
+ return collector.toString();
+ }
+
+ @Before
+ public void setUp() throws Exception {
+ config = new IseHarvestConfigBuilder()
+ .setTenant(TENANT_ID)
+ .setConnectionConfig(new ConnectionConfigBuilder()
+ .setConnectionTimeout(10)
+ .setReadTimeout(20)
+ .setHeader(Collections.singletonList(new HeaderBuilder()
+ .setName("hName")
+ .setValue("hValue")
+ .build()))
+ .setIseRestUrl(ISE_REST_URL)
+ .build())
+ .build();
+
+ harvester = new GbpIseSgtHarvesterImpl(processor);
+ }
+
+ @Test
+ public void testHarvest() throws Exception {
+ Mockito.when(response.getEntity(String.class)).thenReturn(iseReplyAllSgts, iseReplySgtDetail);
+ Mockito.when(builder.get(Matchers.<Class<ClientResponse>>any())).thenReturn(response);
+ Mockito.when(webResource.getRequestBuilder()).thenReturn(builder);
+ Mockito.when(webResource.path(Matchers.anyString())).thenReturn(webResource);
+ Mockito.when(client.resource(Matchers.<String>any())).thenReturn(webResource);
+ stub(method(RestClientFactory.class, "createIseClient")).toReturn(client);
+
+ Mockito.when(processor.processSgtInfo(Matchers.eq(TENANT_ID), Matchers.<List<SgtInfo>>any())).thenReturn(
+ Futures.immediateCheckedFuture(null));
+
+ final ListenableFuture<Integer> harvestResult = harvester.harvest(config);
+
+ final InOrder inOrder = Mockito.inOrder(client, webResource, builder);
+ inOrder.verify(client).resource(ISE_REST_URL.getValue());
+ // all sgts
+ inOrder.verify(webResource).path("/ers/config/sgt");
+ inOrder.verify(webResource).getRequestBuilder();
+ inOrder.verify(builder).header(Matchers.anyString(),Matchers.anyString());
+ inOrder.verify(builder).get(ClientResponse.class);
+ // sgt detail
+ inOrder.verify(webResource).path("/ers/config/sgt/abc123");
+ inOrder.verify(webResource).getRequestBuilder();
+ inOrder.verify(builder).header(Matchers.anyString(),Matchers.anyString());
+ inOrder.verify(builder).get(ClientResponse.class);
+ inOrder.verifyNoMoreInteractions();
+
+ final Integer count = harvestResult.get(2, TimeUnit.SECONDS);
+ Assert.assertEquals(1, count.intValue());
+ }
+}
\ No newline at end of file
Code Review / groupbasedpolicy.git / blobdiff