OLD | NEW |
---|---|
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 package org.chromium.net; | 5 package org.chromium.net; |
6 | 6 |
7 import android.support.test.filters.SmallTest; | 7 import android.support.test.filters.SmallTest; |
8 | 8 |
9 import org.json.JSONObject; | 9 import org.json.JSONObject; |
10 | 10 |
11 import org.chromium.base.Log; | |
11 import org.chromium.base.PathUtils; | 12 import org.chromium.base.PathUtils; |
12 import org.chromium.base.test.util.Feature; | 13 import org.chromium.base.test.util.Feature; |
13 | 14 |
14 import java.io.BufferedReader; | |
15 import java.io.File; | 15 import java.io.File; |
16 import java.io.FileReader; | 16 import java.io.FileInputStream; |
17 import java.io.FileNotFoundException; | |
18 import java.io.IOException; | |
17 | 19 |
18 /** | 20 /** |
19 * Tests for experimental options. | 21 * Tests for experimental options. |
20 */ | 22 */ |
21 public class ExperimentalOptionsTest extends CronetTestBase { | 23 public class ExperimentalOptionsTest extends CronetTestBase { |
22 private static final String TAG = ExperimentalOptionsTest.class.getSimpleNam e(); | 24 private static final String TAG = ExperimentalOptionsTest.class.getSimpleNam e(); |
23 private CronetTestFramework mTestFramework; | 25 private CronetTestFramework mTestFramework; |
24 private ExperimentalCronetEngine.Builder mBuilder; | 26 private ExperimentalCronetEngine.Builder mBuilder; |
25 | 27 |
26 @Override | 28 @Override |
(...skipping 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
59 String url = Http2TestServer.getEchoMethodUrl(); | 61 String url = Http2TestServer.getEchoMethodUrl(); |
60 TestUrlRequestCallback callback = new TestUrlRequestCallback(); | 62 TestUrlRequestCallback callback = new TestUrlRequestCallback(); |
61 UrlRequest.Builder builder = mTestFramework.mCronetEngine.newUrlRequestB uilder( | 63 UrlRequest.Builder builder = mTestFramework.mCronetEngine.newUrlRequestB uilder( |
62 url, callback, callback.getExecutor()); | 64 url, callback, callback.getExecutor()); |
63 UrlRequest urlRequest = builder.build(); | 65 UrlRequest urlRequest = builder.build(); |
64 urlRequest.start(); | 66 urlRequest.start(); |
65 callback.blockForDone(); | 67 callback.blockForDone(); |
66 assertEquals(200, callback.mResponseInfo.getHttpStatusCode()); | 68 assertEquals(200, callback.mResponseInfo.getHttpStatusCode()); |
67 assertEquals("GET", callback.mResponseAsString); | 69 assertEquals("GET", callback.mResponseAsString); |
68 mTestFramework.mCronetEngine.stopNetLog(); | 70 mTestFramework.mCronetEngine.stopNetLog(); |
69 assertTrue(logfile.exists()); | 71 while (true) { |
70 assertTrue(logfile.length() != 0); | 72 Log.i(TAG, "Still waiting for netlog file....."); |
71 BufferedReader logReader = new BufferedReader(new FileReader(logfile)); | 73 Thread.sleep(10000); |
kapishnikov
2017/04/18 16:24:34
This will increase the duration of every test run
xunjieli
2017/04/18 18:30:29
Done. Great idea! Per offline discussion, I have c
| |
72 boolean validFile = false; | 74 boolean contains = false; |
73 try { | 75 if (fileContainsString(logfile, "HostResolverRules")) break; |
74 String logLine; | |
75 while ((logLine = logReader.readLine()) != null) { | |
76 if (logLine.contains("HostResolverRules")) { | |
77 validFile = true; | |
78 break; | |
79 } | |
80 } | |
81 } finally { | |
82 logReader.close(); | |
83 } | 76 } |
84 assertTrue(validFile); | |
85 assertTrue(logfile.delete()); | 77 assertTrue(logfile.delete()); |
86 assertTrue(!logfile.exists()); | 78 assertTrue(!logfile.exists()); |
87 } | 79 } |
88 | 80 |
89 @SmallTest | 81 @SmallTest |
90 @Feature({"Cronet"}) | 82 @Feature({"Cronet"}) |
91 @OnlyRunNativeCronet | 83 @OnlyRunNativeCronet |
92 public void testSetSSLKeyLogFile() throws Exception { | 84 public void testSetSSLKeyLogFile() throws Exception { |
93 String url = Http2TestServer.getEchoMethodUrl(); | 85 String url = Http2TestServer.getEchoMethodUrl(); |
94 File dir = new File(PathUtils.getDataDirectory()); | 86 File dir = new File(PathUtils.getDataDirectory()); |
95 File file = File.createTempFile("ssl_key_log_file", "", dir); | 87 File file = File.createTempFile("ssl_key_log_file", "", dir); |
96 | 88 |
97 JSONObject experimentalOptions = new JSONObject().put("ssl_key_log_file" , file.getPath()); | 89 JSONObject experimentalOptions = new JSONObject().put("ssl_key_log_file" , file.getPath()); |
98 mBuilder.setExperimentalOptions(experimentalOptions.toString()); | 90 mBuilder.setExperimentalOptions(experimentalOptions.toString()); |
99 mTestFramework = new CronetTestFramework(null, null, getContext(), mBuil der); | 91 mTestFramework = new CronetTestFramework(null, null, getContext(), mBuil der); |
100 | 92 |
101 TestUrlRequestCallback callback = new TestUrlRequestCallback(); | 93 TestUrlRequestCallback callback = new TestUrlRequestCallback(); |
102 UrlRequest.Builder builder = mTestFramework.mCronetEngine.newUrlRequestB uilder( | 94 UrlRequest.Builder builder = mTestFramework.mCronetEngine.newUrlRequestB uilder( |
103 url, callback, callback.getExecutor()); | 95 url, callback, callback.getExecutor()); |
104 UrlRequest urlRequest = builder.build(); | 96 UrlRequest urlRequest = builder.build(); |
105 urlRequest.start(); | 97 urlRequest.start(); |
106 callback.blockForDone(); | 98 callback.blockForDone(); |
107 assertEquals(200, callback.mResponseInfo.getHttpStatusCode()); | 99 assertEquals(200, callback.mResponseInfo.getHttpStatusCode()); |
108 assertEquals("GET", callback.mResponseAsString); | 100 assertEquals("GET", callback.mResponseAsString); |
109 | 101 |
110 assertTrue(file.exists()); | 102 while (true) { |
111 assertTrue(file.length() != 0); | 103 Log.i(TAG, "Still waiting for ssl key log file....."); |
112 BufferedReader logReader = new BufferedReader(new FileReader(file)); | 104 Thread.sleep(10000); |
113 boolean validFile = false; | 105 boolean contains = false; |
114 try { | 106 if (fileContainsString(file, "CLIENT_RANDOM")) break; |
115 String logLine; | |
116 while ((logLine = logReader.readLine()) != null) { | |
117 if (logLine.contains("CLIENT_RANDOM")) { | |
118 validFile = true; | |
119 break; | |
120 } | |
121 } | |
122 } finally { | |
123 logReader.close(); | |
124 } | 107 } |
125 assertTrue(validFile); | |
126 assertTrue(file.delete()); | 108 assertTrue(file.delete()); |
127 assertTrue(!file.exists()); | 109 assertTrue(!file.exists()); |
128 } | 110 } |
111 | |
112 // Returns whether a file contains a particular string. | |
113 private boolean fileContainsString(File file, String content) throws IOExcep tion { | |
114 FileInputStream fileInputStream = null; | |
115 try { | |
116 fileInputStream = new FileInputStream(file); | |
117 byte[] data = new byte[(int) file.length()]; | |
118 fileInputStream.read(data); | |
119 return new String(data, "UTF-8").contains(content); | |
120 } catch (FileNotFoundException e) { | |
121 // Ignored this exception since the file will only be created when u pdates are | |
122 // flushed to the disk. | |
123 Log.i(TAG, "file not found"); | |
124 } finally { | |
125 if (fileInputStream != null) { | |
126 fileInputStream.close(); | |
127 } | |
128 } | |
129 return false; | |
130 } | |
129 } | 131 } |
OLD | NEW |