Java tutorial
/* * The MIT License (MIT) * * Copyright (c) 2015 OnDeck * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to deal * in the Software without restriction, including without limitation the rights * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell * copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN * THE SOFTWARE. * * Except as contained in this notice, the name of a copyright holder shall not be used in advertising or otherwise to promote the sale, use or other dealings in this Software without prior written authorization of the copyright holder. */ package com.ondeck.datapipes.http; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertNull; import static org.mockito.Mockito.when; import java.io.IOException; import org.apache.commons.io.IOUtils; import org.apache.http.HttpEntity; import org.apache.http.HttpResponse; import org.apache.http.StatusLine; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mock; import org.mockito.runners.MockitoJUnitRunner; @RunWith(MockitoJUnitRunner.class) public class HttpConnectorResponseHandlerUnitTest { private HttpConnectorResponseHandler underTest; @Mock private HttpResponse httpResponse; @Mock private StatusLine statusLine; @Mock private HttpEntity httpEntity; @Before public void setup() { underTest = new HttpConnectorResponseHandler(); } @Test public void testHandleResponseWithoutEntity() throws IOException { when(httpResponse.getStatusLine()).thenReturn(statusLine); Integer expectedCode = 200; when(statusLine.getStatusCode()).thenReturn(expectedCode); HttpConnectorResponse result = underTest.handleResponse(httpResponse); assertNotNull(result); assertEquals(expectedCode.intValue(), result.getHttpStatusCode()); assertNull(result.getContent()); } @Test public void testHandleResponseWithEntity() throws IOException { when(httpResponse.getStatusLine()).thenReturn(statusLine); Integer expectedCode = 200; when(statusLine.getStatusCode()).thenReturn(expectedCode); when(httpResponse.getEntity()).thenReturn(httpEntity); String expectedContent = "someContent"; when(httpEntity.getContent()).thenReturn(IOUtils.toInputStream(expectedContent)); HttpConnectorResponse result = underTest.handleResponse(httpResponse); assertNotNull(result); assertEquals(expectedCode.intValue(), result.getHttpStatusCode()); assertEquals(expectedContent, result.getContent()); } }