URLConnectionTest.java revision f02c695ed03e708623d9365dec26d533356ef2d0
1e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes/* 2e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * Copyright (C) 2009 The Android Open Source Project 3f33eae7e84eb6d3b0f4e86b59605bb3de73009f3Elliott Hughes * 4e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * Licensed under the Apache License, Version 2.0 (the "License"); 5e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * you may not use this file except in compliance with the License. 6e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * You may obtain a copy of the License at 7f33eae7e84eb6d3b0f4e86b59605bb3de73009f3Elliott Hughes * 8e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * http://www.apache.org/licenses/LICENSE-2.0 9f33eae7e84eb6d3b0f4e86b59605bb3de73009f3Elliott Hughes * 10e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * Unless required by applicable law or agreed to in writing, software 11e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * distributed under the License is distributed on an "AS IS" BASIS, 12e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * See the License for the specific language governing permissions and 14e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * limitations under the License. 15e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes */ 16e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 174557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonpackage libcore.java.net; 18e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 1951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.io.BufferedReader; 20c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilsonimport java.io.ByteArrayOutputStream; 21f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstromimport java.io.FileNotFoundException; 226247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.io.IOException; 2351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.io.InputStream; 24e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughesimport java.io.InputStreamReader; 2502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.io.OutputStream; 264557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.Authenticator; 274557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.CacheRequest; 284557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.CacheResponse; 294557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpRetryException; 304557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpURLConnection; 3100feece22909b7dc79fc96d666d157390b93858eJesse Wilsonimport java.net.InetAddress; 324557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.PasswordAuthentication; 33f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstromimport java.net.Proxy; 344557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.ResponseCache; 3537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilsonimport java.net.SecureCacheResponse; 36f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstromimport java.net.ServerSocket; 37f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstromimport java.net.Socket; 384557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.SocketTimeoutException; 394557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URI; 404557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URISyntaxException; 414557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URL; 424557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URLConnection; 4337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilsonimport java.security.Principal; 4437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilsonimport java.security.cert.Certificate; 45c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.CertificateException; 46c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.X509Certificate; 4751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.ArrayList; 4802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.util.Arrays; 4951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Collections; 5083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.HashSet; 5151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Iterator; 526247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.util.List; 5383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.Map; 5451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Set; 5583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.concurrent.atomic.AtomicReference; 56deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPInputStream; 57deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPOutputStream; 5860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HostnameVerifier; 5960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HttpsURLConnection; 60c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLContext; 61096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilsonimport javax.net.ssl.SSLException; 6260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.SSLSession; 63c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLSocketFactory; 64c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.TrustManager; 65c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.X509TrustManager; 6650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilsonimport libcore.javax.net.ssl.TestSSLContext; 6751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport tests.http.DefaultResponseCache; 6860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockResponse; 6960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockWebServer; 7060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.RecordedRequest; 71e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 72e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughespublic class URLConnectionTest extends junit.framework.TestCase { 73b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 74ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private static final Authenticator SIMPLE_AUTHENTICATOR = new Authenticator() { 75ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson protected PasswordAuthentication getPasswordAuthentication() { 76ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson return new PasswordAuthentication("username", "password".toCharArray()); 77ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 78ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson }; 79ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 8051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockWebServer server = new MockWebServer(); 8100feece22909b7dc79fc96d666d157390b93858eJesse Wilson private String hostname; 8200feece22909b7dc79fc96d666d157390b93858eJesse Wilson 8300feece22909b7dc79fc96d666d157390b93858eJesse Wilson @Override protected void setUp() throws Exception { 8400feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.setUp(); 8500feece22909b7dc79fc96d666d157390b93858eJesse Wilson hostname = InetAddress.getLocalHost().getHostName(); 8600feece22909b7dc79fc96d666d157390b93858eJesse Wilson } 8751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 8851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson @Override protected void tearDown() throws Exception { 8951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(null); 90ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(null); 91984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyHost"); 92984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyPort"); 93984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyHost"); 94984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyPort"); 95984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyHost"); 96984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyPort"); 9751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); 9800feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.tearDown(); 9951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 10051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 10183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testRequestHeaders() throws IOException, InterruptedException { 10283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse()); 10383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 10483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 10583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 10683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "e"); 10783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "f"); 10883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders = urlConnection.getRequestProperties(); 10983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("e", "f"), new HashSet<String>(requestHeaders.get("D"))); 11083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.put("G", Arrays.asList("h")); 11283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 11383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 11483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 11583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.get("D").add("i"); 11783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 11883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 11983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 12183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty(null, "j"); 12283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 12383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 12483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 12683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty(null, "k"); 12783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 12883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 12983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("NullValue", null); // should fail silently! 13183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("AnotherNullValue", null); // should fail silently! 13283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 13383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.getResponseCode(); 13483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson RecordedRequest request = server.takeRequest(); 13583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: e"); 13683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: f"); 13783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "NullValue.*"); 13883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "AnotherNullValue.*"); 13983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "G:.*"); 14083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "null:.*"); 14183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 14283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 14383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("N", "o"); 14483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 14583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 14683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 14783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 14883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("P", "q"); 14983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 15083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 15183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 15283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 15383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 15483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseHeaders() throws IOException, InterruptedException { 15583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse() 15683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setStatus("HTTP/1.0 200 Fantastic") 15783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: b") 15883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: c") 15983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8)); 16083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 16183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 16283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 16383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 16483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 165c1a675c80c69decadb736b245f0366f93a94a462Jesse Wilson assertEquals("HTTP/1.0 200 Fantastic", urlConnection.getHeaderField(null)); 16683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> responseHeaders = urlConnection.getHeaderFields(); 1678ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(Arrays.asList("HTTP/1.0 200 Fantastic"), responseHeaders.get(null)); 16883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("b", "c"), new HashSet<String>(responseHeaders.get("A"))); 16983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 17083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.put("N", Arrays.asList("o")); 17183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 17283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 17383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 17483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 17583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.get("A").add("d"); 17683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 17783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 17883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 17983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 18083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 181e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // Check that if we don't read to the end of a response, the next request on the 182e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // recycled connection doesn't get the unread tail of the first request's response. 183e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // http://code.google.com/p/android/issues/detail?id=2939 184e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes public void test_2939() throws Exception { 185b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8); 186b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 187b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 188b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 189b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 190b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 191c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 192c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 1938baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson } 1948baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson 195977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // Check that we recognize a few basic mime types by extension. 196977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // http://code.google.com/p/android/issues/detail?id=10100 197977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes public void test_10100() throws Exception { 198977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("image/jpeg", URLConnection.guessContentTypeFromName("someFile.jpg")); 199977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("application/pdf", URLConnection.guessContentTypeFromName("stuff.pdf")); 200977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes } 201977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes 2028baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson public void testConnectionsArePooled() throws Exception { 203b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setBody("ABCDEFGHIJKLMNOPQR"); 204b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 205b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 206b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 207b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 208b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 209b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 21006e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/foo").openConnection()); 211c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 21206e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/bar?baz=quux").openConnection()); 213c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 21406e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/z").openConnection()); 215c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 216c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 217c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 218c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testChunkedConnectionsArePooled() throws Exception { 219c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDEFGHIJKLMNOPQR", 5); 220c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 221c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 222c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 223c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 224c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 225c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 22606e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/foo").openConnection()); 227b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 22806e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/bar?baz=quux").openConnection()); 229b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 23006e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/z").openConnection()); 231b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 232e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes } 23302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 234b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson enum WriteKind { BYTE_BY_BYTE, SMALL_BUFFERS, LARGE_BUFFERS } 23502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 23602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_byteByByte() throws Exception { 23751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.BYTE_BY_BYTE); 23802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 24002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_smallBuffers() throws Exception { 24151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.SMALL_BUFFERS); 24202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 24302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 24402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_largeBuffers() throws Exception { 24551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.LARGE_BUFFERS); 24602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 24702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 24802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_byteByByte() throws Exception { 24951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.BYTE_BY_BYTE); 25002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 25102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 25202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_smallBuffers() throws Exception { 25351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.SMALL_BUFFERS); 25402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 25502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 25602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_largeBuffers() throws Exception { 25751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.LARGE_BUFFERS); 25802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 25902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 26051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void doUpload(TransferKind uploadKind, WriteKind writeKind) throws Exception { 26102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes int n = 512*1024; 262b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.setBodyLimit(0); 263b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse()); 264b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 265b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 266b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson HttpURLConnection conn = (HttpURLConnection) server.getUrl("/").openConnection(); 26702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setDoOutput(true); 26802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setRequestMethod("POST"); 26951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 27002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setChunkedStreamingMode(-1); 27102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 27202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setFixedLengthStreamingMode(n); 27302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes OutputStream out = conn.getOutputStream(); 27502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes if (writeKind == WriteKind.BYTE_BY_BYTE) { 27602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; ++i) { 27702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write('x'); 27802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 28002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes byte[] buf = new byte[writeKind == WriteKind.SMALL_BUFFERS ? 256 : 64*1024]; 28102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes Arrays.fill(buf, (byte) 'x'); 28202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; i += buf.length) { 28302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write(buf, 0, Math.min(buf.length, n - i)); 28402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 28502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 28602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.close(); 2874cb7f05dc68abb23ae54a5891c369062185f2210Elliott Hughes assertEquals(200, conn.getResponseCode()); 288b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson RecordedRequest request = server.takeRequest(); 289b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(n, request.getBodySize()); 29051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 291b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().size() > 0); 292b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } else { 293b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().isEmpty()); 294b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } 29502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 2966247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 29751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 29851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Test that response caching is consistent with the RI and the spec. 29951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.4 30051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 3016247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching() throws Exception { 3026247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // Test each documented HTTP/1.1 code, plus the first unused value in each range. 3036247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html 3046247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 3056247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // We can't test 100 because it's not really a response. 3066247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // assertCached(false, 100); 3076247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 101); 3086247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 102); 3096247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 200); 3106247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 201); 3116247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 202); 3126247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 203); 3136247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 204); 3146247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 205); 3156247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 206); 3166247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 207); 31751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_300.) 3186247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 301); 3196247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 302; i <= 308; ++i) { 3206247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3216247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3226247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 400; i <= 406; ++i) { 3236247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3246247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3256247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // (See test_responseCaching_407.) 3266247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 408); 3276247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 409); 32851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_410.) 3296247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 411; i <= 418; ++i) { 3306247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3316247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3326247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 500; i <= 506; ++i) { 3336247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3346247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3356247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3366247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 33751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_300() throws Exception { 33851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: fix this for android 33951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 300); 34051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 34151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3421f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson /** 3431f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * Response code 407 should only come from proxy servers. Android's client 3441f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * throws if it is sent by an origin server. 3451f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson */ 3461f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson public void testOriginServerSends407() throws Exception { 3471f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.enqueue(new MockResponse().setResponseCode(407)); 3481f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.play(); 3491f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson 3501f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson URL url = server.getUrl("/"); 3511f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3521f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson try { 3531f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson conn.getResponseCode(); 3541f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson fail(); 3551f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } catch (IOException expected) { 3561f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } 3576247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3586247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 35951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_410() throws Exception { 36051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // the HTTP spec permits caching 410s, but the RI doesn't. 36151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 410); 36251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 36351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3646247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes private void assertCached(boolean shouldPut, int responseCode) throws Exception { 36551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server = new MockWebServer(); 366b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse() 367b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .setResponseCode(responseCode) 36851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("ABCDE") 369b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .addHeader("WWW-Authenticate: challenge")); 370b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 371b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 37251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache responseCache = new DefaultResponseCache(); 37351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(responseCache); 37451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 37551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3766247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertEquals(responseCode, conn.getResponseCode()); 3776247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 37851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // exhaust the content stream 37951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 38051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: remove special case once testUnauthorizedResponseHandling() is fixed 38151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (responseCode != 401) { 38251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(conn.getInputStream(), Integer.MAX_VALUE); 38351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 38451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException ignored) { 38551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 38651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 38751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson Set<URI> expectedCachedUris = shouldPut 38851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ? Collections.singleton(url.toURI()) 38951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson : Collections.<URI>emptySet(); 39051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(Integer.toString(responseCode), 39151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson expectedCachedUris, responseCache.getContents().keySet()); 39251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); // tearDown() isn't sufficient; this test starts multiple servers 3936247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 39460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 39560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttps() throws IOException, InterruptedException { 39660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 39760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 398059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 399c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 40060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson server.play(); 40160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 402096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 4034559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 40460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 405c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via HTTPS", connection); 40660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 40760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson RecordedRequest request = server.takeRequest(); 40860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 40960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 41060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 411096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson public void testConnectViaHttpsReusingConnections() throws IOException, InterruptedException { 412096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 413096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 414059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 415096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 416096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 417096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.play(); 418096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 419b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 420b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 421b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("this response comes via HTTPS", connection); 422b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 423b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 424b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 425b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("another response via HTTPS", connection); 426b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 427b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 428b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 429b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson } 430b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 4318116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson public void testConnectViaHttpsReusingConnectionsDifferentFactories() 432b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson throws IOException, InterruptedException { 433b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 434b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 435b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 436b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 437b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 438b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.play(); 439b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 440096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson // install a custom SSL socket factory so the server can be authorized 441096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 442059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 443096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertContent("this response comes via HTTPS", connection); 444096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 445096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 446096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson try { 447096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 448b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson fail("without an SSL socket factory, the connection should fail"); 449096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } catch (SSLException expected) { 450096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 451096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 452096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 4534559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom public void testConnectViaHttpsWithSSLFallback() throws IOException, InterruptedException { 4544559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 4554559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4564559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 4574559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setDisconnectAtStart(true)); 4584559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via SSL")); 4594559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.play(); 4604559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4614559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 4624559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 4634559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4644559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertContent("this response comes via SSL", connection); 4654559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4664559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom RecordedRequest request = server.takeRequest(); 4674559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 4684559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom } 4694559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 470984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxyArg() throws Exception { 471984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.CREATE_ARG); 472984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 473984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 474984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxySystemProperty() throws Exception { 475984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.PROXY_SYSTEM_PROPERTY); 476984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 477984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 478984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingHttpProxySystemProperty() throws Exception { 479984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 480984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 481984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 482984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaProxy(ProxyConfig proxyConfig) throws Exception { 483c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse mockResponse = new MockResponse().setBody("this response comes via a proxy"); 48451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(mockResponse); 48551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 48660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 487984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson URL url = new URL("http://android.com/foo"); 488984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpURLConnection connection = proxyConfig.connect(server, url); 489c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a proxy", connection); 49060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 49151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest request = server.takeRequest(); 49260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET http://android.com/foo HTTP/1.1", request.getRequestLine()); 49360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(request.getHeaders(), "Host: android.com"); 49460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 49560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 496c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithContentLengthHeader() throws IOException { 497c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(new MockResponse() 498c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .setBody("abc\r\nYOU SHOULD NOT SEE THIS") 499c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .clearHeaders() 500c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .addHeader("Content-Length: 3")); 501c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 502c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 503c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 504c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 505c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 506c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithChunkedHeader() throws IOException { 507c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse mockResponse = new MockResponse(); 508c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setChunkedBody("abc", 3); 509c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 510c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write(mockResponse.getBody()); 511c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write("\r\nYOU SHOULD NOT SEE THIS".getBytes()); 512c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setBody(bytesOut.toByteArray()); 513c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.clearHeaders(); 514c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.addHeader("Transfer-encoding: chunked"); 515c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 516c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(mockResponse); 517c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 518c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 519c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 520c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 521c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 522f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom public void testConnectViaHttpProxyToHttpsUsingProxyArgWithNoProxy() throws Exception { 523f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom testConnectViaDirectProxyToHttps(ProxyConfig.NO_PROXY); 524f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 525f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 526f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom public void testConnectViaHttpProxyToHttpsUsingHttpProxySystemProperty() throws Exception { 527f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom // https should not use http proxy 528f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom testConnectViaDirectProxyToHttps(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 529f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 530f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 531f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom private void testConnectViaDirectProxyToHttps(ProxyConfig proxyConfig) throws Exception { 532f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 533f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 534f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 535f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 536f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.play(); 537f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 538f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom URL url = server.getUrl("/foo"); 539f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) proxyConfig.connect(server, url); 540f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 541f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 542f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom assertContent("this response comes via HTTPS", connection); 543f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 544f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom RecordedRequest request = server.takeRequest(); 545f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 546f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 547f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 548f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 549984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxyArg() throws Exception { 550984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.CREATE_ARG); 551984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 552984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 553984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 554984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We weren't honoring all of the appropriate proxy system properties when 555984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * connecting via HTTPS. http://b/3097518 556984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 557984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxySystemProperty() throws Exception { 558984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.PROXY_SYSTEM_PROPERTY); 559984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 560984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 561984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingHttpsProxySystemProperty() throws Exception { 562984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.HTTPS_PROXY_SYSTEM_PROPERTY); 563984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 564984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 565984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 566984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We were verifying the wrong hostname when connecting to an HTTPS site 567984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * through a proxy. http://b/3097277 568984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 569984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaHttpProxyToHttps(ProxyConfig proxyConfig) throws Exception { 57060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 571984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 57260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 573059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 574c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 575c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via a secure proxy")); 57651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 57760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 57860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URL url = new URL("https://android.com/foo"); 579984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) proxyConfig.connect(server, url); 580059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 581984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 58260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 583c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a secure proxy", connection); 58460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 58551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest connect = server.takeRequest(); 58660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("Connect line failure on proxy", 58760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson "CONNECT android.com:443 HTTP/1.1", connect.getRequestLine()); 58860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 58960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 59051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest get = server.takeRequest(); 59160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", get.getRequestLine()); 59260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(get.getHeaders(), "Host: android.com"); 593984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 59460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 59560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 596d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson /** 597d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson * Test which headers are sent unencrypted to the HTTP proxy. 598d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson */ 599d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testProxyConnectIncludesProxyHeadersOnly() 600d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson throws IOException, InterruptedException { 601984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 602d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 603d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 604d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 605d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 606d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("encrypted response from the origin server")); 607d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 608d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 609d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson URL url = new URL("https://android.com/foo"); 610d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) url.openConnection( 611d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.toProxyAddress()); 612d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Private", "Secret"); 613d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Proxy-Authorization", "bar"); 614d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("User-Agent", "baz"); 615d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 616984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 617d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContent("encrypted response from the origin server", connection); 618d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 619d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest connect = server.takeRequest(); 620d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContainsNoneMatching(connect.getHeaders(), "Private.*"); 621d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Authorization: bar"); 622d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "User-Agent: baz"); 623d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 624d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Connection: Keep-Alive"); 625d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 626d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest get = server.takeRequest(); 627d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(get.getHeaders(), "Private: Secret"); 628984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 629d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 630d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 631d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testDisconnectedConnection() throws IOException { 632d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("ABCDEFGHIJKLMNOPQR")); 633d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 634d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 635d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 636d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson InputStream in = connection.getInputStream(); 637d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertEquals('A', (char) in.read()); 638d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.disconnect(); 639d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson try { 640d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson in.read(); 641d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson fail("Expected a connection closed exception"); 642d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } catch (IOException expected) { 643d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 644d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 645d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 64651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithFixedLength() throws IOException { 64751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.FIXED_LENGTH); 64851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 64951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithChunkedEncoding() throws IOException { 65151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.CHUNKED); 65251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 65351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithNoLengthHeaders() throws IOException { 65551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.END_OF_STREAM); 65651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 65751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 65951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * HttpURLConnection.getInputStream().skip(long) causes ResponseCache corruption 66051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://code.google.com/p/android/issues/detail?id=8175 66151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 66251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testResponseCaching(TransferKind transferKind) throws IOException { 6638ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson MockResponse response = new MockResponse() 6648ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setStatus("HTTP/1.1 200 Fantastic"); 66551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "I love puppies but hate spiders", 1); 66651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 66751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 66851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 66951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 67051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 67151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 67251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // Make sure that calling skip() doesn't omit bytes from the cache. 6738ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 67451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = urlConnection.getInputStream(); 67551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love ", readAscii(in, "I love ".length())); 67651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reliableSkip(in, "puppies but hate ".length()); 67751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("spiders", readAscii(in, "spiders".length())); 67851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 67951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 68051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 681096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 68251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 6838ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); // cached! 68451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in = urlConnection.getInputStream(); 68551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love puppies but hate spiders", 68651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(in, "I love puppies but hate spiders".length())); 6878ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 6888ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 6898ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 69051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 69151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getMissCount()); 69251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 693096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(1, cache.getSuccessCount()); 694096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 69551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 69651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 69737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson public void testSecureResponseCaching() throws IOException { 69837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 69937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 70037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 70137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.play(); 70237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 70337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 70437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache.setDefault(cache); 70537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 70637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 70737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 70837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 70937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 71037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson // OpenJDK 6 fails on this line, complaining that the connection isn't open yet 71137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson String suite = connection.getCipherSuite(); 71237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson List<Certificate> localCerts = toListOrNull(connection.getLocalCertificates()); 71337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson List<Certificate> serverCerts = toListOrNull(connection.getServerCertificates()); 71437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Principal peerPrincipal = connection.getPeerPrincipal(); 71537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Principal localPrincipal = connection.getLocalPrincipal(); 71637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 71737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // cached! 71837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 71937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 72037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 72137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(1, cache.getMissCount()); 72237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(1, cache.getHitCount()); 72337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 72437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(suite, connection.getCipherSuite()); 72537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(localCerts, toListOrNull(connection.getLocalCertificates())); 72637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(serverCerts, toListOrNull(connection.getServerCertificates())); 72737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(peerPrincipal, connection.getPeerPrincipal()); 72837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(localPrincipal, connection.getLocalPrincipal()); 72937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 73037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 73137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson public void testCacheReturnsInsecureResponseForSecureRequest() throws IOException { 73237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 73337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 73437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 73537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 73637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.play(); 73737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 73837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache insecureResponseCache = new InsecureResponseCache(); 73937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache.setDefault(insecureResponseCache); 74037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 74137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 74237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 74337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 74437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 74537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // not cached! 74637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 74737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("DEF", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 74837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 74937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 7508ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson public void testResponseCachingAndRedirects() throws IOException { 7518ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse() 7528ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_PERM) 7538ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .addHeader("Location: /foo")); 7548ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 7558ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 7568ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.play(); 7578ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7588ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 7598ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson ResponseCache.setDefault(cache); 7608ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7618ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 7628ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 7638ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7648ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection = (HttpURLConnection) server.getUrl("/").openConnection(); // cached! 7658ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 7668ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7678ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getMissCount()); // 1 redirect + 1 final response = 2 7688ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getHitCount()); 7698ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson } 7708ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7718ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson public void testSecureResponseCachingAndRedirects() throws IOException { 7728ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 7738ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 7748ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse() 7758ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_PERM) 7768ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .addHeader("Location: /foo")); 7778ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 7788ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 7798ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.play(); 7808ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7818ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 7828ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson ResponseCache.setDefault(cache); 7838ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7848ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 7858ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 7868ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 7878ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7888ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // cached! 7898ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 7908ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 7918ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 7928ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getMissCount()); // 1 redirect + 1 final response = 2 7938ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getHitCount()); 7948ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson } 7958ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 79683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseCacheRequestHeaders() throws IOException, URISyntaxException { 79783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 79883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 79983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 80083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson final AtomicReference<Map<String, List<String>>> requestHeadersRef 80183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson = new AtomicReference<Map<String, List<String>>>(); 80283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 80383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 80483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 80583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeadersRef.set(requestHeaders); 80683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 80783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 80883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection conn) throws IOException { 80983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 81083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 81183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson }); 81283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 81383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URL url = server.getUrl("/"); 81483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URLConnection urlConnection = url.openConnection(); 81583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("A", "android"); 81683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson readAscii(urlConnection.getInputStream(), Integer.MAX_VALUE); 81783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(Arrays.asList("android"), requestHeadersRef.get().get("A")); 81883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 81983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 82051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void reliableSkip(InputStream in, int length) throws IOException { 82151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson while (length > 0) { 82251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson length -= in.skip(length); 82351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 82451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 82551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 82651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 82751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Reads {@code count} characters from the stream. If the stream is 82851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * exhausted before {@code count} characters can be read, the remaining 82951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * characters are returned and the stream is closed. 83051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 83151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private String readAscii(InputStream in, int count) throws IOException { 83251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson StringBuilder result = new StringBuilder(); 83351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (int i = 0; i < count; i++) { 83451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson int value = in.read(); 83551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (value == -1) { 83651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 83751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 83851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 83951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson result.append((char) value); 84051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 84151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return result.toString(); 84251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 84351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 84451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithContentLengthHeader() throws IOException { 84551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.FIXED_LENGTH); 84651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 84751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 84851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithChunkedEncoding() throws IOException { 84951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.CHUNKED); 85051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 85151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 85251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithNoLengthHeaders() throws IOException { 85351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /* 85451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Intentionally empty. This case doesn't make sense because there's no 85551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * such thing as a premature disconnect when the disconnect itself 85651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * indicates the end of the data stream. 85751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 85851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 85951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 86051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testServerPrematureDisconnect(TransferKind transferKind) throws IOException { 86151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 86251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 16); 86351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(truncateViolently(response, 16)); 86451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 86551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 86651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 86751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 86851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 86951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 87051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson BufferedReader reader = new BufferedReader(new InputStreamReader( 87151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.getUrl("/").openConnection().getInputStream())); 87251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", reader.readLine()); 87351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 87451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reader.readLine(); 87551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("This implementation silently ignored a truncated HTTP body."); 87651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 87751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 87851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 87951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 88051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 88151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 88251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 88351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 88451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 88551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 88651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithContentLengthHeader() throws IOException { 88751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.FIXED_LENGTH); 88851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 88951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 89051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithChunkedEncoding() throws IOException { 89151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.CHUNKED); 89251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 89351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 89451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithNoLengthHeaders() throws IOException { 89551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.END_OF_STREAM); 89651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 89751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 89851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testClientPrematureDisconnect(TransferKind transferKind) throws IOException { 89951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 90051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 1024); 90151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 90251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 90351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 90451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 90551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 90651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 90751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 90851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 90951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 91051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 91151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 91251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.read(); 91351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("Expected an IOException because the stream is closed."); 91451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 91551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 91651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 91751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 91851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 91951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 92051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 92151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 92251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 92351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 92451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 92551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Shortens the body of {@code response} but not the corresponding headers. 92651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Only useful to test how clients respond to the premature conclusion of 92751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the HTTP body. 92851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 92951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) { 93051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 93151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson List<String> headers = new ArrayList<String>(response.getHeaders()); 93251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(Arrays.copyOfRange(response.getBody(), 0, numBytesToKeep)); 93351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().clear(); 93451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().addAll(headers); 93551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return response; 93651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 93751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 93851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithContentLengthHeader() throws IOException { 93951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.FIXED_LENGTH); 94051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 94151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 94251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithChunkedEncoding() throws IOException { 94351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.CHUNKED); 94451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 94551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 94651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithNoLengthHeaders() throws IOException { 94751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.END_OF_STREAM); 94851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 94951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 95051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndReset(TransferKind transferKind) throws IOException { 95151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 95251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDEFGHIJKLMNOPQRSTUVWXYZ", 1024); 95351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 95451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 95551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 95651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 95751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 95851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 95951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 96051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertFalse("This implementation claims to support mark().", in.markSupported()); 96151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.mark(5); 96251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 96351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 96451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.reset(); 96551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail(); 96651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 96751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 96851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("FGHIJKLMNOPQRSTUVWXYZ", readAscii(in, Integer.MAX_VALUE)); 96951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 97051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQRSTUVWXYZ", server.getUrl("/").openConnection()); 97151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 97251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 97351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 97451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 97551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 97651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * We've had a bug where we forget the HTTP response when we see response 97751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * code 401. This causes a new HTTP request to be issued for every call into 97851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the URLConnection. 97951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 98051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testUnauthorizedResponseHandling() throws IOException { 98151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse() 98251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .addHeader("WWW-Authenticate: challenge") 98351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setResponseCode(401) // UNAUTHORIZED 98451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("Unauthorized"); 98551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 98651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 98751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 98851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 98951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 99051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 99151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 99251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 99351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 99451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 99551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 99651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, server.getRequestCount()); 99751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 99851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 9996906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testNonHexChunkSize() throws IOException { 10006906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 10016906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5\r\nABCDE\r\nG\r\nFGHIJKLMNOPQRSTU\r\n0\r\n\r\n") 10026906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 10036906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked")); 10046906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 10056906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 10066906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 10076906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 10086906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 10096906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 10106906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 10116906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10126906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10136906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 10146906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testMissingChunkBody() throws IOException { 10156906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 10166906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5") 10176906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 10186906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked") 10196906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setDisconnectAtEnd(true)); 10206906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 10216906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 10226906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 10236906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 10246906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 10256906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 10266906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 10276906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10286906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10296906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 103050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson /** 103150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * This test checks whether connections are gzipped by default. This 103250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * behavior in not required by the API, so a failure of this test does not 103350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * imply a bug in the implementation. 103450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson */ 103550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testGzipEncodingEnabledByDefault() throws IOException, InterruptedException { 103650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 103750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody(gzip("ABCABCABC".getBytes("UTF-8"))) 103850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: gzip")); 103950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 104050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 104150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 104250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCABCABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 10438116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson assertNull(connection.getContentEncoding()); 104450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 104550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 104650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 104750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 104850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 1049deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testClientConfiguredGzipContentEncoding() throws Exception { 1050deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(new MockResponse() 1051deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .setBody(gzip("ABCDEFGHIJKLMNOPQRSTUVWXYZ".getBytes("UTF-8"))) 1052deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .addHeader("Content-Encoding: gzip")); 1053deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 1054deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1055deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1056deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 1057deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 1058deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("ABCDEFGHIJKLMNOPQRSTUVWXYZ", readAscii(gunzippedIn, Integer.MAX_VALUE)); 1059deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1060deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson RecordedRequest request = server.takeRequest(); 1061deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 1062deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1063deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1064deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithFixedLength() throws Exception { 1065deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.FIXED_LENGTH); 1066deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1067deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1068deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithChunkedEncoding() throws Exception { 1069deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.CHUNKED); 1070deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1071deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 107250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testClientConfiguredCustomContentEncoding() throws Exception { 107350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 107450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody("ABCDE") 107550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: custom")); 107650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 107750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 107850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 107950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson connection.addRequestProperty("Accept-Encoding", "custom"); 108050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCDE", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 108150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 108250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 108350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: custom"); 108450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 108550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 1086deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 1087deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Test a bug where gzip input streams weren't exhausting the input stream, 1088deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * which corrupted the request that followed. 1089deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * http://code.google.com/p/android/issues/detail?id=7059 1090deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1091deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson private void testClientConfiguredGzipContentEncodingAndConnectionReuse( 1092deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson TransferKind transferKind) throws Exception { 1093deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseOne = new MockResponse(); 1094deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson responseOne.addHeader("Content-Encoding: gzip"); 1095deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseOne, gzip("one (gzipped)".getBytes("UTF-8")), 5); 1096deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseOne); 1097deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseTwo = new MockResponse(); 1098deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseTwo, "two (identity)", 5); 1099deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseTwo); 1100deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 1101deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1102deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1103deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 1104deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 1105deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("one (gzipped)", readAscii(gunzippedIn, Integer.MAX_VALUE)); 1106deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1107deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1108deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection = server.getUrl("/").openConnection(); 1109deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("two (identity)", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1110deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 1111deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1112deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1113deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 1114ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * Obnoxiously test that the chunk sizes transmitted exactly equal the 1115ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * requested data+chunk header size. Although setChunkedStreamingMode() 1116ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * isn't specific about whether the size applies to the data or the 1117ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * complete chunk, the RI interprets it as a complete chunk. 1118ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson */ 1119ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testSetChunkedStreamingMode() throws IOException, InterruptedException { 1120ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse()); 1121ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1122ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1123ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1124ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setChunkedStreamingMode(8); 1125ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setDoOutput(true); 1126ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = urlConnection.getOutputStream(); 1127ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write("ABCDEFGHIJKLMNOPQ".getBytes("US-ASCII")); 1128ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1129ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1130ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1131ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("ABCDEFGHIJKLMNOPQ", new String(request.getBody(), "US-ASCII")); 1132ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.asList(3, 3, 3, 3, 3, 2), request.getChunkSizes()); 1133ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1134ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1135ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithFixedLengthStreaming() throws Exception { 1136ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.FIXED_LENGTH); 1137ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1138ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1139ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithChunkedStreaming() throws Exception { 1140ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.CHUNKED); 1141ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1142ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1143ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void testAuthenticateWithStreamingPost(StreamingMode streamingMode) throws Exception { 1144ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1145ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1146ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1147ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1148ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1149ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1150ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1151ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1152ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1153ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1154ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1155ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 1156ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 1157ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 1158ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setChunkedStreamingMode(0); 1159ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1160ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1161ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1162ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1163ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson try { 1164ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.getInputStream(); 1165ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail(); 1166ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } catch (HttpRetryException expected) { 1167ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1168ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1169ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the request... 1170ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1171ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1172ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1173ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1174ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1175ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson enum StreamingMode { 1176ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson FIXED_LENGTH, CHUNKED 1177ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1178ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1179ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithPost() throws Exception { 1180ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1181ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1182ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1183ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1184ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1185ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1186ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1187ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1188ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1189ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1190ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1191ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1192ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1193ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1194ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1195ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1196ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1197ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1198ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1199ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1200ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1201ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1202ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1203ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1204ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1205da289bcd0a9e207cc03c752f7c21c9004056e179Jesse Wilson // ...but the three requests that follow include an authorization header 1206ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1207ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1208ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 1209ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1210ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1211ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1212ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1213ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1214ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1215ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithGet() throws Exception { 1216ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1217ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1218ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1219ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1220ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1221ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1222ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1223ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1224ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1225ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1226ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1227ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1228ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1229ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1230ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1231ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1232ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1233ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1234ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1235ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1236ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...but the three requests that follow requests include an authorization header 1237ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1238ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1239ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("GET / HTTP/1.1", request.getRequestLine()); 1240ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1241ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1242ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1243ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1244ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1245c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithChunkedEncoding() throws Exception { 1246c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.CHUNKED, true); 1247c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1248c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1249c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithContentLengthHeader() throws Exception { 1250c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.FIXED_LENGTH, true); 1251c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1252c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1253c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithNoLengthHeaders() throws Exception { 1254c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.END_OF_STREAM, false); 1255c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1256c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1257c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private void testRedirected(TransferKind transferKind, boolean reuse) throws Exception { 1258c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse response = new MockResponse() 1259c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1260c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo"); 1261c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson transferKind.setBody(response, "This page has moved!", 10); 1262c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(response); 1263c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1264c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1265c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1266c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1267c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1268c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1269c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1270c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1271c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1272c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1273c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1274c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson if (reuse) { 1275c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1276c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1277c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1278c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1279c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedOnHttps() throws IOException, InterruptedException { 1280c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1281059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1282c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1283c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1284c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo") 1285c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1286c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1287c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1288c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1289c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1290059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1291c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1292c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1293c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1294c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1295c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1296c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1297c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1298c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1299c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1300c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1301c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpsToHttp() throws IOException, InterruptedException { 1302c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1303059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1304c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1305c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1306c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: http://anyhost/foo") 1307c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1308c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1309c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1310c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1311059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1312c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1313c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1314c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1315c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1316c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpToHttps() throws IOException, InterruptedException { 1317c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1318c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1319c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: https://anyhost/foo") 1320c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1321c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1322c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1323c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1324c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1325c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1326c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1327c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1328c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectToAnotherOriginServer() throws Exception { 1329c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockWebServer server2 = new MockWebServer(); 1330c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.enqueue(new MockResponse().setBody("This is the 2nd server!")); 1331c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.play(); 1332c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1333c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1334c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1335c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: " + server2.getUrl("/").toString()) 1336c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1337c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the first server again!")); 1338c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1339c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1340c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1341c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the 2nd server!", 1342c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1343c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(server2.getUrl("/"), connection.getURL()); 1344c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1345c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson // make sure the first server was careful to recycle the connection 1346c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the first server again!", 1347c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(server.getUrl("/").openStream(), Integer.MAX_VALUE)); 1348c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1349c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 135000feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertContains(first.getHeaders(), "Host: " + hostname + ":" + server.getPort()); 1351c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest second = server2.takeRequest(); 135200feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertContains(second.getHeaders(), "Host: " + hostname + ":" + server2.getPort()); 1353c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest third = server.takeRequest(); 1354c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, third.getSequenceNumber()); 1355c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1356c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.shutdown(); 1357c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1358c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1359c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testHttpsWithCustomTrustManager() throws Exception { 1360c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 1361c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingTrustManager trustManager = new RecordingTrustManager(); 1362c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLContext sc = SSLContext.getInstance("TLS"); 1363c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson sc.init(null, new TrustManager[] { trustManager }, new java.security.SecureRandom()); 1364c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1365c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HostnameVerifier defaultHostnameVerifier = HttpsURLConnection.getDefaultHostnameVerifier(); 1366c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(hostnameVerifier); 1367c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLSocketFactory defaultSSLSocketFactory = HttpsURLConnection.getDefaultSSLSocketFactory(); 1368c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory()); 1369c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson try { 1370c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1371059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1372c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1373c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 1374c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("GHI")); 1375c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1376c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1377c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URL url = server.getUrl("/"); 1378c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("ABC", readAscii(url.openStream(), Integer.MAX_VALUE)); 1379c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("DEF", readAscii(url.openStream(), Integer.MAX_VALUE)); 1380c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GHI", readAscii(url.openStream(), Integer.MAX_VALUE)); 1381c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 138200feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertEquals(Arrays.asList("verify " + hostname), hostnameVerifier.calls); 13834559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals(Arrays.asList("checkServerTrusted [" 138400feece22909b7dc79fc96d666d157390b93858eJesse Wilson + "CN=" + hostname + " 1, " 13854559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "CN=Test Intermediate Certificate Authority 1, " 13864559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "CN=Test Root Certificate Authority 1" 13874559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "] RSA"), 1388c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson trustManager.calls); 1389c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } finally { 1390c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(defaultHostnameVerifier); 1391c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(defaultSSLSocketFactory); 1392c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1393c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1394c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1395eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testConnectTimeouts() throws IOException { 1396f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom // Set a backlog and use it up so that we can expect the 1397f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom // URLConnection to properly timeout. According to Steven's 1398f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom // 4.5 "listen function", linux adds 3 to the specified 1399f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom // backlog, so we need to connect 4 times before it will hang. 1400f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom ServerSocket serverSocket = new ServerSocket(0, 1); 1401f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom int serverPort = serverSocket.getLocalPort(); 1402f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom Socket[] sockets = new Socket[4]; 1403f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom for (int i = 0; i < sockets.length; i++) { 1404f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom sockets[i] = new Socket("localhost", serverPort); 1405f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 1406f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 1407f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom URLConnection urlConnection = new URL("http://localhost:" + serverPort).openConnection(); 1408eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setConnectTimeout(1000); 1409eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1410eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.getInputStream(); 1411eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1412eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1413eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1414f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 1415f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom for (Socket s : sockets) { 1416f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom s.close(); 1417f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 1418eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1419eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1420eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testReadTimeouts() throws IOException { 1421eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson /* 1422eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * This relies on the fact that MockWebServer doesn't close the 1423eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * connection after a response has been sent. This causes the client to 1424eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * try to read more bytes than are sent, which results in a timeout. 1425eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson */ 1426eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson MockResponse timeout = new MockResponse() 1427eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .setBody("ABC") 1428eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .clearHeaders() 1429eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .addHeader("Content-Length: 4"); 1430eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.enqueue(timeout); 1431eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.play(); 1432eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1433eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 1434eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setReadTimeout(1000); 1435eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson InputStream in = urlConnection.getInputStream(); 1436eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('A', in.read()); 1437eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('B', in.read()); 1438eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('C', in.read()); 1439eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1440eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson in.read(); // if Content-Length was accurate, this would return -1 immediately 1441eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1442eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1443eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1444eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1445eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1446125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson public void testSetChunkedEncodingAsRequestProperty() throws IOException, InterruptedException { 1447125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.enqueue(new MockResponse()); 1448125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.play(); 1449125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1450125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1451125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setRequestProperty("Transfer-encoding", "chunked"); 1452125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setDoOutput(true); 1453125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.getOutputStream().write("ABC".getBytes("UTF-8")); 1454125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1455125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1456125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson RecordedRequest request = server.takeRequest(); 1457125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals("ABC", new String(request.getBody(), "UTF-8")); 1458125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson } 1459125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1460f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInRequest() throws IOException, InterruptedException { 1461f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); // server doesn't honor the connection: close header! 1462f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1463f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1464f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1465f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1466f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson a.setRequestProperty("Connection", "close"); 1467f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1468f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1469f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1470f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1471f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1472f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1473f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1474f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1475f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1476f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1477f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInResponse() throws IOException, InterruptedException { 1478f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().addHeader("Connection: close")); 1479f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1480f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1481f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1482f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1483f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1484f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1485f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1486f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1487f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1488f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1489f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1490f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1491f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1492f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1493f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseWithRedirect() throws IOException, InterruptedException { 1494f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson MockResponse response = new MockResponse() 1495f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1496f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Location: /foo") 1497f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Connection: close"); 1498f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(response); 1499f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1500f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1501f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1502f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1503f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("This is the new location!", 1504f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1505f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1506f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1507f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1508f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1509f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1510f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 151165d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson public void testResponseCodeDisagreesWithHeaders() throws IOException, InterruptedException { 151265d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.enqueue(new MockResponse() 151365d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setResponseCode(HttpURLConnection.HTTP_NO_CONTENT) 151465d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setBody("This body is not allowed!")); 151565d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.play(); 151665d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 151765d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 151865d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson assertEquals("This body is not allowed!", 151965d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 152065d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson } 152165d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 1522ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson public void testSingleByteReadIsSigned() throws IOException { 1523ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.enqueue(new MockResponse().setBody(new byte[] { -2, -1 })); 1524ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.play(); 1525ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1526ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1527ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson InputStream in = connection.getInputStream(); 1528ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(254, in.read()); 1529ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(255, in.read()); 1530ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(-1, in.read()); 1531ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson } 1532ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1533f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithChunkedEncoding() throws IOException { 1534f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.CHUNKED); 1535f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1536f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1537f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithFixedLength() throws IOException { 1538f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.FIXED_LENGTH); 1539f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1540f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1541f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithNoLengthHeaders() throws IOException { 1542f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.END_OF_STREAM); 1543f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1544f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1545f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson /** 1546f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * We explicitly permit apps to close the upload stream even after it has 1547f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * been transmitted. We also permit flush so that buffered streams can 1548f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * do a no-op flush when they are closed. http://b/3038470 1549f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson */ 1550f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson private void testFlushAfterStreamTransmitted(TransferKind transferKind) throws IOException { 1551f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.enqueue(new MockResponse().setBody("abc")); 1552f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.play(); 1553f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1554f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1555f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setDoOutput(true); 1556f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson byte[] upload = "def".getBytes("UTF-8"); 1557f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1558f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson if (transferKind == TransferKind.CHUNKED) { 1559f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setChunkedStreamingMode(0); 1560f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } else if (transferKind == TransferKind.FIXED_LENGTH) { 1561f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setFixedLengthStreamingMode(upload.length); 1562f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1563f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1564f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson OutputStream out = connection.getOutputStream(); 1565f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write(upload); 1566f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson assertEquals("abc", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1567f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1568f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.flush(); // dubious but permitted 1569f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson try { 1570f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write("ghi".getBytes("UTF-8")); 1571f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson fail(); 1572f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } catch (IOException expected) { 1573f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1574f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1575f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1576ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson /** 1577deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Encodes the response body using GZIP and adds the corresponding header. 1578deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1579deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public byte[] gzip(byte[] bytes) throws IOException { 1580deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1581deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson OutputStream gzippedOut = new GZIPOutputStream(bytesOut); 1582deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.write(bytes); 1583deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.close(); 1584deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson return bytesOut.toByteArray(); 1585deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1586deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 158737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private <T> List<T> toListOrNull(T[] arrayOrNull) { 158837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return arrayOrNull != null ? Arrays.asList(arrayOrNull) : null; 158937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 159037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 1591c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson /** 1592c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * Reads at most {@code limit} characters from {@code in} and asserts that 1593c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * content equals {@code expected}. 1594c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson */ 1595c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection, int limit) 1596c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson throws IOException { 1597f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom connection.connect(); 159851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(expected, readAscii(connection.getInputStream(), limit)); 1599c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ((HttpURLConnection) connection).disconnect(); 1600c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1601c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 1602c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection) throws IOException { 1603c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent(expected, connection, Integer.MAX_VALUE); 1604c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1605c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 160660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson private void assertContains(List<String> headers, String header) { 160760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertTrue(headers.toString(), headers.contains(header)); 160860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 160951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1610ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void assertContainsNoneMatching(List<String> headers, String pattern) { 1611ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (String header : headers) { 1612ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (header.matches(pattern)) { 1613ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail("Header " + header + " matches " + pattern); 1614ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1615ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1616ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1617ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1618eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson private Set<String> newSet(String... elements) { 161983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return new HashSet<String>(Arrays.asList(elements)); 162083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 162183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 162251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson enum TransferKind { 162351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson CHUNKED() { 1624deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) 162551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException { 162651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setChunkedBody(content, chunkSize); 162751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 162851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 162951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson FIXED_LENGTH() { 1630deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 163151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 163251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 163351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 163451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson END_OF_STREAM() { 1635deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 163651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 163751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 163851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (Iterator<String> h = response.getHeaders().iterator(); h.hasNext(); ) { 163951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (h.next().startsWith("Content-Length:")) { 164051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson h.remove(); 164151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 164251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 164351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 164451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 164551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }; 164651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1647deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson abstract void setBody(MockResponse response, byte[] content, int chunkSize) 164851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException; 1649deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1650deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson void setBody(MockResponse response, String content, int chunkSize) throws IOException { 1651deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson setBody(response, content.getBytes("UTF-8"), chunkSize); 1652deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 165351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 1654c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1655984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson enum ProxyConfig { 1656f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom NO_PROXY() { 1657f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom @Override public HttpURLConnection connect(MockWebServer server, URL url) 1658f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom throws IOException { 1659f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom return (HttpURLConnection) url.openConnection(Proxy.NO_PROXY); 1660f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 1661f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom }, 1662f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 1663984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson CREATE_ARG() { 1664984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1665984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1666984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(server.toProxyAddress()); 1667984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1668984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1669984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1670984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson PROXY_SYSTEM_PROPERTY() { 1671984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1672984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1673984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyHost", "localhost"); 1674984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyPort", Integer.toString(server.getPort())); 1675984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1676984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1677984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1678984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1679984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTP_PROXY_SYSTEM_PROPERTY() { 1680984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1681984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1682984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyHost", "localhost"); 1683984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyPort", Integer.toString(server.getPort())); 1684984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1685984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1686984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1687984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1688984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTPS_PROXY_SYSTEM_PROPERTY() { 1689984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1690984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1691984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyHost", "localhost"); 1692984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyPort", Integer.toString(server.getPort())); 1693984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1694984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1695984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }; 1696984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1697984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public abstract HttpURLConnection connect(MockWebServer server, URL url) throws IOException; 1698984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1699984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1700c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingTrustManager implements X509TrustManager { 1701c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1702c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1703c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public X509Certificate[] getAcceptedIssuers() { 1704c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("getAcceptedIssuers"); 1705c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return new X509Certificate[] {}; 1706c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1707c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1708c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkClientTrusted(X509Certificate[] chain, String authType) 1709c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1710c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkClientTrusted " + certificatesToString(chain) + " " + authType); 1711c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1712c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1713c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkServerTrusted(X509Certificate[] chain, String authType) 1714c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1715c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkServerTrusted " + certificatesToString(chain) + " " + authType); 1716c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1717c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1718c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private String certificatesToString(X509Certificate[] certificates) { 1719c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson List<String> result = new ArrayList<String>(); 1720c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson for (X509Certificate certificate : certificates) { 1721c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson result.add(certificate.getSubjectDN() + " " + certificate.getSerialNumber()); 1722c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1723c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return result.toString(); 1724c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1725c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1726c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1727c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingHostnameVerifier implements HostnameVerifier { 1728c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1729c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1730c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public boolean verify(String hostname, SSLSession session) { 1731c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("verify " + hostname); 1732c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return true; 1733c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1734c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 173537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 173637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private static class InsecureResponseCache extends ResponseCache { 173737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private final DefaultResponseCache delegate = new DefaultResponseCache(); 173837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 173937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection connection) throws IOException { 174037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return delegate.put(uri, connection); 174137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 174237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 174337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 174437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 174537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson final CacheResponse response = delegate.get(uri, requestMethod, requestHeaders); 174637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson if (response instanceof SecureCacheResponse) { 174737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return new CacheResponse() { 174837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public InputStream getBody() throws IOException { 174937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response.getBody(); 175037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 175137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public Map<String, List<String>> getHeaders() throws IOException { 175237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response.getHeaders(); 175337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 175437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson }; 175537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 175637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response; 175737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 175837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 1759e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes} 1760