1 files changed, 104 insertions, 0 deletions
diff --git a/core/src/test/java/com/android/volley/toolbox/BaseHttpStackTest.java b/core/src/test/java/com/android/volley/toolbox/BaseHttpStackTest.java
new file mode 100644
index 0000000..1049ad0
--- /dev/null
+++ b/core/src/test/java/com/android/volley/toolbox/BaseHttpStackTest.java
@@ -0,0 +1,104 @@
+package com.android.volley.toolbox;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertSame;
+
+import com.android.volley.AuthFailureError;
+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.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
+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 BaseHttpStackTest {
+    private static final Request<?> REQUEST = new TestRequest.Get();
+    private static final Map<String, String> ADDITIONAL_HEADERS = Collections.emptyMap();
+
+    @Mock private InputStream mContent;
+
+    @Before
+    public void setUp() {
+        MockitoAnnotations.initMocks(this);
+    }
+
+    @Test
+    public void legacyRequestWithoutBody() throws Exception {
+        BaseHttpStack stack =
+                new BaseHttpStack() {
+                    @Override
+                    public HttpResponse executeRequest(
+                            Request<?> request, Map<String, String> additionalHeaders)
+                            throws IOException, AuthFailureError {
+                        assertSame(REQUEST, request);
+                        assertSame(ADDITIONAL_HEADERS, additionalHeaders);
+                        return new HttpResponse(12345, Collections.<Header>emptyList());
+                    }
+                };
+        org.apache.http.HttpResponse resp = stack.performRequest(REQUEST, ADDITIONAL_HEADERS);
+        assertEquals(12345, resp.getStatusLine().getStatusCode());
+        assertEquals(0, resp.getAllHeaders().length);
+        assertNull(resp.getEntity());
+    }
+
+    @Test
+    public void legacyResponseWithBody() throws Exception {
+        BaseHttpStack stack =
+                new BaseHttpStack() {
+                    @Override
+                    public HttpResponse executeRequest(
+                            Request<?> request, Map<String, String> additionalHeaders)
+                            throws IOException, AuthFailureError {
+                        assertSame(REQUEST, request);
+                        assertSame(ADDITIONAL_HEADERS, additionalHeaders);
+                        return new HttpResponse(
+                                12345, Collections.<Header>emptyList(), 555, mContent);
+                    }
+                };
+        org.apache.http.HttpResponse resp = stack.performRequest(REQUEST, ADDITIONAL_HEADERS);
+        assertEquals(12345, resp.getStatusLine().getStatusCode());
+        assertEquals(0, resp.getAllHeaders().length);
+        assertEquals(555L, resp.getEntity().getContentLength());
+        assertSame(mContent, resp.getEntity().getContent());
+    }
+
+    @Test
+    public void legacyResponseHeaders() throws Exception {
+        BaseHttpStack stack =
+                new BaseHttpStack() {
+                    @Override
+                    public HttpResponse executeRequest(
+                            Request<?> request, Map<String, String> additionalHeaders)
+                            throws IOException, AuthFailureError {
+                        assertSame(REQUEST, request);
+                        assertSame(ADDITIONAL_HEADERS, additionalHeaders);
+                        List<Header> headers = new ArrayList<>();
+                        headers.add(new Header("HeaderA", "ValueA"));
+                        headers.add(new Header("HeaderB", "ValueB_1"));
+                        headers.add(new Header("HeaderB", "ValueB_2"));
+                        return new HttpResponse(12345, headers);
+                    }
+                };
+        org.apache.http.HttpResponse resp = stack.performRequest(REQUEST, ADDITIONAL_HEADERS);
+        assertEquals(12345, resp.getStatusLine().getStatusCode());
+        assertEquals(3, resp.getAllHeaders().length);
+        assertEquals("HeaderA", resp.getAllHeaders()[0].getName());
+        assertEquals("ValueA", resp.getAllHeaders()[0].getValue());
+        assertEquals("HeaderB", resp.getAllHeaders()[1].getName());
+        assertEquals("ValueB_1", resp.getAllHeaders()[1].getValue());
+        assertEquals("HeaderB", resp.getAllHeaders()[2].getName());
+        assertEquals("ValueB_2", resp.getAllHeaders()[2].getValue());
+        assertNull(resp.getEntity());
+    }
+}