Added Network Quality Estimator Real-time interface to Cronet

components/cronet/android/test/javatests/src/org/chromium/net/CronetUrlRequestContextTest.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;
 
 import android.content.Context;
 import android.content.ContextWrapper;
 import android.os.ConditionVariable;
 import android.os.Handler;
 import android.os.Looper;
 import android.test.suitebuilder.annotation.SmallTest;
 
 import org.chromium.base.PathUtils;
 import org.chromium.base.test.util.Feature;
 import org.chromium.net.TestUrlRequestListener.ResponseStep;
 
 import java.io.BufferedReader;
 import java.io.File;
 import java.io.FileReader;
+import java.util.LinkedList;
+import java.util.NoSuchElementException;
+import java.util.concurrent.Executor;
 
 /**
  * Test CronetUrlRequestContext.
  */
 public class CronetUrlRequestContextTest extends CronetTestBase {
     // URLs used for tests.
     private static final String TEST_URL = "http://127.0.0.1:8000";
     private static final String URL_404 = "http://127.0.0.1:8000/notfound404";
     private static final String MOCK_CRONET_TEST_FAILED_URL =
             "http://mock.failed.request/-2";
@@ skipping 41 matching lines @@
 
         @Override
         public void onFailed(UrlRequest request,
                 ResponseInfo info,
                 UrlRequestException error) {
             super.onFailed(request, info, error);
             mActivity.mUrlRequestContext.shutdown();
         }
     }
 
+    static class TestExecutor implements Executor {
+        private final LinkedList<Runnable> mTaskQueue = new LinkedList<Runnable>();
+
+        @Override
+        public void execute(Runnable task) {
+            mTaskQueue.add(task);
+        }
+
+        public void runAllTasks() {
+            try {
+                while (mTaskQueue.size() > 0) {
+                    mTaskQueue.remove().run();
+                }
+            } catch (NoSuchElementException e) {
+            }
+        }
+    }
+
+    static class TestNetworkQualityListener
+            implements NetworkQualityRttListener, NetworkQualityThroughputListener {
+        int mRttObservationCount;
+        int mThroughputObservationCount;
+
+        public void onRttObservation(int rttMs, long when, int source) {
+            mRttObservationCount++;
+        }
+
+        public void onThroughputObservation(int throughputKbps, long when, int source) {
+            mThroughputObservationCount++;
+        }
+
+        public int rttObservationCount() {
+            return mRttObservationCount;
+        }
+
+        public int throughputObservationCount() {
+            return mThroughputObservationCount;
+        }
+    }
+
     @SmallTest
     @Feature({"Cronet"})
     public void testConfigUserAgent() throws Exception {
         String userAgentName = "User-Agent";
         String userAgentValue = "User-Agent-Value";
         UrlRequestContextConfig config = new UrlRequestContextConfig();
         config.setUserAgent(userAgentValue);
         config.setLibraryName("cronet_tests");
         String[] commandLineArgs = {
                 CronetTestActivity.CONFIG_KEY, config.toString()
@@ skipping 55 matching lines @@
         // Proxy logic configured to use the test server as its proxy.
         assertEquals(200, listener.mResponseInfo.getHttpStatusCode());
         assertEquals(serverHostPort, listener.mResponseInfo.getProxyServer());
         assertEquals(
                 "http://DomainThatDoesnt.Resolve/datareductionproxysuccess.txt",
                 listener.mResponseInfo.getUrl());
     }
 
     @SmallTest
     @Feature({"Cronet"})
+    public void testRealTimeNetworkQualityObservationsNotEnabled() throws Exception {
+        mActivity = launchCronetTestApp();
+        TestNetworkQualityListener networkQualityListener = new TestNetworkQualityListener();
+        try {
+            mActivity.mUrlRequestContext.addRttListener(networkQualityListener);
+            fail("Should throw an exception.");
+        } catch (IllegalStateException e) {
+        }
+        try {
+            mActivity.mUrlRequestContext.addThroughputListener(networkQualityListener);
+            fail("Should throw an exception.");
+        } catch (IllegalStateException e) {
+        }
+        TestUrlRequestListener listener = new TestUrlRequestListener();
+        UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
+                TEST_URL, listener, listener.getExecutor());
+        urlRequest.start();
+        listener.blockForDone();
+        assertEquals(0, networkQualityListener.rttObservationCount());
+        assertEquals(0, networkQualityListener.throughputObservationCount());
+        mActivity.mUrlRequestContext.shutdown();
+    }
+
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testRealTimeNetworkQualityObservationsListenerRemoved() throws Exception {
+        mActivity = launchCronetTestApp();
+        TestExecutor testExecutor = new TestExecutor();
+        TestNetworkQualityListener networkQualityListener = new TestNetworkQualityListener();
+        mActivity.mUrlRequestContext.enableNetworkQualityEstimatorForTesting(
+                true, true, testExecutor);
+        mActivity.mUrlRequestContext.addRttListener(networkQualityListener);
+        mActivity.mUrlRequestContext.addThroughputListener(networkQualityListener);
+        mActivity.mUrlRequestContext.removeRttListener(networkQualityListener);
+        mActivity.mUrlRequestContext.removeThroughputListener(networkQualityListener);
+        TestUrlRequestListener listener = new TestUrlRequestListener();
+        UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
+                TEST_URL, listener, listener.getExecutor());
+        urlRequest.start();
+        listener.blockForDone();
+        testExecutor.runAllTasks();
+        assertEquals(0, networkQualityListener.rttObservationCount());
+        assertEquals(0, networkQualityListener.throughputObservationCount());
+        mActivity.mUrlRequestContext.shutdown();
+    }
+
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testRealTimeNetworkQualityObservations() throws Exception {
+        mActivity = launchCronetTestApp();
+        TestExecutor testExecutor = new TestExecutor();
+        TestNetworkQualityListener networkQualityListener = new TestNetworkQualityListener();
+        mActivity.mUrlRequestContext.enableNetworkQualityEstimatorForTesting(
+                true, true, testExecutor);
+        mActivity.mUrlRequestContext.addRttListener(networkQualityListener);
+        mActivity.mUrlRequestContext.addThroughputListener(networkQualityListener);
+        TestUrlRequestListener listener = new TestUrlRequestListener();
+        UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
+                TEST_URL, listener, listener.getExecutor());
+        urlRequest.start();
+        listener.blockForDone();
+        testExecutor.runAllTasks();
+        assertTrue(networkQualityListener.rttObservationCount() > 0);
+        assertTrue(networkQualityListener.throughputObservationCount() > 0);
+        mActivity.mUrlRequestContext.shutdown();
+    }
+
+    @SmallTest
+    @Feature({"Cronet"})
     public void testShutdown() throws Exception {
         mActivity = launchCronetTestApp();
         TestUrlRequestListener listener = new ShutdownTestUrlRequestListener();
         // Block listener when response starts to verify that shutdown fails
         // if there are active requests.
         listener.setAutoAdvance(false);
         UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
                 TEST_URL, listener, listener.getExecutor());
         urlRequest.start();
         try {
@@ skipping 501 matching lines @@
                 new CronetUrlRequestContext(mActivity, mActivity.getContextConfig());
         CronetUrlRequestContext secondContext = new CronetUrlRequestContext(
                 mActivity.getApplicationContext(), mActivity.getContextConfig());
         CronetUrlRequestContext thirdContext = new CronetUrlRequestContext(
                 new ContextWrapper(mActivity), mActivity.getContextConfig());
         firstContext.shutdown();
         secondContext.shutdown();
         thirdContext.shutdown();
     }
 }