[Cronet] Implement getOutputStream in CronetHttpURLConnection

components/cronet/android/test/javatests/src/org/chromium/net/urlconnection/CronetHttpURLConnectionTest.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.urlconnection;
 
 import android.os.Build;
 import android.test.suitebuilder.annotation.SmallTest;
 
 import org.chromium.base.annotations.SuppressFBWarnings;
 import org.chromium.base.test.util.Feature;
 import org.chromium.net.CronetTestActivity;
 import org.chromium.net.CronetTestBase;
 import org.chromium.net.MockUrlRequestJobFactory;
 import org.chromium.net.NativeTestServer;
 import org.chromium.net.UrlRequestContextConfig;
 import org.chromium.net.UrlRequestException;
 
 import java.io.ByteArrayOutputStream;
 import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.OutputStream;
+import java.lang.reflect.Method;
 import java.net.HttpURLConnection;
 import java.net.MalformedURLException;
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
@@ skipping 35 matching lines @@
     public void testBasicGet() throws Exception {
         URL url = new URL(NativeTestServer.getEchoMethodURL());
         HttpURLConnection urlConnection =
                 (HttpURLConnection) url.openConnection();
         assertEquals(200, urlConnection.getResponseCode());
         assertEquals("OK", urlConnection.getResponseMessage());
         assertEquals("GET", getResponseAsString(urlConnection));
         urlConnection.disconnect();
     }
 
+    /**
+     * Tests that using reflection to find {@code fixedContentLengthLong} works.
+     */
+    @SmallTest
+    @Feature({"Cronet"})
+    @OnlyRunCronetHttpURLConnection
+    public void testSetFixedLengthStreamingModeLong() throws Exception {
+        URL url = new URL(NativeTestServer.getEchoBodyURL());
+        HttpURLConnection connection =
+                (HttpURLConnection) url.openConnection();
+        connection.setDoOutput(true);
+        connection.setRequestMethod("POST");
+        if (Build.VERSION.SDK_INT >= 19) {
+            String dataString = "some very important data";
+            byte[] data = dataString.getBytes();
+            Class<?> c = connection.getClass();
+            Method method = c.getMethod("setFixedLengthStreamingMode",
+                    new Class[] {long.class});
+            method.invoke(connection, (long) data.length);
+            OutputStream out = connection.getOutputStream();
+            out.write(data);
+            assertEquals(200, connection.getResponseCode());
+            assertEquals("OK", connection.getResponseMessage());
+            assertEquals(dataString, getResponseAsString(connection));
+            connection.disconnect();
+        }
+    }
+
+    @SmallTest
+    @Feature({"Cronet"})
+    @OnlyRunCronetHttpURLConnection
+    // TODO(xunjieli): Change the test after chunked support is added.
+    public void testPostChunked() throws Exception {
+        URL url = new URL(NativeTestServer.getEchoBodyURL());
+        HttpURLConnection connection =
+                (HttpURLConnection) url.openConnection();
+        connection.setDoOutput(true);
+        connection.setRequestMethod("POST");
+        try {
+            connection.setChunkedStreamingMode(0);
+        } catch (UnsupportedOperationException e) {
+            assertEquals("Chunked mode not supported yet", e.getMessage());
+        }
+    }
+
     @SmallTest
     @Feature({"Cronet"})
     @CompareDefaultWithCronet
     public void testNotFoundURLRequest() throws Exception {
         URL url = new URL(NativeTestServer.getFileURL("/notfound.html"));
         HttpURLConnection urlConnection =
                 (HttpURLConnection) url.openConnection();
         assertEquals(404, urlConnection.getResponseCode());
         assertEquals("Not Found", urlConnection.getResponseMessage());
         try {
@@ skipping 131 matching lines @@
             InputStream in = urlConnection.getInputStream();
             fail();
         } catch (Exception e) {
             // Ignored.
         }
     }
 
     @SmallTest
     @Feature({"Cronet"})
     @CompareDefaultWithCronet
+    public void testMultipleDisconnect() throws Exception {
+        URL url = new URL(NativeTestServer.getEchoMethodURL());
+        HttpURLConnection urlConnection =
+                (HttpURLConnection) url.openConnection();
+        assertEquals(200, urlConnection.getResponseCode());
+        assertEquals("OK", urlConnection.getResponseMessage());
+        assertEquals("GET", getResponseAsString(urlConnection));
+        // Disconnect multiple times should be fine.
+        for (int i = 0; i < 10; i++) {
+            urlConnection.disconnect();
+        }
+    }
+
+    @SmallTest
+    @Feature({"Cronet"})
+    @CompareDefaultWithCronet
     public void testAddRequestProperty() throws Exception {
         URL url = new URL(NativeTestServer.getEchoAllHeadersURL());
         HttpURLConnection connection = (HttpURLConnection) url.openConnection();
         connection.addRequestProperty("foo-header", "foo");
         connection.addRequestProperty("bar-header", "bar");
 
         // Before connection is made, check request headers are set.
         Map<String, List<String>> requestHeadersMap =
                 connection.getRequestProperties();
         List<String> fooValues = requestHeadersMap.get("foo-header");
@@ skipping 708 matching lines @@
             String headerName) {
         Pattern pattern = Pattern.compile(headerName + ":\\s(.*)\\r\\n");
         Matcher matcher = pattern.matcher(allHeaders);
         List<String> headerValues = new ArrayList<String>();
         while (matcher.find()) {
             headerValues.add(matcher.group(1));
         }
         return headerValues;
     }
 }