URLConnectionTest.java revision d4bddd7d1fb7b1b7f0836648228235c6e4b56a18
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; 216247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.io.IOException; 2251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.io.InputStream; 23e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughesimport java.io.InputStreamReader; 2402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.io.OutputStream; 254557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.Authenticator; 264557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.CacheRequest; 274557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.CacheResponse; 284557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpRetryException; 294557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpURLConnection; 304557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.PasswordAuthentication; 314557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.ResponseCache; 324557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.SocketTimeoutException; 334557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URI; 344557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URISyntaxException; 354557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URL; 364557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URLConnection; 37c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.CertificateException; 38c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.X509Certificate; 3951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.ArrayList; 4002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.util.Arrays; 4151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Collections; 4283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.HashSet; 4351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Iterator; 446247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.util.List; 4583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.Map; 4651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Set; 4783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.concurrent.atomic.AtomicReference; 48deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPInputStream; 49deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPOutputStream; 5060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HostnameVerifier; 5160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HttpsURLConnection; 52c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLContext; 53096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilsonimport javax.net.ssl.SSLException; 5460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.SSLSession; 55c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLSocketFactory; 56c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.TrustManager; 57c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.X509TrustManager; 5850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilsonimport libcore.javax.net.ssl.TestSSLContext; 5951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport tests.http.DefaultResponseCache; 6060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockResponse; 6160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockWebServer; 6260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.RecordedRequest; 63e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 64e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughespublic class URLConnectionTest extends junit.framework.TestCase { 65b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 66ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private static final Authenticator SIMPLE_AUTHENTICATOR = new Authenticator() { 67ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson protected PasswordAuthentication getPasswordAuthentication() { 68ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson return new PasswordAuthentication("username", "password".toCharArray()); 69ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 70ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson }; 71ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 72d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson private final HostnameVerifier ALWAYS_TRUST_VERIFIER = new HostnameVerifier() { 73d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public boolean verify(String hostname, SSLSession session) { 74d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson return true; 75d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 76d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson }; 77d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 7851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockWebServer server = new MockWebServer(); 7951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 8051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson @Override protected void tearDown() throws Exception { 8151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(null); 82ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(null); 8351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); 8451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 8551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 8683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testRequestHeaders() throws IOException, InterruptedException { 8783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse()); 8883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 8983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 9083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 9183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "e"); 9283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "f"); 9383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders = urlConnection.getRequestProperties(); 9483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("e", "f"), new HashSet<String>(requestHeaders.get("D"))); 9583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 9683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.put("G", Arrays.asList("h")); 9783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 9883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 9983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 10083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 10183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.get("D").add("i"); 10283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 10383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 10483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 10583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 10683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty(null, "j"); 10783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 10883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 10983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 11083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty(null, "k"); 11283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 11383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 11483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 11583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("NullValue", null); // should fail silently! 11683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("AnotherNullValue", null); // should fail silently! 11783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 11883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.getResponseCode(); 11983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson RecordedRequest request = server.takeRequest(); 12083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: e"); 12183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: f"); 12283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "NullValue.*"); 12383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "AnotherNullValue.*"); 12483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "G:.*"); 12583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "null:.*"); 12683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 12783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 12883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("N", "o"); 12983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 13083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 13183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 13383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("P", "q"); 13483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 13583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 13683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 13983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseHeaders() throws IOException, InterruptedException { 14083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse() 14183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setStatus("HTTP/1.0 200 Fantastic") 14283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: b") 14383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: c") 14483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8)); 14583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 14683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 14783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 14883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 14983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 15083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> responseHeaders = urlConnection.getHeaderFields(); 15183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("b", "c"), new HashSet<String>(responseHeaders.get("A"))); 15283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 15383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.put("N", Arrays.asList("o")); 15483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 15583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 15683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 15783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 15883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.get("A").add("d"); 15983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 16083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 16183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 16283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 16383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 164e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // Check that if we don't read to the end of a response, the next request on the 165e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // recycled connection doesn't get the unread tail of the first request's response. 166e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // http://code.google.com/p/android/issues/detail?id=2939 167e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes public void test_2939() throws Exception { 168b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8); 169b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 170b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 171b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 172b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 173b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 174c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 175c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 1768baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson } 1778baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson 178977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // Check that we recognize a few basic mime types by extension. 179977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // http://code.google.com/p/android/issues/detail?id=10100 180977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes public void test_10100() throws Exception { 181977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("image/jpeg", URLConnection.guessContentTypeFromName("someFile.jpg")); 182977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("application/pdf", URLConnection.guessContentTypeFromName("stuff.pdf")); 183977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes } 184977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes 1858baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson public void testConnectionsArePooled() throws Exception { 186b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setBody("ABCDEFGHIJKLMNOPQR"); 187b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 188b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 189b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 190b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 191b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 192b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 193c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 194c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 195c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 196c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 197c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 198c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 199c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 200c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 201c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testChunkedConnectionsArePooled() throws Exception { 202c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDEFGHIJKLMNOPQR", 5); 203c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 204c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 205c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 206c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 207c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 208c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 209c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 210b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 211c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 212b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 213c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 214b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 215e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes } 21602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 217b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson enum WriteKind { BYTE_BY_BYTE, SMALL_BUFFERS, LARGE_BUFFERS } 21802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 21902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_byteByByte() throws Exception { 22051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.BYTE_BY_BYTE); 22102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 22202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 22302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_smallBuffers() throws Exception { 22451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.SMALL_BUFFERS); 22502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 22602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 22702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_largeBuffers() throws Exception { 22851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.LARGE_BUFFERS); 22902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 23102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_byteByByte() throws Exception { 23251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.BYTE_BY_BYTE); 23302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 23502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_smallBuffers() throws Exception { 23651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.SMALL_BUFFERS); 23702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 23902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_largeBuffers() throws Exception { 24051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.LARGE_BUFFERS); 24102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 24202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 24351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void doUpload(TransferKind uploadKind, WriteKind writeKind) throws Exception { 24402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes int n = 512*1024; 245b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.setBodyLimit(0); 246b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse()); 247b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 248b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 249b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson HttpURLConnection conn = (HttpURLConnection) server.getUrl("/").openConnection(); 25002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setDoOutput(true); 25102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setRequestMethod("POST"); 25251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 25302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setChunkedStreamingMode(-1); 25402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 25502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setFixedLengthStreamingMode(n); 25602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 25702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes OutputStream out = conn.getOutputStream(); 25802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes if (writeKind == WriteKind.BYTE_BY_BYTE) { 25902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; ++i) { 26002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write('x'); 26102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 26202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 26302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes byte[] buf = new byte[writeKind == WriteKind.SMALL_BUFFERS ? 256 : 64*1024]; 26402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes Arrays.fill(buf, (byte) 'x'); 26502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; i += buf.length) { 26602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write(buf, 0, Math.min(buf.length, n - i)); 26702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 26802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 26902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.close(); 2704cb7f05dc68abb23ae54a5891c369062185f2210Elliott Hughes assertEquals(200, conn.getResponseCode()); 271b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson RecordedRequest request = server.takeRequest(); 272b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(n, request.getBodySize()); 27351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 274b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().size() > 0); 275b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } else { 276b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().isEmpty()); 277b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } 27802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 2796247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 28051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 28151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Test that response caching is consistent with the RI and the spec. 28251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.4 28351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 2846247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching() throws Exception { 2856247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // Test each documented HTTP/1.1 code, plus the first unused value in each range. 2866247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html 2876247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 2886247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // We can't test 100 because it's not really a response. 2896247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // assertCached(false, 100); 2906247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 101); 2916247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 102); 2926247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 200); 2936247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 201); 2946247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 202); 2956247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 203); 2966247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 204); 2976247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 205); 2986247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 206); 2996247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 207); 30051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_300.) 3016247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 301); 3026247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 302; i <= 308; ++i) { 3036247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3046247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3056247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 400; i <= 406; ++i) { 3066247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3076247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3086247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // (See test_responseCaching_407.) 3096247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 408); 3106247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 409); 31151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_410.) 3126247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 411; i <= 418; ++i) { 3136247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3146247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3156247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 500; i <= 506; ++i) { 3166247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3176247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3186247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3196247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 32051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_300() throws Exception { 32151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: fix this for android 32251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 300); 32351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 32451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3256247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching_407() throws Exception { 3266247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // This test will fail on Android because we throw if we're not using a proxy. 3276247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // This isn't true of the RI, but it seems like useful debugging behavior. 3286247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 407); 3296247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3306247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 33151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_410() throws Exception { 33251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // the HTTP spec permits caching 410s, but the RI doesn't. 33351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 410); 33451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 33551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3366247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes private void assertCached(boolean shouldPut, int responseCode) throws Exception { 33751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server = new MockWebServer(); 338b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse() 339b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .setResponseCode(responseCode) 34051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("ABCDE") 341b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .addHeader("WWW-Authenticate: challenge")); 342b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 343b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 34451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache responseCache = new DefaultResponseCache(); 34551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(responseCache); 34651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 34751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3486247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertEquals(responseCode, conn.getResponseCode()); 3496247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 35051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // exhaust the content stream 35151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 35251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: remove special case once testUnauthorizedResponseHandling() is fixed 35351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (responseCode != 401) { 35451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(conn.getInputStream(), Integer.MAX_VALUE); 35551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 35651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException ignored) { 35751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 35851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 35951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson Set<URI> expectedCachedUris = shouldPut 36051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ? Collections.singleton(url.toURI()) 36151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson : Collections.<URI>emptySet(); 36251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(Integer.toString(responseCode), 36351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson expectedCachedUris, responseCache.getContents().keySet()); 36451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); // tearDown() isn't sufficient; this test starts multiple servers 3656247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 36660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 36760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttps() throws IOException, InterruptedException { 36860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 36960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 370059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 371c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 37260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson server.play(); 37360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 374096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 3754559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 37660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 377c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via HTTPS", connection); 37860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 37960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson RecordedRequest request = server.takeRequest(); 38060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 38160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 38260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 383096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson public void testConnectViaHttpsReusingConnections() throws IOException, InterruptedException { 384096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 385096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 386059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 387096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 388096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 389096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.play(); 390096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 391b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 392b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 393b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("this response comes via HTTPS", connection); 394b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 395b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 396b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 397b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("another response via HTTPS", connection); 398b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 399b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 400b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 401b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson } 402b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 4038116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson public void testConnectViaHttpsReusingConnectionsDifferentFactories() 404b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson throws IOException, InterruptedException { 405b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 406b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 407b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 408b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 409b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 410b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.play(); 411b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 412096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson // install a custom SSL socket factory so the server can be authorized 413096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 414059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 415096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertContent("this response comes via HTTPS", connection); 416096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 417096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 418096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson try { 419096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 420b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson fail("without an SSL socket factory, the connection should fail"); 421096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } catch (SSLException expected) { 422096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 423096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 424096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 4254559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom public void testConnectViaHttpsWithSSLFallback() throws IOException, InterruptedException { 4264559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 4274559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4284559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 4294559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setDisconnectAtStart(true)); 4304559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via SSL")); 4314559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.play(); 4324559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4334559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 4344559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 4354559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4364559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertContent("this response comes via SSL", connection); 4374559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4384559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom RecordedRequest request = server.takeRequest(); 4394559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 4404559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom } 4414559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 44260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaProxy() throws IOException, InterruptedException { 443c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse mockResponse = new MockResponse().setBody("this response comes via a proxy"); 44451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(mockResponse); 44551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 44660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 44760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URLConnection connection = new URL("http://android.com/foo").openConnection( 44851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.toProxyAddress()); 449c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a proxy", connection); 45060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 45151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest request = server.takeRequest(); 45260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET http://android.com/foo HTTP/1.1", request.getRequestLine()); 45360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(request.getHeaders(), "Host: android.com"); 45460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 45560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 456c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithContentLengthHeader() throws IOException { 457c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(new MockResponse() 458c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .setBody("abc\r\nYOU SHOULD NOT SEE THIS") 459c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .clearHeaders() 460c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .addHeader("Content-Length: 3")); 461c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 462c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 463c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 464c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 465c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 466c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithChunkedHeader() throws IOException { 467c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse mockResponse = new MockResponse(); 468c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setChunkedBody("abc", 3); 469c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 470c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write(mockResponse.getBody()); 471c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write("\r\nYOU SHOULD NOT SEE THIS".getBytes()); 472c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setBody(bytesOut.toByteArray()); 473c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.clearHeaders(); 474c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.addHeader("Transfer-encoding: chunked"); 475c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 476c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(mockResponse); 477c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 478c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 479c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 480c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 481c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 48260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttpProxyToHttps() throws IOException, InterruptedException { 48360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 48460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 485059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 486c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 487c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via a secure proxy")); 48851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 48960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 49060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URL url = new URL("https://android.com/foo"); 49160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) url.openConnection( 49251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.toProxyAddress()); 493059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 494d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.setHostnameVerifier(ALWAYS_TRUST_VERIFIER); 49560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 496c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a secure proxy", connection); 49760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 49851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest connect = server.takeRequest(); 49960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("Connect line failure on proxy", 50060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson "CONNECT android.com:443 HTTP/1.1", connect.getRequestLine()); 50160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 50260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 50351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest get = server.takeRequest(); 50460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", get.getRequestLine()); 50560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(get.getHeaders(), "Host: android.com"); 50660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 50760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 508d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson /** 509d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson * Test which headers are sent unencrypted to the HTTP proxy. 510d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson */ 511d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testProxyConnectIncludesProxyHeadersOnly() 512d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson throws IOException, InterruptedException { 513d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 514d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 515d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 516d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 517d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("encrypted response from the origin server")); 518d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 519d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 520d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson URL url = new URL("https://android.com/foo"); 521d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) url.openConnection( 522d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.toProxyAddress()); 523d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Private", "Secret"); 524d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Proxy-Authorization", "bar"); 525d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("User-Agent", "baz"); 526d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 527d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.setHostnameVerifier(ALWAYS_TRUST_VERIFIER); 528d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContent("encrypted response from the origin server", connection); 529d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 530d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest connect = server.takeRequest(); 531d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContainsNoneMatching(connect.getHeaders(), "Private.*"); 532d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Authorization: bar"); 533d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "User-Agent: baz"); 534d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 535d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Connection: Keep-Alive"); 536d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 537d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest get = server.takeRequest(); 538d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(get.getHeaders(), "Private: Secret"); 539d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 540d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 541d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testDisconnectedConnection() throws IOException { 542d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("ABCDEFGHIJKLMNOPQR")); 543d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 544d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 545d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 546d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson InputStream in = connection.getInputStream(); 547d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertEquals('A', (char) in.read()); 548d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.disconnect(); 549d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson try { 550d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson in.read(); 551d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson fail("Expected a connection closed exception"); 552d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } catch (IOException expected) { 553d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 554d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 555d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 55651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithFixedLength() throws IOException { 55751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.FIXED_LENGTH); 55851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 55951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 56051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithChunkedEncoding() throws IOException { 56151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.CHUNKED); 56251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 56351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 56451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithNoLengthHeaders() throws IOException { 56551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.END_OF_STREAM); 56651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 56751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 56851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 56951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * HttpURLConnection.getInputStream().skip(long) causes ResponseCache corruption 57051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://code.google.com/p/android/issues/detail?id=8175 57151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 57251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testResponseCaching(TransferKind transferKind) throws IOException { 57351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 57451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "I love puppies but hate spiders", 1); 57551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 57651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 57751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 57851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 57951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 58051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 58151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // Make sure that calling skip() doesn't omit bytes from the cache. 58251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 58351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = urlConnection.getInputStream(); 58451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love ", readAscii(in, "I love ".length())); 58551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reliableSkip(in, "puppies but hate ".length()); 58651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("spiders", readAscii(in, "spiders".length())); 58751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 58851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 58951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 590096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 59151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 59251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson urlConnection = server.getUrl("/").openConnection(); // this response is cached! 59351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in = urlConnection.getInputStream(); 59451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love puppies but hate spiders", 59551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(in, "I love puppies but hate spiders".length())); 59651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 59751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getMissCount()); 59851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 599096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(1, cache.getSuccessCount()); 600096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 60151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 60251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 60383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseCacheRequestHeaders() throws IOException, URISyntaxException { 60483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 60583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 60683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 60783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson final AtomicReference<Map<String, List<String>>> requestHeadersRef 60883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson = new AtomicReference<Map<String, List<String>>>(); 60983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 61083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 61183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 61283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeadersRef.set(requestHeaders); 61383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 61483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 61583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection conn) throws IOException { 61683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 61783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 61883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson }); 61983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 62083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URL url = server.getUrl("/"); 62183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URLConnection urlConnection = url.openConnection(); 62283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("A", "android"); 62383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson readAscii(urlConnection.getInputStream(), Integer.MAX_VALUE); 62483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(Arrays.asList("android"), requestHeadersRef.get().get("A")); 62583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 62683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 62751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void reliableSkip(InputStream in, int length) throws IOException { 62851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson while (length > 0) { 62951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson length -= in.skip(length); 63051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 63151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 63251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 63351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 63451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Reads {@code count} characters from the stream. If the stream is 63551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * exhausted before {@code count} characters can be read, the remaining 63651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * characters are returned and the stream is closed. 63751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 63851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private String readAscii(InputStream in, int count) throws IOException { 63951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson StringBuilder result = new StringBuilder(); 64051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (int i = 0; i < count; i++) { 64151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson int value = in.read(); 64251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (value == -1) { 64351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 64451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 64551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 64651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson result.append((char) value); 64751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 64851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return result.toString(); 64951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 65051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithContentLengthHeader() throws IOException { 65251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.FIXED_LENGTH); 65351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 65451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithChunkedEncoding() throws IOException { 65651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.CHUNKED); 65751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 65851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithNoLengthHeaders() throws IOException { 66051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /* 66151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Intentionally empty. This case doesn't make sense because there's no 66251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * such thing as a premature disconnect when the disconnect itself 66351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * indicates the end of the data stream. 66451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 66551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 66651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 66751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testServerPrematureDisconnect(TransferKind transferKind) throws IOException { 66851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 66951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 16); 67051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(truncateViolently(response, 16)); 67151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 67251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 67351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 67451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 67551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 67651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 67751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson BufferedReader reader = new BufferedReader(new InputStreamReader( 67851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.getUrl("/").openConnection().getInputStream())); 67951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", reader.readLine()); 68051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 68151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reader.readLine(); 68251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("This implementation silently ignored a truncated HTTP body."); 68351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 68451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 68551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 68651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 68751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 68851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 68951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 69051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 69151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 69251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 69351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithContentLengthHeader() throws IOException { 69451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.FIXED_LENGTH); 69551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 69651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 69751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithChunkedEncoding() throws IOException { 69851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.CHUNKED); 69951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 70051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 70151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithNoLengthHeaders() throws IOException { 70251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.END_OF_STREAM); 70351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 70451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 70551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testClientPrematureDisconnect(TransferKind transferKind) throws IOException { 70651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 70751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 1024); 70851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 70951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 71051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 71151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 71251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 71351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 71451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 71551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 71651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 71751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 71851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 71951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.read(); 72051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("Expected an IOException because the stream is closed."); 72151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 72251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 72351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 72451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 72551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 72651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 72751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 72851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 72951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 73051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 73151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 73251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Shortens the body of {@code response} but not the corresponding headers. 73351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Only useful to test how clients respond to the premature conclusion of 73451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the HTTP body. 73551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 73651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) { 73751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 73851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson List<String> headers = new ArrayList<String>(response.getHeaders()); 73951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(Arrays.copyOfRange(response.getBody(), 0, numBytesToKeep)); 74051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().clear(); 74151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().addAll(headers); 74251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return response; 74351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 74451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 74551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithContentLengthHeader() throws IOException { 74651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.FIXED_LENGTH); 74751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 74851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 74951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithChunkedEncoding() throws IOException { 75051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.CHUNKED); 75151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 75251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 75351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithNoLengthHeaders() throws IOException { 75451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.END_OF_STREAM); 75551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 75651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 75751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndReset(TransferKind transferKind) throws IOException { 75851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 75951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDEFGHIJKLMNOPQRSTUVWXYZ", 1024); 76051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 76151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 76251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 76351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 76451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 76551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 76651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 76751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertFalse("This implementation claims to support mark().", in.markSupported()); 76851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.mark(5); 76951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 77051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 77151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.reset(); 77251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail(); 77351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 77451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 77551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("FGHIJKLMNOPQRSTUVWXYZ", readAscii(in, Integer.MAX_VALUE)); 77651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 77751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQRSTUVWXYZ", server.getUrl("/").openConnection()); 77851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 77951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 78051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 78151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 78251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 78351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * We've had a bug where we forget the HTTP response when we see response 78451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * code 401. This causes a new HTTP request to be issued for every call into 78551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the URLConnection. 78651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 78751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testUnauthorizedResponseHandling() throws IOException { 78851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse() 78951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .addHeader("WWW-Authenticate: challenge") 79051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setResponseCode(401) // UNAUTHORIZED 79151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("Unauthorized"); 79251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 79351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 79451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 79551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 79651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 79751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 79851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 79951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 80051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 80151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 80251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 80351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, server.getRequestCount()); 80451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 80551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 8066906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testNonHexChunkSize() throws IOException { 8076906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 8086906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5\r\nABCDE\r\nG\r\nFGHIJKLMNOPQRSTU\r\n0\r\n\r\n") 8096906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 8106906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked")); 8116906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 8126906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 8136906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 8146906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 8156906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 8166906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 8176906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 8186906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8196906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8206906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 8216906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testMissingChunkBody() throws IOException { 8226906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 8236906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5") 8246906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 8256906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked") 8266906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setDisconnectAtEnd(true)); 8276906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 8286906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 8296906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 8306906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 8316906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 8326906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 8336906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 8346906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8356906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8366906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 83750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson /** 83850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * This test checks whether connections are gzipped by default. This 83950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * behavior in not required by the API, so a failure of this test does not 84050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * imply a bug in the implementation. 84150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson */ 84250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testGzipEncodingEnabledByDefault() throws IOException, InterruptedException { 84350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 84450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody(gzip("ABCABCABC".getBytes("UTF-8"))) 84550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: gzip")); 84650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 84750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 84850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 84950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCABCABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 8508116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson assertNull(connection.getContentEncoding()); 85150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 85250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 85350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 85450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 85550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 856deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testClientConfiguredGzipContentEncoding() throws Exception { 857deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(new MockResponse() 858deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .setBody(gzip("ABCDEFGHIJKLMNOPQRSTUVWXYZ".getBytes("UTF-8"))) 859deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .addHeader("Content-Encoding: gzip")); 860deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 861deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 862deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 863deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 864deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 865deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("ABCDEFGHIJKLMNOPQRSTUVWXYZ", readAscii(gunzippedIn, Integer.MAX_VALUE)); 866deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 867deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson RecordedRequest request = server.takeRequest(); 868deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 869deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 870deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 871deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithFixedLength() throws Exception { 872deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.FIXED_LENGTH); 873deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 874deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 875deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithChunkedEncoding() throws Exception { 876deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.CHUNKED); 877deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 878deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 87950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testClientConfiguredCustomContentEncoding() throws Exception { 88050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 88150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody("ABCDE") 88250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: custom")); 88350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 88450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 88550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 88650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson connection.addRequestProperty("Accept-Encoding", "custom"); 88750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCDE", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 88850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 88950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 89050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: custom"); 89150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 89250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 893deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 894deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Test a bug where gzip input streams weren't exhausting the input stream, 895deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * which corrupted the request that followed. 896deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * http://code.google.com/p/android/issues/detail?id=7059 897deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 898deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson private void testClientConfiguredGzipContentEncodingAndConnectionReuse( 899deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson TransferKind transferKind) throws Exception { 900deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseOne = new MockResponse(); 901deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson responseOne.addHeader("Content-Encoding: gzip"); 902deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseOne, gzip("one (gzipped)".getBytes("UTF-8")), 5); 903deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseOne); 904deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseTwo = new MockResponse(); 905deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseTwo, "two (identity)", 5); 906deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseTwo); 907deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 908deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 909deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 910deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 911deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 912deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("one (gzipped)", readAscii(gunzippedIn, Integer.MAX_VALUE)); 913deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 914deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 915deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection = server.getUrl("/").openConnection(); 916deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("two (identity)", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 917deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 918deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 919deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 920deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 921ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * Obnoxiously test that the chunk sizes transmitted exactly equal the 922ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * requested data+chunk header size. Although setChunkedStreamingMode() 923ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * isn't specific about whether the size applies to the data or the 924ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * complete chunk, the RI interprets it as a complete chunk. 925ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson */ 926ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testSetChunkedStreamingMode() throws IOException, InterruptedException { 927ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse()); 928ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 929ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 930ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 931ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setChunkedStreamingMode(8); 932ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setDoOutput(true); 933ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = urlConnection.getOutputStream(); 934ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write("ABCDEFGHIJKLMNOPQ".getBytes("US-ASCII")); 935ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 936ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 937ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 938ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("ABCDEFGHIJKLMNOPQ", new String(request.getBody(), "US-ASCII")); 939ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.asList(3, 3, 3, 3, 3, 2), request.getChunkSizes()); 940ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 941ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 942ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithFixedLengthStreaming() throws Exception { 943ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.FIXED_LENGTH); 944ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 945ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 946ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithChunkedStreaming() throws Exception { 947ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.CHUNKED); 948ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 949ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 950ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void testAuthenticateWithStreamingPost(StreamingMode streamingMode) throws Exception { 951ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 952ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 953ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 954ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 955ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 956ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 957ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 958ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 959ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 960ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 961ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 962ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 963ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 964ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 965ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setChunkedStreamingMode(0); 966ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 967ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 968ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 969ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 970ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson try { 971ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.getInputStream(); 972ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail(); 973ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } catch (HttpRetryException expected) { 974ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 975ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 976ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the request... 977ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 978ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 979ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 980ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 981ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 982ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson enum StreamingMode { 983ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson FIXED_LENGTH, CHUNKED 984ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 985ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 986ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithPost() throws Exception { 987ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 988ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 989ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 990ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 991ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 992ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 993ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 994ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 995ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 996ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 997ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 998ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 999ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1000ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1001ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1002ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1003ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1004ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1005ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1006ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1007ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1008ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1009ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1010ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1011ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1012da289bcd0a9e207cc03c752f7c21c9004056e179Jesse Wilson // ...but the three requests that follow include an authorization header 1013ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1014ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1015ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 1016ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1017ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1018ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1019ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1020ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1021ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1022ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithGet() throws Exception { 1023ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1024ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1025ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1026ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1027ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1028ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1029ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1030ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1031ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1032ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1033ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1034ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1035ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1036ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1037ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1038ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1039ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1040ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1041ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1042ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1043ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...but the three requests that follow requests include an authorization header 1044ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1045ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1046ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("GET / HTTP/1.1", request.getRequestLine()); 1047ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1048ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1049ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1050ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1051ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1052c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithChunkedEncoding() throws Exception { 1053c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.CHUNKED, true); 1054c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1055c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1056c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithContentLengthHeader() throws Exception { 1057c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.FIXED_LENGTH, true); 1058c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1059c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1060c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithNoLengthHeaders() throws Exception { 1061c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.END_OF_STREAM, false); 1062c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1063c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1064c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private void testRedirected(TransferKind transferKind, boolean reuse) throws Exception { 1065c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse response = new MockResponse() 1066c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1067c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo"); 1068c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson transferKind.setBody(response, "This page has moved!", 10); 1069c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(response); 1070c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1071c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1072c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1073c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1074c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1075c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1076c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1077c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1078c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1079c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1080c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1081c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson if (reuse) { 1082c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1083c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1084c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1085c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1086c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedOnHttps() throws IOException, InterruptedException { 1087c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1088059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1089c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1090c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1091c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo") 1092c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1093c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1094c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1095c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1096c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1097059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1098c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1099c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1100c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1101c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1102c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1103c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1104c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1105c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1106c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1107c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1108c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpsToHttp() throws IOException, InterruptedException { 1109c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1110059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1111c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1112c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1113c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: http://anyhost/foo") 1114c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1115c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1116c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1117c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1118059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1119c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1120c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1121c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1122c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1123c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpToHttps() throws IOException, InterruptedException { 1124c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1125c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1126c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: https://anyhost/foo") 1127c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1128c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1129c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1130c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1131c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1132c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1133c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1134c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1135c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectToAnotherOriginServer() throws Exception { 1136c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockWebServer server2 = new MockWebServer(); 1137c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.enqueue(new MockResponse().setBody("This is the 2nd server!")); 1138c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.play(); 1139c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1140c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1141c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1142c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: " + server2.getUrl("/").toString()) 1143c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1144c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the first server again!")); 1145c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1146c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1147c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1148c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the 2nd server!", 1149c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1150c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(server2.getUrl("/"), connection.getURL()); 1151c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1152c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson // make sure the first server was careful to recycle the connection 1153c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the first server again!", 1154c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(server.getUrl("/").openStream(), Integer.MAX_VALUE)); 1155c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1156c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1157c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertContains(first.getHeaders(), "Host: localhost:" + server.getPort()); 1158c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest second = server2.takeRequest(); 1159c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertContains(second.getHeaders(), "Host: localhost:" + server2.getPort()); 1160c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest third = server.takeRequest(); 1161c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, third.getSequenceNumber()); 1162c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1163c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.shutdown(); 1164c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1165c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1166c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testHttpsWithCustomTrustManager() throws Exception { 1167c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 1168c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingTrustManager trustManager = new RecordingTrustManager(); 1169c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLContext sc = SSLContext.getInstance("TLS"); 1170c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson sc.init(null, new TrustManager[] { trustManager }, new java.security.SecureRandom()); 1171c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1172c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HostnameVerifier defaultHostnameVerifier = HttpsURLConnection.getDefaultHostnameVerifier(); 1173c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(hostnameVerifier); 1174c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLSocketFactory defaultSSLSocketFactory = HttpsURLConnection.getDefaultSSLSocketFactory(); 1175c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory()); 1176c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson try { 1177c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1178059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1179c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1180c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 1181c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("GHI")); 1182c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1183c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1184c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URL url = server.getUrl("/"); 1185c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("ABC", readAscii(url.openStream(), Integer.MAX_VALUE)); 1186c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("DEF", readAscii(url.openStream(), Integer.MAX_VALUE)); 1187c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GHI", readAscii(url.openStream(), Integer.MAX_VALUE)); 1188c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1189c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(Arrays.asList("verify localhost"), hostnameVerifier.calls); 11904559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals(Arrays.asList("checkServerTrusted [" 11914559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "CN=localhost 1, " 11924559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "CN=Test Intermediate Certificate Authority 1, " 11934559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "CN=Test Root Certificate Authority 1" 11944559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "] RSA"), 1195c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson trustManager.calls); 1196c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } finally { 1197c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(defaultHostnameVerifier); 1198c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(defaultSSLSocketFactory); 1199c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1200c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1201c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1202eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testConnectTimeouts() throws IOException { 1203eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson // 10.0.0.0 is non-routable and will time out on every network 1204eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = new URL("http://10.0.0.0/").openConnection(); 1205eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setConnectTimeout(1000); 1206eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1207eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.getInputStream(); 1208eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1209eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1210eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1211eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1212eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1213eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testReadTimeouts() throws IOException { 1214eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson /* 1215eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * This relies on the fact that MockWebServer doesn't close the 1216eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * connection after a response has been sent. This causes the client to 1217eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * try to read more bytes than are sent, which results in a timeout. 1218eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson */ 1219eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson MockResponse timeout = new MockResponse() 1220eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .setBody("ABC") 1221eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .clearHeaders() 1222eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .addHeader("Content-Length: 4"); 1223eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.enqueue(timeout); 1224eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.play(); 1225eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1226eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 1227eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setReadTimeout(1000); 1228eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson InputStream in = urlConnection.getInputStream(); 1229eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('A', in.read()); 1230eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('B', in.read()); 1231eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('C', in.read()); 1232eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1233eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson in.read(); // if Content-Length was accurate, this would return -1 immediately 1234eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1235eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1236eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1237eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1238eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1239125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson public void testSetChunkedEncodingAsRequestProperty() throws IOException, InterruptedException { 1240125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.enqueue(new MockResponse()); 1241125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.play(); 1242125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1243125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1244125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setRequestProperty("Transfer-encoding", "chunked"); 1245125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setDoOutput(true); 1246125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.getOutputStream().write("ABC".getBytes("UTF-8")); 1247125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1248125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1249125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson RecordedRequest request = server.takeRequest(); 1250125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals("ABC", new String(request.getBody(), "UTF-8")); 1251125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson } 1252125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1253f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInRequest() throws IOException, InterruptedException { 1254f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); // server doesn't honor the connection: close header! 1255f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1256f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1257f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1258f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1259f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson a.setRequestProperty("Connection", "close"); 1260f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1261f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1262f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1263f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1264f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1265f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1266f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1267f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1268f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1269f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1270f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInResponse() throws IOException, InterruptedException { 1271f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().addHeader("Connection: close")); 1272f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1273f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1274f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1275f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1276f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1277f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1278f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1279f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1280f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1281f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1282f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1283f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1284f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1285f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1286f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseWithRedirect() throws IOException, InterruptedException { 1287f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson MockResponse response = new MockResponse() 1288f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1289f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Location: /foo") 1290f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Connection: close"); 1291f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(response); 1292f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1293f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1294f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1295f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1296f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("This is the new location!", 1297f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1298f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1299f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1300f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1301f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1302f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1303f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 130465d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson public void testResponseCodeDisagreesWithHeaders() throws IOException, InterruptedException { 130565d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.enqueue(new MockResponse() 130665d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setResponseCode(HttpURLConnection.HTTP_NO_CONTENT) 130765d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setBody("This body is not allowed!")); 130865d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.play(); 130965d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 131065d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 131165d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson assertEquals("This body is not allowed!", 131265d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 131365d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson } 131465d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 1315ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson /** 1316deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Encodes the response body using GZIP and adds the corresponding header. 1317deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1318deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public byte[] gzip(byte[] bytes) throws IOException { 1319deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1320deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson OutputStream gzippedOut = new GZIPOutputStream(bytesOut); 1321deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.write(bytes); 1322deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.close(); 1323deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson return bytesOut.toByteArray(); 1324deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1325deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1326c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson /** 1327c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * Reads at most {@code limit} characters from {@code in} and asserts that 1328c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * content equals {@code expected}. 1329c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson */ 1330c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection, int limit) 1331c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson throws IOException { 133251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(expected, readAscii(connection.getInputStream(), limit)); 1333c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ((HttpURLConnection) connection).disconnect(); 1334c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1335c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 1336c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection) throws IOException { 1337c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent(expected, connection, Integer.MAX_VALUE); 1338c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1339c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 134060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson private void assertContains(List<String> headers, String header) { 134160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertTrue(headers.toString(), headers.contains(header)); 134260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 134351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1344ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void assertContainsNoneMatching(List<String> headers, String pattern) { 1345ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (String header : headers) { 1346ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (header.matches(pattern)) { 1347ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail("Header " + header + " matches " + pattern); 1348ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1349ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1350ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1351ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1352eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson private Set<String> newSet(String... elements) { 135383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return new HashSet<String>(Arrays.asList(elements)); 135483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 135583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 135651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson enum TransferKind { 135751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson CHUNKED() { 1358deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) 135951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException { 136051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setChunkedBody(content, chunkSize); 136151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 136251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 136351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson FIXED_LENGTH() { 1364deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 136551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 136651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 136751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 136851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson END_OF_STREAM() { 1369deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 137051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 137151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 137251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (Iterator<String> h = response.getHeaders().iterator(); h.hasNext(); ) { 137351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (h.next().startsWith("Content-Length:")) { 137451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson h.remove(); 137551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 137651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 137751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 137851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 137951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }; 138051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1381deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson abstract void setBody(MockResponse response, byte[] content, int chunkSize) 138251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException; 1383deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1384deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson void setBody(MockResponse response, String content, int chunkSize) throws IOException { 1385deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson setBody(response, content.getBytes("UTF-8"), chunkSize); 1386deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 138751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 1388c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1389c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingTrustManager implements X509TrustManager { 1390c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1391c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1392c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public X509Certificate[] getAcceptedIssuers() { 1393c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("getAcceptedIssuers"); 1394c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return new X509Certificate[] {}; 1395c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1396c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1397c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkClientTrusted(X509Certificate[] chain, String authType) 1398c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1399c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkClientTrusted " + certificatesToString(chain) + " " + authType); 1400c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1401c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1402c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkServerTrusted(X509Certificate[] chain, String authType) 1403c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1404c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkServerTrusted " + certificatesToString(chain) + " " + authType); 1405c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1406c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1407c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private String certificatesToString(X509Certificate[] certificates) { 1408c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson List<String> result = new ArrayList<String>(); 1409c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson for (X509Certificate certificate : certificates) { 1410c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson result.add(certificate.getSubjectDN() + " " + certificate.getSerialNumber()); 1411c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1412c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return result.toString(); 1413c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1414c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1415c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1416c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingHostnameVerifier implements HostnameVerifier { 1417c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1418c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1419c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public boolean verify(String hostname, SSLSession session) { 1420c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("verify " + hostname); 1421c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return true; 1422c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1423c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1424e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes} 1425