Fix a pair of Cronet upload bugs and add tests.

components/cronet/android/test/javatests/src/org/chromium/cronet_test_apk/UploadTest.java

Index: components/cronet/android/test/javatests/src/org/chromium/cronet_test_apk/UploadTest.java
diff --git a/components/cronet/android/test/javatests/src/org/chromium/cronet_test_apk/UploadTest.java b/components/cronet/android/test/javatests/src/org/chromium/cronet_test_apk/UploadTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..bc0156f7eca0430b6fed47b9f6b796b20988b0cb
--- /dev/null
+++ b/components/cronet/android/test/javatests/src/org/chromium/cronet_test_apk/UploadTest.java
@@ -0,0 +1,249 @@
+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.cronet_test_apk;
+
+import android.test.suitebuilder.annotation.SmallTest;
+
+import org.chromium.base.test.util.Feature;
+import org.chromium.net.HttpUrlRequest;
+import org.chromium.net.HttpUrlRequestListener;
+
+import java.io.ByteArrayInputStream;
+import java.io.InputStream;
+
+import java.nio.channels.Channels;
+import java.nio.channels.ReadableByteChannel;
+
+import java.util.HashMap;
+
+/**
+ * Example test that just starts the cronet sample.
+ */
+public class UploadTest extends CronetTestBase {
+    private static final String UPLOAD_DATA = "Nifty upload data!";
+    private static final String UPLOAD_CHANNEL_DATA = "Upload channel data";
+
+    private HttpUrlRequest createRequest(
+            String url, HttpUrlRequestListener listener,
+            CronetTestActivity activity) {
+        HashMap<String, String> headers = new HashMap<String, String>();
+        return activity.mRequestFactory.createRequest(
+                url, HttpUrlRequest.REQUEST_PRIORITY_MEDIUM, headers, listener);
+    }
+
+    // Sets request to have an upload channel containing the given data.
+    // uploadDataLength should generally be uploadData.length(), unless a test
+    // needs to get a read error.
+    private void setUploadChannel(HttpUrlRequest request,
+                                  String contentType,
+                                  String uploadData,
+                                  int uploadDataLength) {
+        InputStream uploadDataStream = new ByteArrayInputStream(
+                uploadData.getBytes());
+        ReadableByteChannel uploadDataChannel =
+                Channels.newChannel(uploadDataStream);
+        request.setUploadChannel(
+                contentType, uploadDataChannel, uploadDataLength);
+    }
+
+    // Tests uploading an in-memory string.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadData() throws Exception {
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+        HttpUrlRequest request = createRequest(
+                UploadTestServer.getEchoBodyURL(), listener, activity);
+        request.setUploadData("text/plain", UPLOAD_DATA.getBytes("UTF8"));
+        request.start();
+        listener.blockForComplete();
+
+        assertEquals(200, listener.mHttpStatusCode);
+        assertEquals(UPLOAD_DATA, listener.mResponseAsString);
+    }
+
+    // Tests uploading an in-memory string with a redirect that preserves the
+    // POST body.  This makes sure the body is correctly sent again.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadDataWithRedirect() throws Exception {
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+        HttpUrlRequest request = createRequest(
+                UploadTestServer.getRedirectToEchoBody(), listener, activity);
+        request.setUploadData("text/plain", UPLOAD_DATA.getBytes("UTF8"));
+        request.start();
+        listener.blockForComplete();
+
+        assertEquals(200, listener.mHttpStatusCode);
+        assertEquals(UPLOAD_DATA, listener.mResponseAsString);
+    }
+
+    // Tests Content-Type can be set when uploading an in-memory string.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadDataContentType() throws Exception {
+        String contentTypes[] = {"text/plain", "chicken/spicy"};
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        for (String contentType : contentTypes) {
+            TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+            HttpUrlRequest request = createRequest(
+                    UploadTestServer.getEchoHeaderURL("Content-Type"), listener,
+                    activity);
+            request.setUploadData(contentType, UPLOAD_DATA.getBytes("UTF8"));
+            request.start();
+            listener.blockForComplete();
+
+            assertEquals(200, listener.mHttpStatusCode);
+            assertEquals(contentType, listener.mResponseAsString);
+        }
+    }
+
+    // Tests the default method when uploading.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testDefaultUploadMethod() throws Exception {
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+        HttpUrlRequest request = createRequest(
+                UploadTestServer.getEchoMethodURL(), listener, activity);
+        request.setUploadData("text/plain", UPLOAD_DATA.getBytes("UTF8"));
+        request.start();
+        listener.blockForComplete();
+
+        assertEquals(200, listener.mHttpStatusCode);
+        assertEquals("POST", listener.mResponseAsString);
+    }
+
+    // Tests methods can be set when uploading.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadMethods() throws Exception {
+        String uploadMethods[] = {"POST", "PUT"};
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        for (String uploadMethod : uploadMethods) {
+            TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+            HttpUrlRequest request = createRequest(
+                    UploadTestServer.getEchoMethodURL(), listener, activity);
+            request.setHttpMethod(uploadMethod);
+            request.setUploadData("text/plain", UPLOAD_DATA.getBytes("UTF8"));
+            request.start();
+            listener.blockForComplete();
+
+            assertEquals(200, listener.mHttpStatusCode);
+            assertEquals(uploadMethod, listener.mResponseAsString);
+        }
+    }
+
+    // Tests uploading from a channel.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadChannel() throws Exception {
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+        HttpUrlRequest request = createRequest(
+                UploadTestServer.getEchoBodyURL(), listener, activity);
+        setUploadChannel(request, "text/plain", UPLOAD_CHANNEL_DATA,
+                         UPLOAD_CHANNEL_DATA.length());
+        request.start();
+        listener.blockForComplete();
+
+        assertEquals(200, listener.mHttpStatusCode);
+        assertEquals(UPLOAD_CHANNEL_DATA, listener.mResponseAsString);
+    }
+
+    // Tests uploading from a channel in the case a redirect preserves the post
+    // body.  Since channels can't be rewound, the request fails when we try to
+    // rewind it to send the second request.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadChannelWithRedirect() throws Exception {
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+        HttpUrlRequest request = createRequest(
+                UploadTestServer.getRedirectToEchoBody(), listener, activity);
+        setUploadChannel(request, "text/plain", UPLOAD_CHANNEL_DATA,
+                         UPLOAD_CHANNEL_DATA.length());
+        request.start();
+        listener.blockForComplete();
+
+        assertEquals(0, listener.mHttpStatusCode);
+    }
+
+    // Tests uploading from a channel when there's a read error.  The body
+    // should be 0-padded.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadChannelWithReadError() throws Exception {
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+        HttpUrlRequest request = createRequest(
+                UploadTestServer.getEchoBodyURL(), listener, activity);
+        setUploadChannel(request, "text/plain", UPLOAD_CHANNEL_DATA,
+                         UPLOAD_CHANNEL_DATA.length() + 2);
+        request.start();
+        listener.blockForComplete();
+
+        assertEquals(200, listener.mHttpStatusCode);
+        assertEquals(UPLOAD_CHANNEL_DATA + "\0\0", listener.mResponseAsString);
+    }
+
+    // Tests Content-Type can be set when uploading from a channel.
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadChannelContentType() throws Exception {
+        String contentTypes[] = {"text/plain", "chicken/spicy"};
+        CronetTestActivity activity = launchCronetTestApp();
+        assertNotNull(activity);
+        assertTrue(UploadTestServer.startUploadTestServer());
+        MockUrlRequestJobUtil.addUrlInterceptors();
+
+        for (String contentType : contentTypes) {
+            TestHttpUrlRequestListener listener = new TestHttpUrlRequestListener();
+            HttpUrlRequest request = createRequest(
+                    UploadTestServer.getEchoHeaderURL("Content-Type"), listener,
+                    activity);
+            setUploadChannel(request, contentType, UPLOAD_CHANNEL_DATA,
+                             UPLOAD_CHANNEL_DATA.length());
+            request.start();
+            listener.blockForComplete();
+
+            assertEquals(200, listener.mHttpStatusCode);
+            assertEquals(contentType, listener.mResponseAsString);
+        }
+    }
+}