Make QuicSimpleServerStream supports multipart response body

components/cronet/android/test/quic_test_server.cc

 // 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.
 
 #include "quic_test_server.h"
 
 #include "base/android/jni_android.h"
+#include "base/android/jni_array.h"
 #include "base/android/jni_string.h"
 #include "base/android/path_utils.h"
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
+#include "base/strings/string_number_conversions.h"
 #include "base/threading/thread.h"
 #include "components/cronet/android/test/cronet_test_util.h"
 #include "jni/QuicTestServer_jni.h"
 #include "net/base/ip_endpoint.h"
 #include "net/base/test_data_directory.h"
 #include "net/quic/crypto/proof_source_chromium.h"
 #include "net/tools/quic/quic_in_memory_cache.h"
 #include "net/tools/quic/quic_simple_server.h"
 
 namespace cronet {
@@ skipping 39 matching lines @@
 }
 
 void ShutdownOnServerThread() {
   DCHECK(g_quic_server_thread->task_runner()->BelongsToCurrentThread());
   g_quic_server->Shutdown();
   delete g_quic_server;
 }
 
 }  // namespace
 
+void AddResponse(JNIEnv* env,
+                 const JavaParamRef<jclass>& /*jcaller*/,
+                 const JavaParamRef<jstring>& jpath,
+                 const JavaParamRef<jobjectArray>& jbody_chunks,
+                 const JavaParamRef<jobjectArray>& jtrailers) {
+  std::string host(kFakeQuicDomain);
+  host.append(":");
+  host.append(base::IntToString(kServerPort));
+  std::string path;
+  base::android::ConvertJavaStringToUTF8(env, jpath, &path);
+  std::vector<std::string> body_chunks;
+  base::android::AppendJavaStringArrayToStringVector(env, jbody_chunks,
+                                                     &body_chunks);
+  std::vector<std::string> trailers;
+  base::android::AppendJavaStringArrayToStringVector(env, jtrailers, &trailers);
+  net::SpdyHeaderBlock trailers_block;
+  for (int i = 0; i < trailers.size() / 2; i++) {
+    trailers_block.ReplaceOrAppendHeader(trailers[i * 2], trailers[i * 2 + 1]);
+  }
+
+  net::QuicInMemoryCache::GetInstance()->AddBidirectionalResponse(
+      host, path, 200, body_chunks, trailers_block);
+}
+
 // Quic server is currently hardcoded to run on port 6121 of the localhost on
 // the device.
 void StartQuicTestServer(JNIEnv* env,
                          const JavaParamRef<jclass>& /*jcaller*/,
                          const JavaParamRef<jstring>& jtest_files_root) {
   DCHECK(!g_quic_server_thread);
   g_quic_server_thread = new base::Thread("quic server thread");
   base::Thread::Options thread_options;
   thread_options.message_loop_type = base::MessageLoop::TYPE_IO;
   bool started = g_quic_server_thread->StartWithOptions(thread_options);
@@ skipping 20 matching lines @@
 
 int GetServerPort(JNIEnv* env, const JavaParamRef<jclass>& /*jcaller*/) {
   return kServerPort;
 }
 
 bool RegisterQuicTestServer(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 }  // namespace cronet