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

components/cronet/android/test/src/org/chromium/net/TestUploadDataStreamHandler.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 junit.framework.Assert;
 
 import org.chromium.base.annotations.CalledByNative;
 import org.chromium.base.annotations.JNINamespace;
 import org.chromium.base.annotations.NativeClassQualifiedName;
 
 /**
  * A wrapper class on top of the native net::UploadDataStream. This class is
  * used in tests to drive the native UploadDataStream directly.
  */
 @JNINamespace("cronet")
 public final class TestUploadDataStreamHandler {
+    private CronetEngine mCronetEngine;
     private long mTestUploadDataStreamHandler;
     private ConditionVariable mWaitInitCalled = new ConditionVariable();
     private ConditionVariable mWaitInitComplete = new ConditionVariable();
     private ConditionVariable mWaitReadComplete = new ConditionVariable();
     private ConditionVariable mWaitResetComplete = new ConditionVariable();
     // Waits for checkIfInitCallbackInvoked() returns result asynchronously.
     private ConditionVariable mWaitCheckInit = new ConditionVariable();
     // Waits for checkIfReadCallbackInvoked() returns result asynchronously.
     private ConditionVariable mWaitCheckRead = new ConditionVariable();
     // If true, init completes synchronously.
     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();
         }
     }
 
     /**
      * Init and returns whether init completes synchronously.
      */
     public boolean init() {
         mData = "";
         nativeInit(mTestUploadDataStreamHandler);
         mWaitInitCalled.block();
@@ skipping 108 matching lines @@
             long nativePtr);
 
     @NativeClassQualifiedName("TestUploadDataStreamHandler")
     private native void nativeCheckReadCallbackNotInvoked(
             long nativePtr);
 
     @NativeClassQualifiedName("TestUploadDataStreamHandler")
     private native void nativeDestroy(long nativePtr);
 
     private native long nativeCreateTestUploadDataStreamHandler(
-            long uploadDataStream);
+            long uploadDataStream, long singleThreadTaskRunner);
 }