1 files changed, 128 insertions, 0 deletions
diff --git a/core/src/test/java/com/android/volley/toolbox/AdaptedHttpStackTest.java b/core/src/test/java/com/android/volley/toolbox/AdaptedHttpStackTest.java
new file mode 100644
index 0000000..dbd6535
--- /dev/null
+++ b/core/src/test/java/com/android/volley/toolbox/AdaptedHttpStackTest.java
@@ -0,0 +1,128 @@
+package com.android.volley.toolbox;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertSame;
+import static org.mockito.Mockito.when;
+
+import com.android.volley.Header;
+import com.android.volley.Request;
+import com.android.volley.mock.TestRequest;
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.SocketTimeoutException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+import org.apache.http.HttpEntity;
+import org.apache.http.HttpResponse;
+import org.apache.http.StatusLine;
+import org.apache.http.conn.ConnectTimeoutException;
+import org.apache.http.message.BasicHeader;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.robolectric.RobolectricTestRunner;
+
+@RunWith(RobolectricTestRunner.class)
+public class AdaptedHttpStackTest {
+    private static final Request<?> REQUEST = new TestRequest.Get();
+    private static final Map<String, String> ADDITIONAL_HEADERS = Collections.emptyMap();
+
+    @Mock private HttpStack mHttpStack;
+    @Mock private HttpResponse mHttpResponse;
+    @Mock private StatusLine mStatusLine;
+    @Mock private HttpEntity mHttpEntity;
+    @Mock private InputStream mContent;
+
+    private AdaptedHttpStack mAdaptedHttpStack;
+
+    @Before
+    public void setUp() {
+        MockitoAnnotations.initMocks(this);
+        mAdaptedHttpStack = new AdaptedHttpStack(mHttpStack);
+        when(mHttpResponse.getStatusLine()).thenReturn(mStatusLine);
+    }
+
+    @Test(expected = SocketTimeoutException.class)
+    public void requestTimeout() throws Exception {
+        when(mHttpStack.performRequest(REQUEST, ADDITIONAL_HEADERS))
+                .thenThrow(new ConnectTimeoutException());
+
+        mAdaptedHttpStack.executeRequest(REQUEST, ADDITIONAL_HEADERS);
+    }
+
+    @Test
+    public void emptyResponse() throws Exception {
+        when(mHttpStack.performRequest(REQUEST, ADDITIONAL_HEADERS)).thenReturn(mHttpResponse);
+        when(mStatusLine.getStatusCode()).thenReturn(12345);
+        when(mHttpResponse.getAllHeaders()).thenReturn(new org.apache.http.Header[0]);
+
+        com.android.volley.toolbox.HttpResponse response =
+                mAdaptedHttpStack.executeRequest(REQUEST, ADDITIONAL_HEADERS);
+
+        assertEquals(12345, response.getStatusCode());
+        assertEquals(Collections.emptyList(), response.getHeaders());
+        assertNull(response.getContent());
+    }
+
+    @Test
+    public void nonEmptyResponse() throws Exception {
+        when(mHttpStack.performRequest(REQUEST, ADDITIONAL_HEADERS)).thenReturn(mHttpResponse);
+        when(mStatusLine.getStatusCode()).thenReturn(12345);
+        when(mHttpResponse.getAllHeaders()).thenReturn(new org.apache.http.Header[0]);
+        when(mHttpResponse.getEntity()).thenReturn(mHttpEntity);
+        when(mHttpEntity.getContentLength()).thenReturn((long) Integer.MAX_VALUE);
+        when(mHttpEntity.getContent()).thenReturn(mContent);
+
+        com.android.volley.toolbox.HttpResponse response =
+                mAdaptedHttpStack.executeRequest(REQUEST, ADDITIONAL_HEADERS);
+
+        assertEquals(12345, response.getStatusCode());
+        assertEquals(Collections.emptyList(), response.getHeaders());
+        assertEquals(Integer.MAX_VALUE, response.getContentLength());
+        assertSame(mContent, response.getContent());
+    }
+
+    @Test(expected = IOException.class)
+    public void responseTooBig() throws Exception {
+        when(mHttpStack.performRequest(REQUEST, ADDITIONAL_HEADERS)).thenReturn(mHttpResponse);
+        when(mStatusLine.getStatusCode()).thenReturn(12345);
+        when(mHttpResponse.getAllHeaders()).thenReturn(new org.apache.http.Header[0]);
+        when(mHttpResponse.getEntity()).thenReturn(mHttpEntity);
+        when(mHttpEntity.getContentLength()).thenReturn(Integer.MAX_VALUE + 1L);
+        when(mHttpEntity.getContent()).thenReturn(mContent);
+
+        mAdaptedHttpStack.executeRequest(REQUEST, ADDITIONAL_HEADERS);
+    }
+
+    @Test
+    public void responseWithHeaders() throws Exception {
+        when(mHttpStack.performRequest(REQUEST, ADDITIONAL_HEADERS)).thenReturn(mHttpResponse);
+        when(mStatusLine.getStatusCode()).thenReturn(12345);
+        when(mHttpResponse.getAllHeaders())
+                .thenReturn(
+                        new org.apache.http.Header[] {
+                            new BasicHeader("header1", "value1_B"),
+                            new BasicHeader("header3", "value3"),
+                            new BasicHeader("HEADER2", "value2"),
+                            new BasicHeader("header1", "value1_A")
+                        });
+
+        com.android.volley.toolbox.HttpResponse response =
+                mAdaptedHttpStack.executeRequest(REQUEST, ADDITIONAL_HEADERS);
+
+        assertEquals(12345, response.getStatusCode());
+        assertNull(response.getContent());
+
+        List<Header> expectedHeaders = new ArrayList<>();
+        expectedHeaders.add(new Header("header1", "value1_B"));
+        expectedHeaders.add(new Header("header3", "value3"));
+        expectedHeaders.add(new Header("HEADER2", "value2"));
+        expectedHeaders.add(new Header("header1", "value1_A"));
+        assertEquals(expectedHeaders, response.getHeaders());
+    }
+}