Make QuicSimpleServerStream supports multipart response body

components/cronet/android/test/src/org/chromium/net/QuicTestServer.java

 // Copyright 2015 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.content.Context;
 import android.os.ConditionVariable;
 
 import org.chromium.base.Log;
 import org.chromium.base.annotations.CalledByNative;
 import org.chromium.base.annotations.JNINamespace;
 
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * Wrapper class to start a Quic test server.
  */
 @JNINamespace("cronet")
 public final class QuicTestServer {
     private static final ConditionVariable sBlock = new ConditionVariable();
     private static final String TAG = "cr.QuicTestServer";
 
     private static final String CERT_USED = "quic_test.example.com.crt";
     private static final String KEY_USED = "quic_test.example.com.key";
     private static final String[] CERTS_USED = {CERT_USED};
 
+    private static final List<Response> sResponses = new ArrayList<Response>();
+
+    /**
+     * Represents a response returned by the server.
+     */
+    public static class Response {
+        private final String mPath;
+        private final String[] mBodyChunks;
+        private final String[] mTrailers;
+
+        /**
+         * Constructs a Response.
+         * @param path path of the request url
+         * @param bodyChunks multipart response body
+         * @param trailers an array of response headers with keys at the even indices
+         *       followed by the corresponding values at the odd indices.
+         */
+        public Response(String path, String[] bodyChunks, String[] trailers) {
+            mPath = path;
+            mBodyChunks = bodyChunks.clone();
+            mTrailers = trailers.clone();
+        }
+
+        public final String getPath() {
+            return mPath;
+        }
+
+        public final String[] getBodyChunks() {
+            return mBodyChunks.clone();
+        }
+
+        public final String[] getTrailers() {
+            return mTrailers.clone();
+        }
+    }
+
+    public static void addResponse(Response response) {
+        sResponses.add(response);
+    }
+
     public static void startQuicTestServer(Context context) {
         TestFilesInstaller.installIfNeeded(context);
+        for (Response r : sResponses) {
+            nativeAddResponse(r.getPath(), r.getBodyChunks(), r.getTrailers());
+        }
         nativeStartQuicTestServer(TestFilesInstaller.getInstalledPath(context));
         sBlock.block();
     }
 
     public static void shutdownQuicTestServer() {
         nativeShutdownQuicTestServer();
         sBlock.close();
     }
 
     public static String getServerURL() {
@@ skipping 19 matching lines @@
     public static long createMockCertVerifier() {
         return MockCertVerifier.createMockCertVerifier(CERTS_USED);
     }
 
     @CalledByNative
     private static void onServerStarted() {
         Log.i(TAG, "Quic server started.");
         sBlock.open();
     }
 
+    private static native void nativeAddResponse(
+            String path, String[] bodyChunks, String[] trailers);
     private static native void nativeStartQuicTestServer(String filePath);
     private static native void nativeShutdownQuicTestServer();
     private static native String nativeGetServerHost();
     private static native int nativeGetServerPort();
 }