Allow setting Cronet NetLog level to LOG_ALL

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;
 
 /**
  * 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 241 matching lines @@
     public void testNetLog() throws Exception {
         Context context = getInstrumentation().getTargetContext();
         File directory = new File(PathUtils.getDataDirectory(context));
         File file = File.createTempFile("cronet", "json", directory);
         CronetUrlRequestContext requestContext = new CronetUrlRequestContext(
                 context,
                 new UrlRequestContextConfig().setLibraryName("cronet_tests"));
         // Start NetLog immediately after the request context is created to make
         // sure that the call won't crash the app even when the native request
         // context is not fully initialized. See crbug.com/470196.
-        requestContext.startNetLogToFile(file.getPath());
+        requestContext.startNetLogToFile(file.getPath(), false);
 
         // Start a request.
         TestUrlRequestListener listener = new TestUrlRequestListener();
         UrlRequest request = requestContext.createRequest(
                 TEST_URL, listener, listener.getExecutor());
         request.start();
         listener.blockForDone();
         requestContext.stopNetLog();
         assertTrue(file.exists());
         assertTrue(file.length() != 0);
+        assertFalse(hasBytesInNetLog(file));
         assertTrue(file.delete());
         assertTrue(!file.exists());
     }
 
     @SmallTest
     @Feature({"Cronet"})
     public void testNetLogAfterShutdown() throws Exception {
         mActivity = launchCronetTestApp();
         TestUrlRequestListener listener = new TestUrlRequestListener();
         UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
                 TEST_URL, listener, listener.getExecutor());
         urlRequest.start();
         listener.blockForDone();
         mActivity.mUrlRequestContext.shutdown();
 
         File directory = new File(PathUtils.getDataDirectory(
                 getInstrumentation().getTargetContext()));
         File file = File.createTempFile("cronet", "json", directory);
         try {
-            mActivity.mUrlRequestContext.startNetLogToFile(file.getPath());
+            mActivity.mUrlRequestContext.startNetLogToFile(file.getPath(),
+                    false);
             fail("Should throw an exception.");
         } catch (Exception e) {
             assertEquals("Context is shut down.", e.getMessage());
         }
+        assertFalse(hasBytesInNetLog(file));
         assertTrue(file.delete());
         assertTrue(!file.exists());
     }
 
     @SmallTest
     @Feature({"Cronet"})
     public void testNetLogStartMultipleTimes() throws Exception {
         mActivity = launchCronetTestApp();
         File directory = new File(PathUtils.getDataDirectory(
                 getInstrumentation().getTargetContext()));
         File file = File.createTempFile("cronet", "json", directory);
         // Start NetLog multiple times.
-        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath());
-        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath());
-        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath());
-        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath());
+        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath(), false);
+        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath(), false);
+        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath(), false);
+        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath(), false);
         // Start a request.
         TestUrlRequestListener listener = new TestUrlRequestListener();
         UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
                 TEST_URL, listener, listener.getExecutor());
         urlRequest.start();
         listener.blockForDone();
         mActivity.mUrlRequestContext.stopNetLog();
         assertTrue(file.exists());
         assertTrue(file.length() != 0);
+        assertFalse(hasBytesInNetLog(file));
         assertTrue(file.delete());
         assertTrue(!file.exists());
     }
 
     @SmallTest
     @Feature({"Cronet"})
     public void testNetLogStopMultipleTimes() throws Exception {
         mActivity = launchCronetTestApp();
         File directory = new File(PathUtils.getDataDirectory(
                 getInstrumentation().getTargetContext()));
         File file = File.createTempFile("cronet", "json", directory);
-        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath());
+        mActivity.mUrlRequestContext.startNetLogToFile(file.getPath(), false);
         // Start a request.
         TestUrlRequestListener listener = new TestUrlRequestListener();
         UrlRequest urlRequest = mActivity.mUrlRequestContext.createRequest(
                 TEST_URL, listener, listener.getExecutor());
         urlRequest.start();
         listener.blockForDone();
         // Stop NetLog multiple times.
         mActivity.mUrlRequestContext.stopNetLog();
         mActivity.mUrlRequestContext.stopNetLog();
         mActivity.mUrlRequestContext.stopNetLog();
         mActivity.mUrlRequestContext.stopNetLog();
         mActivity.mUrlRequestContext.stopNetLog();
         assertTrue(file.exists());
         assertTrue(file.length() != 0);
+        assertFalse(hasBytesInNetLog(file));
         assertTrue(file.delete());
         assertTrue(!file.exists());
     }
 
+    @SmallTest
+    @Feature({"Cronet"})
+    public void testNetLogWithBytes() throws Exception {
+        Context context = getInstrumentation().getTargetContext();
+        File directory = new File(PathUtils.getDataDirectory(context));
+        File file = File.createTempFile("cronet", "json", directory);
+        CronetUrlRequestContext requestContext = new CronetUrlRequestContext(
+                context,
+                new UrlRequestContextConfig().setLibraryName("cronet_tests"));
+        // Start NetLog with logAll as true.
+        requestContext.startNetLogToFile(file.getPath(), true);
+        // Start a request.
+        TestUrlRequestListener listener = new TestUrlRequestListener();
+        UrlRequest request = requestContext.createRequest(
+                TEST_URL, listener, listener.getExecutor());
+        request.start();
+        listener.blockForDone();
+        requestContext.stopNetLog();
+        assertTrue(file.exists());
+        assertTrue(file.length() != 0);
+        assertTrue(hasBytesInNetLog(file));
+        assertTrue(file.delete());
+        assertTrue(!file.exists());
+    }
+
+    private boolean hasBytesInNetLog(File logFile) throws Exception {
+        BufferedReader logReader = new BufferedReader(new FileReader(logFile));
+        try {
+            String logLine;
+            while ((logLine = logReader.readLine()) != null) {
+                if (logLine.contains("\"hex_encoded_bytes\"")) {
+                    return true;
+                }
+            }
+            return false;
+        } finally {
+            logReader.close();
+        }
+    }
+
     private void enableCache(UrlRequestContextConfig.HttpCache cacheType)
             throws Exception {
         UrlRequestContextConfig config = new UrlRequestContextConfig();
         config.setLibraryName("cronet_tests");
         if (cacheType == UrlRequestContextConfig.HttpCache.DISK
                 || cacheType == UrlRequestContextConfig.HttpCache.DISK_NO_HTTP) {
             config.setStoragePath(prepareTestStorage());
         }
 
         config.enableHttpCache(cacheType, 1000 * 1024);
@@ skipping 207 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();
     }
 }