cadi/client/src/test/java/org/onap/aaf/cadi/http/test/JU_HClient.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.cadi.http.test;
  23
  24 import static org.junit.Assert.*;
  25 import static org.mockito.Mockito.*;
  26
  27 import java.io.ByteArrayInputStream;
  28 import java.io.IOException;
  29 import java.io.InputStream;
  30 import java.lang.reflect.Field;
  31 import java.net.HttpURLConnection;
  32 import java.net.URI;
  33 import java.net.URISyntaxException;
  34
  35 import javax.servlet.ServletOutputStream;
  36 import javax.servlet.http.HttpServletResponse;
  37
  38 import static org.hamcrest.CoreMatchers.*;
  39 import org.junit.*;
  40 import org.mockito.*;
  41 import org.onap.aaf.cadi.CadiException;
  42 import org.onap.aaf.cadi.LocatorException;
  43 import org.onap.aaf.cadi.SecuritySetter;
  44 import org.onap.aaf.cadi.client.EClient.Transfer;
  45 import org.onap.aaf.cadi.client.Future;
  46 import org.onap.aaf.cadi.http.HClient;
  47 import org.onap.aaf.cadi.http.HClient.HFuture;
  48 import org.onap.aaf.misc.env.APIException;
  49 import org.onap.aaf.misc.rosetta.env.RosettaDF;
  50 import org.onap.aaf.misc.rosetta.env.RosettaData;
  51
  52 public class JU_HClient {
  53
  54     @Mock private SecuritySetter<HttpURLConnection> ssMock;
  55     @Mock private Transfer transferMock;
  56     @Mock private HttpURLConnection hucMock;
  57     @Mock private HttpServletResponse respMock;
  58     @Mock private RosettaDF<HttpURLConnection> dfMock;
  59     @Mock private RosettaData<HttpURLConnection> dataMock;
  60
  61     private static final String uriString = "http://example.com:8080/path/to/a/file.txt";
  62     private static final String fragment = "fragment";
  63     private static final String method = "method";
  64     private static final String pathinfo = "pathinfo";
  65     private static final String queryParams = "queryParams";
  66
  67     private static final String errorString = "error string";
  68     private static final String successString = "success string";
  69
  70     private static final String tag1 = "tag1";
  71     private static final String tag2 = "tag2";
  72     private static final String value1 = "value1";
  73     private static final String value2 = "value2";
  74
  75     private URI uri;
  76
  77     @Before
  78     public void setup() throws URISyntaxException {
  79         MockitoAnnotations.initMocks(this);
  80
  81         uri = new URI(uriString);
  82     }
  83
  84     @Test
  85     public void accessorsMutatorsTest() throws LocatorException {
  86         HClient client = new HClient(ssMock, uri, 0);
  87         client.setFragment(fragment);
  88         client.setMethod(method);
  89         client.setPathInfo(pathinfo);
  90         client.setPayload(transferMock);
  91         client.setQueryParams(queryParams);
  92         assertThat(client.getURI(), is(uri));
  93         assertThat(client.timeout(), is(0));
  94         assertThat(client.toString(), is("HttpURLConnection Client configured to " + uri.toString()));
  95     }
  96
  97     @Test
  98     public void sendTest() throws LocatorException, APIException, URISyntaxException {
  99         HClientStub client;
 100         client = new HClientStub(ssMock, uri, 0, null);
 101         client.send();
 102
 103         client.setPathInfo("/pathinfo");
 104         client.send();
 105
 106         client.setPathInfo("pathinfo");
 107         client.send();
 108
 109         client = new HClientStub(null, uri, 0, null);
 110         client.send();
 111
 112         client.addHeader(tag1, value1);
 113         client.addHeader(tag2, value2);
 114         client.send();
 115
 116         client.setPayload(transferMock);
 117         client.send();
 118     }
 119
 120     @Test(expected = APIException.class)
 121     public void sendThrows1Test() throws APIException, LocatorException, URISyntaxException {
 122         HClientStub client = new HClientStub(ssMock, new URI("mailto:me@domain.com"), 0, null);
 123         client.send();
 124     }
 125
 126     @Test(expected = APIException.class)
 127     public void sendThrows2Test() throws APIException, LocatorException, URISyntaxException {
 128         HClientStub client = new HClientStub(ssMock, new URI("mailto:me@domain.com"), 0, null);
 129         client.addHeader(tag1, value1);
 130         client.addHeader(tag2, value2);
 131         client.send();
 132     }
 133
 134     @Test
 135     public void futureCreateTest() throws LocatorException, CadiException, IOException {
 136         HClient client = new HClientStub(ssMock, uri, 0, hucMock);
 137         HFuture<HttpURLConnection> future = (HFuture<HttpURLConnection>) client.futureCreate(HttpURLConnection.class);
 138
 139         // Test a bad response code (default 0) without output
 140         assertThat(future.get(0), is(false));
 141         assertThat(future.body().length(), is(0));
 142
 143         // Test a bad response code (default 0) with output
 144         ByteArrayInputStream bais = new ByteArrayInputStream(errorString.getBytes());
 145         when(hucMock.getInputStream()).thenReturn(bais);
 146         assertThat(future.get(0), is(false));
 147         assertThat(future.body(), is(errorString));
 148
 149         // Test a good response code
 150         when(hucMock.getResponseCode()).thenReturn(201);
 151         assertThat(future.get(0), is(true));
 152     }
 153
 154     @Test
 155     public void futureReadStringTest() throws LocatorException, CadiException, IOException {
 156         HClient client = new HClientStub(ssMock, uri, 0, hucMock);
 157         Future<String> future = client.futureReadString();
 158
 159         // Test a bad response code (default 0) without output
 160         assertThat(future.get(0), is(false));
 161         assertThat(future.body().length(), is(0));
 162
 163         // Test a bad response code (default 0) with output
 164         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(errorString.getBytes()));
 165         assertThat(future.get(0), is(false));
 166         assertThat(future.body(), is(errorString));
 167
 168         // Test a good response code
 169         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(successString.getBytes()));
 170         when(hucMock.getResponseCode()).thenReturn(200);
 171         assertThat(future.get(0), is(true));
 172         assertThat(future.body(), is(successString));
 173     }
 174
 175     @Test
 176     public void futureReadTest() throws LocatorException, CadiException, IOException, APIException {
 177         HClient client = new HClientStub(ssMock, uri, 0, hucMock);
 178         Future<HttpURLConnection> future = client.futureRead(dfMock, null);
 179
 180         // Test a bad response code (default 0) without output
 181         assertThat(future.get(0), is(false));
 182         assertThat(future.body().length(), is(0));
 183
 184         // Test a bad response code (default 0) with output
 185         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(errorString.getBytes()));
 186         assertThat(future.get(0), is(false));
 187         assertThat(future.body(), is(errorString));
 188
 189         // Test a good response code
 190         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(successString.getBytes()));
 191         when(dfMock.newData()).thenReturn(dataMock);
 192         when(dataMock.in(null)).thenReturn(dataMock);
 193         when(dataMock.load((InputStream)any())).thenReturn(dataMock);
 194         when(dataMock.asObject()).thenReturn(hucMock);
 195         when(dataMock.asString()).thenReturn(successString);
 196         when(hucMock.getResponseCode()).thenReturn(200);
 197         assertThat(future.get(0), is(true));
 198         assertThat(future.body(), is(successString));
 199     }
 200
 201     @Test
 202     public void future1Test() throws LocatorException, CadiException, IOException, APIException {
 203         HClient client = new HClientStub(ssMock, uri, 0, hucMock);
 204         Future<HttpURLConnection> future = client.future(hucMock);
 205
 206         // Test a good response code
 207         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(successString.getBytes()));
 208         when(hucMock.getResponseCode()).thenReturn(200);
 209         assertThat(future.get(0), is(true));
 210         assertThat(future.body(), is("200"));
 211
 212         // Test a bad response code
 213         when(hucMock.getResponseCode()).thenReturn(0);
 214         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(errorString.getBytes()));
 215         assertThat(future.get(0), is(false));
 216         assertThat(future.body(), is(errorString));
 217     }
 218
 219     @Test
 220     public void future2Test() throws LocatorException, CadiException, IOException, APIException {
 221         HClient client = new HClientStub(ssMock, uri, 0, hucMock);
 222         Future<Void> future = client.future(respMock, 200);
 223
 224         ServletOutputStream sos = new ServletOutputStream() {
 225             @Override public void write(int arg0) throws IOException { }
 226         };
 227         when(respMock.getOutputStream()).thenReturn(sos);
 228
 229         // Test a good response code
 230         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(successString.getBytes()));
 231         when(hucMock.getResponseCode()).thenReturn(200);
 232         assertThat(future.get(0), is(true));
 233         assertThat(future.body(), is(nullValue()));
 234
 235         // Test a bad response code
 236         when(hucMock.getResponseCode()).thenReturn(0);
 237         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(errorString.getBytes()));
 238         assertThat(future.get(0), is(false));
 239         assertThat(future.body(), is(""));
 240     }
 241
 242     @Test
 243     public void hfutureTest() throws CadiException, IOException, LocatorException {
 244         HClient client = new HClientStub(ssMock, uri, 0, hucMock);
 245         HFutureStub future = new HFutureStub(client, hucMock);
 246         assertThat(future.get(0), is(false));
 247
 248         // Test a bad response code (default 0) with output
 249         when(hucMock.getInputStream()).thenReturn(new ByteArrayInputStream(errorString.getBytes()));
 250         assertThat(future.get(0), is(false));
 251
 252         assertThat(future.get(0), is(false));
 253
 254         when(hucMock.getResponseCode()).thenReturn(200);
 255         assertThat(future.get(0), is(true));
 256
 257         StringBuilder sb = future.inputStreamToString(new ByteArrayInputStream(errorString.getBytes()));
 258         assertThat(sb.toString(), is(errorString));
 259
 260         assertThat(future.code(), is(200));
 261         assertThat(future.huc(), is(hucMock));
 262
 263         assertThat(future.exception(), is(nullValue()));
 264         assertThat(future.header("string"), is(nullValue()));
 265
 266         // coverage...
 267         future.setHuc(null);
 268         future.close();
 269     }
 270
 271     @Test
 272     public void headerTest() throws LocatorException {
 273         HClient client = new HClientStub(ssMock, uri, 0, hucMock);
 274         String tag1 = "tag1";
 275         String tag2 = "tag2";
 276         String value1 = "value1";
 277         String value2 = "value2";
 278         client.addHeader(tag1, value1);
 279         client.addHeader(tag2, value2);
 280     }
 281
 282     @Test(expected = LocatorException.class)
 283     public void throws1Test() throws LocatorException {
 284         @SuppressWarnings("unused")
 285         HClient client = new HClient(ssMock, null, 0);
 286     }
 287
 288     private class HClientStub extends HClient {
 289         public HClientStub(SecuritySetter<HttpURLConnection> ss, URI uri, int connectTimeout, HttpURLConnection huc) throws LocatorException {
 290             super(ss, uri, connectTimeout);
 291             setHuc(huc);
 292         }
 293         public void setHuc(HttpURLConnection huc) {
 294             Field field;
 295             try {
 296                 field = HClient.class.getDeclaredField("huc");
 297                 field.setAccessible(true);
 298                 field.set(this, huc);
 299                 field.setAccessible(false);
 300             } catch (NoSuchFieldException | SecurityException | IllegalArgumentException | IllegalAccessException e) {
 301                 e.printStackTrace();
 302                 fail("Caught an exception: " + e.getMessage());
 303             }
 304         }
 305         @Override
 306         public HttpURLConnection getConnection(URI uri, StringBuilder pi) throws IOException {
 307             return hucMock;
 308         }
 309     }
 310
 311     private class HFutureStub extends HFuture<HttpURLConnection> {
 312         public HFutureStub(HClient hClient, HttpURLConnection huc) {
 313             hClient.super(huc);
 314         }
 315
 316         @Override public String body() { return null; }
 317         public void setHuc(HttpURLConnection huc) { this.huc = huc; }
 318     }
 319
 320 }