[Cronet] Test the libcronet that's shipped, not libcronet_test

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

Index: components/cronet/android/test/src/org/chromium/net/TestUploadDataStreamHandler.java
diff --git a/components/cronet/android/test/src/org/chromium/net/TestUploadDataStreamHandler.java b/components/cronet/android/test/src/org/chromium/net/TestUploadDataStreamHandler.java
index 0d2e6100869a29f4cc2a5857b5b707a77428030e..7e3bc217355feeee5c39c2334244ecbea655a3b1 100644
--- a/components/cronet/android/test/src/org/chromium/net/TestUploadDataStreamHandler.java
+++ b/components/cronet/android/test/src/org/chromium/net/TestUploadDataStreamHandler.java
@@ -4,6 +4,7 @@
 
 package org.chromium.net;
 
+import android.content.Context;
 import android.os.ConditionVariable;
 
 import junit.framework.Assert;
@@ -18,6 +19,7 @@ import org.chromium.base.annotations.NativeClassQualifiedName;
  */
 @JNINamespace("cronet")
 public final class TestUploadDataStreamHandler {
+    private CronetEngine mCronetEngine;
     private long mTestUploadDataStreamHandler;
     private ConditionVariable mWaitInitCalled = new ConditionVariable();
     private ConditionVariable mWaitInitComplete = new ConditionVariable();
@@ -31,15 +33,19 @@ public final class TestUploadDataStreamHandler {
     private boolean mInitCompletedSynchronously = false;
     private String mData = "";
 
-    public TestUploadDataStreamHandler(final long uploadDataStream) {
-        mTestUploadDataStreamHandler =
-                nativeCreateTestUploadDataStreamHandler(uploadDataStream);
+    public TestUploadDataStreamHandler(Context context, final long uploadDataStream) {
+        mCronetEngine = new CronetEngine.Builder(context).build();
+        CronetTestUtil.prepareNetworkThreadForTesting(mCronetEngine);
+        mTestUploadDataStreamHandler = nativeCreateTestUploadDataStreamHandler(
+                uploadDataStream, CronetTestUtil.getNativeSingleThreadTaskRunner());
     }
 
     public void destroyNativeObjects() {
         if (mTestUploadDataStreamHandler != 0) {
             nativeDestroy(mTestUploadDataStreamHandler);
             mTestUploadDataStreamHandler = 0;
+            CronetTestUtil.cleanupNetorkThreadForTesting();
+            mCronetEngine.shutdown();
         }
     }
 
@@ -168,5 +174,5 @@ public final class TestUploadDataStreamHandler {
     private native void nativeDestroy(long nativePtr);
 
     private native long nativeCreateTestUploadDataStreamHandler(
-            long uploadDataStream);
+            long uploadDataStream, long singleThreadTaskRunner);
 }