| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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.test.suitebuilder.annotation.LargeTest; | 7 import android.test.suitebuilder.annotation.LargeTest; |
| 8 import android.test.suitebuilder.annotation.SmallTest; | 8 import android.test.suitebuilder.annotation.SmallTest; |
| 9 | 9 |
| 10 import org.chromium.base.Log; | 10 import org.chromium.base.Log; |
| (...skipping 12 matching lines...) Expand all Loading... |
| 23 public class QuicTest extends CronetTestBase { | 23 public class QuicTest extends CronetTestBase { |
| 24 private static final String TAG = "cr.QuicTest"; | 24 private static final String TAG = "cr.QuicTest"; |
| 25 private CronetTestActivity mActivity; | 25 private CronetTestActivity mActivity; |
| 26 | 26 |
| 27 @Override | 27 @Override |
| 28 protected void setUp() throws Exception { | 28 protected void setUp() throws Exception { |
| 29 super.setUp(); | 29 super.setUp(); |
| 30 // Load library first, since we need the Quic test server's URL. | 30 // Load library first, since we need the Quic test server's URL. |
| 31 System.loadLibrary("cronet_tests"); | 31 System.loadLibrary("cronet_tests"); |
| 32 QuicTestServer.startQuicTestServer(getInstrumentation().getTargetContext
()); | 32 QuicTestServer.startQuicTestServer(getInstrumentation().getTargetContext
()); |
| 33 UrlRequestContextConfig config = new UrlRequestContextConfig(); | 33 CronetEngine.Builder builder = new CronetEngine.Builder(mActivity); |
| 34 config.enableQUIC(true); | 34 builder.enableQUIC(true); |
| 35 config.addQuicHint(QuicTestServer.getServerHost(), QuicTestServer.getSer
verPort(), | 35 builder.addQuicHint(QuicTestServer.getServerHost(), QuicTestServer.getSe
rverPort(), |
| 36 QuicTestServer.getServerPort()); | 36 QuicTestServer.getServerPort()); |
| 37 config.setExperimentalQuicConnectionOptions("PACE,IW10,FOO,DEADBEEF"); | 37 builder.setExperimentalQuicConnectionOptions("PACE,IW10,FOO,DEADBEEF"); |
| 38 | 38 |
| 39 String[] commandLineArgs = {CronetTestActivity.CONFIG_KEY, config.toStri
ng(), | 39 String[] commandLineArgs = {CronetTestActivity.CONFIG_KEY, builder.toStr
ing(), |
| 40 CronetTestActivity.CACHE_KEY, CronetTestActivity.CACHE_DISK_NO_H
TTP}; | 40 CronetTestActivity.CACHE_KEY, CronetTestActivity.CACHE_DISK_NO_H
TTP}; |
| 41 mActivity = launchCronetTestAppWithUrlAndCommandLineArgs(null, commandLi
neArgs); | 41 mActivity = launchCronetTestAppWithUrlAndCommandLineArgs(null, commandLi
neArgs); |
| 42 } | 42 } |
| 43 | 43 |
| 44 @Override | 44 @Override |
| 45 protected void tearDown() throws Exception { | 45 protected void tearDown() throws Exception { |
| 46 QuicTestServer.shutdownQuicTestServer(); | 46 QuicTestServer.shutdownQuicTestServer(); |
| 47 super.tearDown(); | 47 super.tearDown(); |
| 48 } | 48 } |
| 49 | 49 |
| (...skipping 23 matching lines...) Expand all Loading... |
| 73 @LargeTest | 73 @LargeTest |
| 74 @Feature({"Cronet"}) | 74 @Feature({"Cronet"}) |
| 75 public void testQuicLoadUrl() throws Exception { | 75 public void testQuicLoadUrl() throws Exception { |
| 76 String quicURL = QuicTestServer.getServerURL() + "/simple.txt"; | 76 String quicURL = QuicTestServer.getServerURL() + "/simple.txt"; |
| 77 TestUrlRequestListener listener = new TestUrlRequestListener(); | 77 TestUrlRequestListener listener = new TestUrlRequestListener(); |
| 78 | 78 |
| 79 // Although the native stack races QUIC and SPDY for the first request, | 79 // Although the native stack races QUIC and SPDY for the first request, |
| 80 // since there is no http server running on the corresponding TCP port, | 80 // since there is no http server running on the corresponding TCP port, |
| 81 // QUIC will always succeed with a 200 (see | 81 // QUIC will always succeed with a 200 (see |
| 82 // net::HttpStreamFactoryImpl::Request::OnStreamFailed). | 82 // net::HttpStreamFactoryImpl::Request::OnStreamFailed). |
| 83 UrlRequest request = mActivity.mUrlRequestContext.createRequest( | 83 UrlRequest.Builder requestBuilder = new UrlRequest.Builder( |
| 84 quicURL, listener, listener.getExecutor()); | 84 quicURL, listener, listener.getExecutor(), mActivity.mCronetEngi
ne); |
| 85 request.start(); | 85 requestBuilder.build().start(); |
| 86 listener.blockForDone(); | 86 listener.blockForDone(); |
| 87 | 87 |
| 88 assertEquals(200, listener.mResponseInfo.getHttpStatusCode()); | 88 assertEquals(200, listener.mResponseInfo.getHttpStatusCode()); |
| 89 String expectedContent = "This is a simple text file served by QUIC.\n"; | 89 String expectedContent = "This is a simple text file served by QUIC.\n"; |
| 90 assertEquals(expectedContent, listener.mResponseAsString); | 90 assertEquals(expectedContent, listener.mResponseAsString); |
| 91 assertEquals("quic/1+spdy/3", listener.mResponseInfo.getNegotiatedProtoc
ol()); | 91 assertEquals("quic/1+spdy/3", listener.mResponseInfo.getNegotiatedProtoc
ol()); |
| 92 // The total received bytes should be larger than the content length, to
account for | 92 // The total received bytes should be larger than the content length, to
account for |
| 93 // headers. | 93 // headers. |
| 94 assertTrue( | 94 assertTrue( |
| 95 listener.mExtendedResponseInfo.getTotalReceivedBytes() > expecte
dContent.length()); | 95 listener.mExtendedResponseInfo.getTotalReceivedBytes() > expecte
dContent.length()); |
| 96 | 96 |
| 97 // This test takes a long time, since the update will only be scheduled | 97 // This test takes a long time, since the update will only be scheduled |
| 98 // after kUpdatePrefsDelayMs in http_server_properties_manager.cc. | 98 // after kUpdatePrefsDelayMs in http_server_properties_manager.cc. |
| 99 while (true) { | 99 while (true) { |
| 100 Log.i(TAG, "Still waiting for pref file update....."); | 100 Log.i(TAG, "Still waiting for pref file update....."); |
| 101 Thread.sleep(10000); | 101 Thread.sleep(10000); |
| 102 boolean contains = false; | 102 boolean contains = false; |
| 103 try { | 103 try { |
| 104 if (fileContainsString("local_prefs.json", "quic")) break; | 104 if (fileContainsString("local_prefs.json", "quic")) break; |
| 105 } catch (FileNotFoundException e) { | 105 } catch (FileNotFoundException e) { |
| 106 // Ignored this exception since the file will only be created wh
en updates are | 106 // Ignored this exception since the file will only be created wh
en updates are |
| 107 // flushed to the disk. | 107 // flushed to the disk. |
| 108 } | 108 } |
| 109 } | 109 } |
| 110 assertTrue(fileContainsString("local_prefs.json", | 110 assertTrue(fileContainsString("local_prefs.json", |
| 111 QuicTestServer.getServerHost() + ":" + QuicTestServer.getServerP
ort())); | 111 QuicTestServer.getServerHost() + ":" + QuicTestServer.getServerP
ort())); |
| 112 mActivity.mUrlRequestContext.shutdown(); | 112 mActivity.mCronetEngine.shutdown(); |
| 113 | 113 |
| 114 // Make another request using a new context but with no QUIC hints. | 114 // Make another request using a new context but with no QUIC hints. |
| 115 UrlRequestContextConfig config = new UrlRequestContextConfig(); | 115 CronetEngine.Builder builder = |
| 116 config.setStoragePath(mActivity.getTestStorage()); | 116 new CronetEngine.Builder(getInstrumentation().getTargetContext()
); |
| 117 config.enableHttpCache(UrlRequestContextConfig.HTTP_CACHE_DISK, 1000 * 1
024); | 117 builder.setStoragePath(mActivity.getTestStorage()); |
| 118 config.enableQUIC(true); | 118 builder.enableHttpCache(CronetEngine.Builder.HTTP_CACHE_DISK, 1000 * 102
4); |
| 119 CronetUrlRequestContext newContext = | 119 builder.enableQUIC(true); |
| 120 new CronetUrlRequestContext(getInstrumentation().getTargetContex
t(), config); | 120 CronetEngine newEngine = new CronetUrlRequestContext(builder); |
| 121 TestUrlRequestListener listener2 = new TestUrlRequestListener(); | 121 TestUrlRequestListener listener2 = new TestUrlRequestListener(); |
| 122 UrlRequest request2 = newContext.createRequest(quicURL, listener2, liste
ner2.getExecutor()); | 122 requestBuilder = |
| 123 request2.start(); | 123 new UrlRequest.Builder(quicURL, listener2, listener2.getExecutor
(), newEngine); |
| 124 requestBuilder.build().start(); |
| 124 listener2.blockForDone(); | 125 listener2.blockForDone(); |
| 125 assertEquals(200, listener2.mResponseInfo.getHttpStatusCode()); | 126 assertEquals(200, listener2.mResponseInfo.getHttpStatusCode()); |
| 126 assertEquals(expectedContent, listener2.mResponseAsString); | 127 assertEquals(expectedContent, listener2.mResponseAsString); |
| 127 assertEquals("quic/1+spdy/3", listener2.mResponseInfo.getNegotiatedProto
col()); | 128 assertEquals("quic/1+spdy/3", listener2.mResponseInfo.getNegotiatedProto
col()); |
| 128 // The total received bytes should be larger than the content length, to
account for | 129 // The total received bytes should be larger than the content length, to
account for |
| 129 // headers. | 130 // headers. |
| 130 assertTrue( | 131 assertTrue( |
| 131 listener2.mExtendedResponseInfo.getTotalReceivedBytes() > expect
edContent.length()); | 132 listener2.mExtendedResponseInfo.getTotalReceivedBytes() > expect
edContent.length()); |
| 132 } | 133 } |
| 133 | 134 |
| 134 // Returns whether a file contains a particular string. | 135 // Returns whether a file contains a particular string. |
| 135 @SuppressFBWarnings("OBL_UNSATISFIED_OBLIGATION_EXCEPTION_EDGE") | 136 @SuppressFBWarnings("OBL_UNSATISFIED_OBLIGATION_EXCEPTION_EDGE") |
| 136 private boolean fileContainsString(String filename, String content) throws I
OException { | 137 private boolean fileContainsString(String filename, String content) throws I
OException { |
| 137 File file = new File(mActivity.getTestStorage() + "/" + filename); | 138 File file = new File(mActivity.getTestStorage() + "/" + filename); |
| 138 FileInputStream fileInputStream = new FileInputStream(file); | 139 FileInputStream fileInputStream = new FileInputStream(file); |
| 139 byte[] data = new byte[(int) file.length()]; | 140 byte[] data = new byte[(int) file.length()]; |
| 140 fileInputStream.read(data); | 141 fileInputStream.read(data); |
| 141 fileInputStream.close(); | 142 fileInputStream.close(); |
| 142 return new String(data, "UTF-8").contains(content); | 143 return new String(data, "UTF-8").contains(content); |
| 143 } | 144 } |
| 144 } | 145 } |
| OLD | NEW |