[Cronet] Enable chunked upload in CronetUrlRequest

components/cronet/android/test/javatests/src/org/chromium/net/CronetUrlRequestTest.java

 // 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.net;
 
 import android.test.suitebuilder.annotation.SmallTest;
 import android.util.Pair;
 
 import org.chromium.base.test.util.Feature;
@@ skipping 792 matching lines @@
         assertEquals(1, dataProvider.getNumReadCalls());
         assertEquals(1, dataProvider.getNumRewindCalls());
 
         assertEquals("Exception received from UploadDataProvider", listener.mError.getMessage());
         assertEquals("Thrown rewind failure", listener.mError.getCause().getMessage());
         assertEquals(null, listener.mResponseInfo);
     }
 
     @SmallTest
     @Feature({"Cronet"})
-    public void testUploadChunkedNotSupported() throws Exception {
+    public void testUploadChunked() throws Exception {
         TestUrlRequestListener listener = new TestUrlRequestListener();
         UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
-                NativeTestServer.getRedirectToEchoBody(), listener, listener.getExecutor());
+                NativeTestServer.getEchoBodyURL(), listener, listener.getExecutor());
 
         TestUploadDataProvider dataProvider = new TestUploadDataProvider(
                 TestUploadDataProvider.SuccessCallbackMode.SYNC, listener.getExecutor());
+        dataProvider.addRead("test hello".getBytes());
         dataProvider.setChunked(true);
-        try {
-            urlRequest.setUploadDataProvider(dataProvider, listener.getExecutor());
-            fail("Exception not thrown");
-        } catch (IllegalArgumentException e) {
-            assertEquals("Chunked uploads not supported.", e.getMessage());
-        }
+        urlRequest.setUploadDataProvider(dataProvider, listener.getExecutor());
+        urlRequest.addHeader("Content-Type", "useless/string");
+
+        assertEquals(-1, dataProvider.getLength());
+
+        urlRequest.start();
+        listener.blockForDone();
+
+        // 1 read call for one data chunk.
+        assertEquals(1, dataProvider.getNumReadCalls());
+        assertEquals("test hello", listener.mResponseAsString);
+    }
+
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testUploadChunkedLastReadZeroLengthBody() throws Exception {
+        TestUrlRequestListener listener = new TestUrlRequestListener();
+        UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
+                NativeTestServer.getEchoBodyURL(), listener, listener.getExecutor());
+
+        TestUploadDataProvider dataProvider = new TestUploadDataProvider(
+                TestUploadDataProvider.SuccessCallbackMode.SYNC, listener.getExecutor());
+        // Add 3 reads. The last read has a 0-length body.
+        dataProvider.addRead("hello there".getBytes());
+        dataProvider.addRead("!".getBytes());
+        dataProvider.addRead("".getBytes());
+        dataProvider.setChunked(true);
+        urlRequest.setUploadDataProvider(dataProvider, listener.getExecutor());
+        urlRequest.addHeader("Content-Type", "useless/string");
+
+        assertEquals(-1, dataProvider.getLength());
+
+        urlRequest.start();
+        listener.blockForDone();
+
+        // 2 read call for the first two data chunks, and 1 for final chunk.
+        assertEquals(3, dataProvider.getNumReadCalls());
+        assertEquals("hello there!", listener.mResponseAsString);
     }
 
     private void throwOrCancel(FailureType failureType, ResponseStep failureStep,
             boolean expectResponseInfo, boolean expectError) {
         TestUrlRequestListener listener = new TestUrlRequestListener();
         listener.setFailure(failureType, failureStep);
         UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
                 MockUrlRequestJobFactory.REDIRECT_URL,
                 listener, listener.getExecutor());
         urlRequest.start();
@@ skipping 41 matching lines @@
         urlRequest.start();
         listener.blockForDone();
         assertTrue(listener.mOnRedirectCalled);
         assertEquals(listener.mResponseStep, ResponseStep.ON_SUCCEEDED);
         assertFalse(urlRequest.isCanceled());
         assertNotNull(listener.mResponseInfo);
         assertNull(listener.mError);
         assertFalse(listener.mOnErrorCalled);
     }
 }