URLConnectionTest.java revision 977a9954414ec41256b218e6278a8544ea135d45
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 17e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughespackage 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; 25c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.CertificateException; 26c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.X509Certificate; 2751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.ArrayList; 2802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.util.Arrays; 2951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Collections; 3083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.HashSet; 3151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Iterator; 326247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.util.List; 3383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.Map; 3451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Set; 3583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.concurrent.atomic.AtomicReference; 36deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPInputStream; 37deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPOutputStream; 3860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HostnameVerifier; 3960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HttpsURLConnection; 40c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLContext; 41096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilsonimport javax.net.ssl.SSLException; 4260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.SSLSession; 43c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLSocketFactory; 4460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.TestSSLContext; 45c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.TrustManager; 46c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.X509TrustManager; 4751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport tests.http.DefaultResponseCache; 4860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockResponse; 4960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockWebServer; 5060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.RecordedRequest; 51e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 52e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughespublic class URLConnectionTest extends junit.framework.TestCase { 53b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 54ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private static final Authenticator SIMPLE_AUTHENTICATOR = new Authenticator() { 55ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson protected PasswordAuthentication getPasswordAuthentication() { 56ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson return new PasswordAuthentication("username", "password".toCharArray()); 57ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 58ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson }; 59ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 6051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockWebServer server = new MockWebServer(); 6151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 6251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson @Override protected void tearDown() throws Exception { 6351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(null); 64ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(null); 6551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); 6651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 6751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 6883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testRequestHeaders() throws IOException, InterruptedException { 6983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse()); 7083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 7183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 7283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 7383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "e"); 7483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "f"); 7583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders = urlConnection.getRequestProperties(); 7683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("e", "f"), new HashSet<String>(requestHeaders.get("D"))); 7783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 7883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.put("G", Arrays.asList("h")); 7983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 8083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 8183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 8283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 8383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.get("D").add("i"); 8483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 8583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 8683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 8783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 8883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty(null, "j"); 8983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 9083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 9183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 9283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 9383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty(null, "k"); 9483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 9583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 9683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 9783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("NullValue", null); // should fail silently! 9883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("AnotherNullValue", null); // should fail silently! 9983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 10083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.getResponseCode(); 10183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson RecordedRequest request = server.takeRequest(); 10283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: e"); 10383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: f"); 10483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "NullValue.*"); 10583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "AnotherNullValue.*"); 10683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "G:.*"); 10783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "null:.*"); 10883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 10983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("N", "o"); 11183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 11283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 11383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 11483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("P", "q"); 11683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 11783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 11883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 11983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 12183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseHeaders() throws IOException, InterruptedException { 12283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse() 12383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setStatus("HTTP/1.0 200 Fantastic") 12483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: b") 12583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: c") 12683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8)); 12783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 12883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 12983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 13083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 13183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 13283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> responseHeaders = urlConnection.getHeaderFields(); 13383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("b", "c"), new HashSet<String>(responseHeaders.get("A"))); 13483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 13583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.put("N", Arrays.asList("o")); 13683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 13783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 13883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 14083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.get("A").add("d"); 14183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 14283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 14383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 14483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 14583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 146e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // Check that if we don't read to the end of a response, the next request on the 147e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // recycled connection doesn't get the unread tail of the first request's response. 148e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // http://code.google.com/p/android/issues/detail?id=2939 149e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes public void test_2939() throws Exception { 150b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8); 151b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 152b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 153b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 154b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 155b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 156c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 157c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 1588baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson } 1598baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson 160977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // Check that we recognize a few basic mime types by extension. 161977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // http://code.google.com/p/android/issues/detail?id=10100 162977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes public void test_10100() throws Exception { 163977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("image/jpeg", URLConnection.guessContentTypeFromName("someFile.jpg")); 164977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("application/pdf", URLConnection.guessContentTypeFromName("stuff.pdf")); 165977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes } 166977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes 1678baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson public void testConnectionsArePooled() throws Exception { 168b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setBody("ABCDEFGHIJKLMNOPQR"); 169b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 170b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 171b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 172b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 173b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 174b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 175c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 176c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 177c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 178c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 179c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 180c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 181c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 182c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 183c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testChunkedConnectionsArePooled() throws Exception { 184c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDEFGHIJKLMNOPQR", 5); 185c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 186c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 187c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 188c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 189c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 190c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 191c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 192b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 193c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 194b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 195c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 196b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 197e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes } 19802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 199b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson enum WriteKind { BYTE_BY_BYTE, SMALL_BUFFERS, LARGE_BUFFERS } 20002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 20102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_byteByByte() throws Exception { 20251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.BYTE_BY_BYTE); 20302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 20402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 20502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_smallBuffers() throws Exception { 20651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.SMALL_BUFFERS); 20702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 20802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 20902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_largeBuffers() throws Exception { 21051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.LARGE_BUFFERS); 21102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 21202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 21302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_byteByByte() throws Exception { 21451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.BYTE_BY_BYTE); 21502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 21602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 21702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_smallBuffers() throws Exception { 21851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.SMALL_BUFFERS); 21902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 22002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 22102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_largeBuffers() throws Exception { 22251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.LARGE_BUFFERS); 22302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 22402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 22551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void doUpload(TransferKind uploadKind, WriteKind writeKind) throws Exception { 22602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes int n = 512*1024; 227b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.setBodyLimit(0); 228b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse()); 229b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 230b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 231b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson HttpURLConnection conn = (HttpURLConnection) server.getUrl("/").openConnection(); 23202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setDoOutput(true); 23302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setRequestMethod("POST"); 23451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 23502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setChunkedStreamingMode(-1); 23602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 23702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setFixedLengthStreamingMode(n); 23802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes OutputStream out = conn.getOutputStream(); 24002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes if (writeKind == WriteKind.BYTE_BY_BYTE) { 24102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; ++i) { 24202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write('x'); 24302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 24402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 24502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes byte[] buf = new byte[writeKind == WriteKind.SMALL_BUFFERS ? 256 : 64*1024]; 24602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes Arrays.fill(buf, (byte) 'x'); 24702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; i += buf.length) { 24802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write(buf, 0, Math.min(buf.length, n - i)); 24902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 25002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 25102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.close(); 2524cb7f05dc68abb23ae54a5891c369062185f2210Elliott Hughes assertEquals(200, conn.getResponseCode()); 253b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson RecordedRequest request = server.takeRequest(); 254b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(n, request.getBodySize()); 25551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 256b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().size() > 0); 257b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } else { 258b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().isEmpty()); 259b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } 26002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 2616247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 26251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 26351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Test that response caching is consistent with the RI and the spec. 26451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.4 26551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 2666247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching() throws Exception { 2676247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // Test each documented HTTP/1.1 code, plus the first unused value in each range. 2686247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html 2696247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 2706247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // We can't test 100 because it's not really a response. 2716247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // assertCached(false, 100); 2726247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 101); 2736247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 102); 2746247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 200); 2756247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 201); 2766247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 202); 2776247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 203); 2786247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 204); 2796247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 205); 2806247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 206); 2816247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 207); 28251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_300.) 2836247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 301); 2846247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 302; i <= 308; ++i) { 2856247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 2866247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 2876247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 400; i <= 406; ++i) { 2886247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 2896247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 2906247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // (See test_responseCaching_407.) 2916247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 408); 2926247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 409); 29351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_410.) 2946247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 411; i <= 418; ++i) { 2956247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 2966247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 2976247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 500; i <= 506; ++i) { 2986247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 2996247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3006247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3016247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 30251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_300() throws Exception { 30351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: fix this for android 30451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 300); 30551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 30651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3076247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching_407() throws Exception { 3086247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // This test will fail on Android because we throw if we're not using a proxy. 3096247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // This isn't true of the RI, but it seems like useful debugging behavior. 3106247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 407); 3116247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3126247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 31351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_410() throws Exception { 31451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // the HTTP spec permits caching 410s, but the RI doesn't. 31551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 410); 31651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 31751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3186247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes private void assertCached(boolean shouldPut, int responseCode) throws Exception { 31951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server = new MockWebServer(); 320b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse() 321b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .setResponseCode(responseCode) 32251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("ABCDE") 323b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .addHeader("WWW-Authenticate: challenge")); 324b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 325b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 32651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache responseCache = new DefaultResponseCache(); 32751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(responseCache); 32851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 32951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3306247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertEquals(responseCode, conn.getResponseCode()); 3316247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 33251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // exhaust the content stream 33351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 33451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: remove special case once testUnauthorizedResponseHandling() is fixed 33551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (responseCode != 401) { 33651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(conn.getInputStream(), Integer.MAX_VALUE); 33751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 33851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException ignored) { 33951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 34051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 34151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson Set<URI> expectedCachedUris = shouldPut 34251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ? Collections.singleton(url.toURI()) 34351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson : Collections.<URI>emptySet(); 34451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(Integer.toString(responseCode), 34551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson expectedCachedUris, responseCache.getContents().keySet()); 34651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); // tearDown() isn't sufficient; this test starts multiple servers 3476247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 34860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 34960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttps() throws IOException, InterruptedException { 35060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 35160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 352059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 353c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 35460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson server.play(); 35560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 356096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 357059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.serverContext.getSocketFactory()); 35860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 359c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via HTTPS", connection); 36060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 36160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson RecordedRequest request = server.takeRequest(); 36260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 36360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 36460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 365096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson public void testConnectViaHttpsReusingConnections() throws IOException, InterruptedException { 366096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 367096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 368059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 369096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 370096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 371096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.play(); 372096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 373096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson // install a custom SSL socket factory so the server can be authorized 374096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 375059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 376096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertContent("this response comes via HTTPS", connection); 377096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 378096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson // without an SSL socket factory, the connection should fail 379096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 380096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson try { 381096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 382096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson fail(); 383096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } catch (SSLException expected) { 384096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 385096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 386096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 38760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaProxy() throws IOException, InterruptedException { 388c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse mockResponse = new MockResponse().setBody("this response comes via a proxy"); 38951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(mockResponse); 39051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 39160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 39260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URLConnection connection = new URL("http://android.com/foo").openConnection( 39351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.toProxyAddress()); 394c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a proxy", connection); 39560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 39651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest request = server.takeRequest(); 39760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET http://android.com/foo HTTP/1.1", request.getRequestLine()); 39860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(request.getHeaders(), "Host: android.com"); 39960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 40060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 401c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithContentLengthHeader() throws IOException { 402c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(new MockResponse() 403c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .setBody("abc\r\nYOU SHOULD NOT SEE THIS") 404c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .clearHeaders() 405c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .addHeader("Content-Length: 3")); 406c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 407c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 408c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 409c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 410c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 411c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithChunkedHeader() throws IOException { 412c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse mockResponse = new MockResponse(); 413c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setChunkedBody("abc", 3); 414c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 415c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write(mockResponse.getBody()); 416c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write("\r\nYOU SHOULD NOT SEE THIS".getBytes()); 417c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setBody(bytesOut.toByteArray()); 418c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.clearHeaders(); 419c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.addHeader("Transfer-encoding: chunked"); 420c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 421c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(mockResponse); 422c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 423c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 424c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 425c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 426c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 42760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttpProxyToHttps() throws IOException, InterruptedException { 42860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 42960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 430059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 431c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 432c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via a secure proxy")); 43351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 43460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 43560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URL url = new URL("https://android.com/foo"); 43660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) url.openConnection( 43751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.toProxyAddress()); 438059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 43960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson connection.setHostnameVerifier(new HostnameVerifier() { 44060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public boolean verify(String hostname, SSLSession session) { 44160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson return true; 44260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 44360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson }); 44460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 445c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a secure proxy", connection); 44660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 44751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest connect = server.takeRequest(); 44860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("Connect line failure on proxy", 44960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson "CONNECT android.com:443 HTTP/1.1", connect.getRequestLine()); 45060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 45160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 45251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest get = server.takeRequest(); 45360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", get.getRequestLine()); 45460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(get.getHeaders(), "Host: android.com"); 45560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 45660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 45751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithFixedLength() throws IOException { 45851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.FIXED_LENGTH); 45951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 46051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 46151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithChunkedEncoding() throws IOException { 46251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.CHUNKED); 46351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 46451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 46551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithNoLengthHeaders() throws IOException { 46651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.END_OF_STREAM); 46751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 46851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 46951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 47051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * HttpURLConnection.getInputStream().skip(long) causes ResponseCache corruption 47151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://code.google.com/p/android/issues/detail?id=8175 47251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 47351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testResponseCaching(TransferKind transferKind) throws IOException { 47451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 47551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "I love puppies but hate spiders", 1); 47651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 47751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 47851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 47951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 48051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 48151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 48251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // Make sure that calling skip() doesn't omit bytes from the cache. 48351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 48451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = urlConnection.getInputStream(); 48551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love ", readAscii(in, "I love ".length())); 48651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reliableSkip(in, "puppies but hate ".length()); 48751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("spiders", readAscii(in, "spiders".length())); 48851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 48951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 49051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 491096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 49251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 49351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson urlConnection = server.getUrl("/").openConnection(); // this response is cached! 49451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in = urlConnection.getInputStream(); 49551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love puppies but hate spiders", 49651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(in, "I love puppies but hate spiders".length())); 49751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 49851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getMissCount()); 49951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 500096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(1, cache.getSuccessCount()); 501096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 50251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 50351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 50483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseCacheRequestHeaders() throws IOException, URISyntaxException { 50583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 50683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 50783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 50883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson final AtomicReference<Map<String, List<String>>> requestHeadersRef 50983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson = new AtomicReference<Map<String, List<String>>>(); 51083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 51183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 51283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 51383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeadersRef.set(requestHeaders); 51483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 51583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 51683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection conn) throws IOException { 51783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 51883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 51983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson }); 52083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 52183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URL url = server.getUrl("/"); 52283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URLConnection urlConnection = url.openConnection(); 52383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("A", "android"); 52483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson readAscii(urlConnection.getInputStream(), Integer.MAX_VALUE); 52583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(Arrays.asList("android"), requestHeadersRef.get().get("A")); 52683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 52783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 52851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void reliableSkip(InputStream in, int length) throws IOException { 52951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson while (length > 0) { 53051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson length -= in.skip(length); 53151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 53251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 53351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 53451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 53551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Reads {@code count} characters from the stream. If the stream is 53651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * exhausted before {@code count} characters can be read, the remaining 53751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * characters are returned and the stream is closed. 53851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 53951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private String readAscii(InputStream in, int count) throws IOException { 54051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson StringBuilder result = new StringBuilder(); 54151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (int i = 0; i < count; i++) { 54251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson int value = in.read(); 54351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (value == -1) { 54451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 54551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 54651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 54751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson result.append((char) value); 54851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 54951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return result.toString(); 55051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 55151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 55251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithContentLengthHeader() throws IOException { 55351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.FIXED_LENGTH); 55451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 55551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 55651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithChunkedEncoding() throws IOException { 55751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.CHUNKED); 55851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 55951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 56051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithNoLengthHeaders() throws IOException { 56151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /* 56251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Intentionally empty. This case doesn't make sense because there's no 56351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * such thing as a premature disconnect when the disconnect itself 56451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * indicates the end of the data stream. 56551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 56651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 56751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 56851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testServerPrematureDisconnect(TransferKind transferKind) throws IOException { 56951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 57051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 16); 57151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(truncateViolently(response, 16)); 57251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 57351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 57451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 57551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 57651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 57751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 57851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson BufferedReader reader = new BufferedReader(new InputStreamReader( 57951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.getUrl("/").openConnection().getInputStream())); 58051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", reader.readLine()); 58151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 58251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reader.readLine(); 58351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("This implementation silently ignored a truncated HTTP body."); 58451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 58551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 58651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 58751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 58851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 58951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 59051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 59151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 59251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 59351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 59451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithContentLengthHeader() throws IOException { 59551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.FIXED_LENGTH); 59651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 59751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 59851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithChunkedEncoding() throws IOException { 59951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.CHUNKED); 60051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 60151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 60251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithNoLengthHeaders() throws IOException { 60351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.END_OF_STREAM); 60451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 60551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 60651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testClientPrematureDisconnect(TransferKind transferKind) throws IOException { 60751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 60851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 1024); 60951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 61051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 61151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 61251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 61351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 61451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 61551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 61651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 61751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 61851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 61951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 62051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.read(); 62151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("Expected an IOException because the stream is closed."); 62251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 62351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 62451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 62551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 62651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 62751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 62851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 62951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 63051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 63151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 63251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 63351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Shortens the body of {@code response} but not the corresponding headers. 63451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Only useful to test how clients respond to the premature conclusion of 63551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the HTTP body. 63651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 63751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) { 63851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 63951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson List<String> headers = new ArrayList<String>(response.getHeaders()); 64051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(Arrays.copyOfRange(response.getBody(), 0, numBytesToKeep)); 64151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().clear(); 64251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().addAll(headers); 64351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return response; 64451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 64551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 64651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithContentLengthHeader() throws IOException { 64751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.FIXED_LENGTH); 64851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 64951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithChunkedEncoding() throws IOException { 65151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.CHUNKED); 65251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 65351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithNoLengthHeaders() throws IOException { 65551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.END_OF_STREAM); 65651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 65751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndReset(TransferKind transferKind) throws IOException { 65951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 66051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDEFGHIJKLMNOPQRSTUVWXYZ", 1024); 66151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 66251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 66351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 66451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 66551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 66651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 66751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 66851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertFalse("This implementation claims to support mark().", in.markSupported()); 66951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.mark(5); 67051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 67151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 67251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.reset(); 67351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail(); 67451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 67551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 67651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("FGHIJKLMNOPQRSTUVWXYZ", readAscii(in, Integer.MAX_VALUE)); 67751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 67851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQRSTUVWXYZ", server.getUrl("/").openConnection()); 67951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 68051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 68151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 68251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 68351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 68451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * We've had a bug where we forget the HTTP response when we see response 68551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * code 401. This causes a new HTTP request to be issued for every call into 68651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the URLConnection. 68751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 68851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testUnauthorizedResponseHandling() throws IOException { 68951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse() 69051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .addHeader("WWW-Authenticate: challenge") 69151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setResponseCode(401) // UNAUTHORIZED 69251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("Unauthorized"); 69351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 69451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 69551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 69651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 69751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 69851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 69951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 70051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 70151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 70251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 70351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 70451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, server.getRequestCount()); 70551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 70651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 7076906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testNonHexChunkSize() throws IOException { 7086906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 7096906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5\r\nABCDE\r\nG\r\nFGHIJKLMNOPQRSTU\r\n0\r\n\r\n") 7106906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 7116906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked")); 7126906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 7136906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 7146906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 7156906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 7166906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 7176906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 7186906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 7196906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 7206906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 7216906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 7226906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testMissingChunkBody() throws IOException { 7236906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 7246906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5") 7256906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 7266906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked") 7276906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setDisconnectAtEnd(true)); 7286906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 7296906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 7306906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 7316906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 7326906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 7336906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 7346906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 7356906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 7366906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 7376906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 738deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testClientConfiguredGzipContentEncoding() throws Exception { 739deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(new MockResponse() 740deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .setBody(gzip("ABCDEFGHIJKLMNOPQRSTUVWXYZ".getBytes("UTF-8"))) 741deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .addHeader("Content-Encoding: gzip")); 742deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 743deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 744deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 745deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 746deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 747deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("ABCDEFGHIJKLMNOPQRSTUVWXYZ", readAscii(gunzippedIn, Integer.MAX_VALUE)); 748deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 749deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson RecordedRequest request = server.takeRequest(); 750deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 751deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 752deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 753deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithFixedLength() throws Exception { 754deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.FIXED_LENGTH); 755deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 756deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 757deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithChunkedEncoding() throws Exception { 758deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.CHUNKED); 759deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 760deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 761deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 762deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Test a bug where gzip input streams weren't exhausting the input stream, 763deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * which corrupted the request that followed. 764deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * http://code.google.com/p/android/issues/detail?id=7059 765deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 766deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson private void testClientConfiguredGzipContentEncodingAndConnectionReuse( 767deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson TransferKind transferKind) throws Exception { 768deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseOne = new MockResponse(); 769deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson responseOne.addHeader("Content-Encoding: gzip"); 770deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseOne, gzip("one (gzipped)".getBytes("UTF-8")), 5); 771deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseOne); 772deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseTwo = new MockResponse(); 773deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseTwo, "two (identity)", 5); 774deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseTwo); 775deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 776deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 777deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 778deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 779deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 780deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("one (gzipped)", readAscii(gunzippedIn, Integer.MAX_VALUE)); 781deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 782deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 783deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection = server.getUrl("/").openConnection(); 784deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("two (identity)", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 785deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 786deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 787deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 788deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 789ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * Obnoxiously test that the chunk sizes transmitted exactly equal the 790ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * requested data+chunk header size. Although setChunkedStreamingMode() 791ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * isn't specific about whether the size applies to the data or the 792ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * complete chunk, the RI interprets it as a complete chunk. 793ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson */ 794ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testSetChunkedStreamingMode() throws IOException, InterruptedException { 795ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse()); 796ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 797ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 798ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 799ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setChunkedStreamingMode(8); 800ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setDoOutput(true); 801ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = urlConnection.getOutputStream(); 802ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write("ABCDEFGHIJKLMNOPQ".getBytes("US-ASCII")); 803ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 804ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 805ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 806ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("ABCDEFGHIJKLMNOPQ", new String(request.getBody(), "US-ASCII")); 807ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.asList(3, 3, 3, 3, 3, 2), request.getChunkSizes()); 808ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 809ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 810ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithFixedLengthStreaming() throws Exception { 811ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.FIXED_LENGTH); 812ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 813ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 814ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithChunkedStreaming() throws Exception { 815ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.CHUNKED); 816ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 817ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 818ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void testAuthenticateWithStreamingPost(StreamingMode streamingMode) throws Exception { 819ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 820ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 821ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 822ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 823ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 824ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 825ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 826ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 827ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 828ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 829ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 830ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 831ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 832ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 833ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setChunkedStreamingMode(0); 834ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 835ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 836ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 837ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 838ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson try { 839ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.getInputStream(); 840ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail(); 841ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } catch (HttpRetryException expected) { 842ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 843ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 844ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the request... 845ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 846ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 847ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 848ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 849ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 850ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson enum StreamingMode { 851ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson FIXED_LENGTH, CHUNKED 852ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 853ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 854ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithPost() throws Exception { 855ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 856ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 857ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 858ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 859ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 860ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 861ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 862ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 863ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 864ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 865ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 866ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 867ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 868ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 869ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 870ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 871ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 872ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 873ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 874ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 875ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 876ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 877ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 878ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 879ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 880da289bcd0a9e207cc03c752f7c21c9004056e179Jesse Wilson // ...but the three requests that follow include an authorization header 881ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 882ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 883ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 884ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 885ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 886ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 887ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 888ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 889ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 890ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithGet() throws Exception { 891ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 892ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 893ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 894ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 895ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 896ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 897ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 898ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 899ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 900ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 901ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 902ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 903ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 904ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 905ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 906ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 907ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 908ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 909ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 910ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 911ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...but the three requests that follow requests include an authorization header 912ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 913ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 914ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("GET / HTTP/1.1", request.getRequestLine()); 915ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 916ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 917ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 918ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 919ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 920c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithChunkedEncoding() throws Exception { 921c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.CHUNKED, true); 922c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 923c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 924c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithContentLengthHeader() throws Exception { 925c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.FIXED_LENGTH, true); 926c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 927c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 928c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithNoLengthHeaders() throws Exception { 929c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.END_OF_STREAM, false); 930c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 931c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 932c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private void testRedirected(TransferKind transferKind, boolean reuse) throws Exception { 933c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse response = new MockResponse() 934c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 935c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo"); 936c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson transferKind.setBody(response, "This page has moved!", 10); 937c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(response); 938c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 939c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 940c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 941c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 942c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 943c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 944c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 945c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 946c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 947c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 948c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 949c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson if (reuse) { 950c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 951c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 952c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 953c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 954c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedOnHttps() throws IOException, InterruptedException { 955c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 956059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 957c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 958c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 959c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo") 960c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 961c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 962c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 963c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 964c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 965059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 966c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 967c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 968c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 969c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 970c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 971c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 972c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 973c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 974c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 975c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 976c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpsToHttp() throws IOException, InterruptedException { 977c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 978059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 979c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 980c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 981c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: http://anyhost/foo") 982c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 983c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 984c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 985c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 986059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 987c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 988c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 989c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 990c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 991c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpToHttps() throws IOException, InterruptedException { 992c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 993c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 994c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: https://anyhost/foo") 995c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 996c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 997c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 998c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 999c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1000c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1001c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1002c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1003c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectToAnotherOriginServer() throws Exception { 1004c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockWebServer server2 = new MockWebServer(); 1005c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.enqueue(new MockResponse().setBody("This is the 2nd server!")); 1006c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.play(); 1007c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1008c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1009c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1010c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: " + server2.getUrl("/").toString()) 1011c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1012c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the first server again!")); 1013c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1014c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1015c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1016c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the 2nd server!", 1017c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1018c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(server2.getUrl("/"), connection.getURL()); 1019c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1020c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson // make sure the first server was careful to recycle the connection 1021c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the first server again!", 1022c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(server.getUrl("/").openStream(), Integer.MAX_VALUE)); 1023c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1024c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1025c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertContains(first.getHeaders(), "Host: localhost:" + server.getPort()); 1026c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest second = server2.takeRequest(); 1027c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertContains(second.getHeaders(), "Host: localhost:" + server2.getPort()); 1028c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest third = server.takeRequest(); 1029c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, third.getSequenceNumber()); 1030c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1031c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.shutdown(); 1032c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1033c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1034c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testHttpsWithCustomTrustManager() throws Exception { 1035c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 1036c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingTrustManager trustManager = new RecordingTrustManager(); 1037c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLContext sc = SSLContext.getInstance("TLS"); 1038c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson sc.init(null, new TrustManager[] { trustManager }, new java.security.SecureRandom()); 1039c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1040c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HostnameVerifier defaultHostnameVerifier = HttpsURLConnection.getDefaultHostnameVerifier(); 1041c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(hostnameVerifier); 1042c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLSocketFactory defaultSSLSocketFactory = HttpsURLConnection.getDefaultSSLSocketFactory(); 1043c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory()); 1044c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson try { 1045c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1046059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1047c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1048c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 1049c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("GHI")); 1050c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1051c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1052c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URL url = server.getUrl("/"); 1053c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("ABC", readAscii(url.openStream(), Integer.MAX_VALUE)); 1054c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("DEF", readAscii(url.openStream(), Integer.MAX_VALUE)); 1055c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GHI", readAscii(url.openStream(), Integer.MAX_VALUE)); 1056c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1057c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(Arrays.asList("verify localhost"), hostnameVerifier.calls); 1058c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(Arrays.asList("checkServerTrusted [CN=localhost 1] RSA"), 1059c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson trustManager.calls); 1060c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } finally { 1061c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(defaultHostnameVerifier); 1062c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(defaultSSLSocketFactory); 1063c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1064c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1065c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1066eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testConnectTimeouts() throws IOException { 1067eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson // 10.0.0.0 is non-routable and will time out on every network 1068eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = new URL("http://10.0.0.0/").openConnection(); 1069eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setConnectTimeout(1000); 1070eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1071eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.getInputStream(); 1072eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1073eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1074eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1075eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1076eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1077eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testReadTimeouts() throws IOException { 1078eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson /* 1079eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * This relies on the fact that MockWebServer doesn't close the 1080eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * connection after a response has been sent. This causes the client to 1081eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * try to read more bytes than are sent, which results in a timeout. 1082eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson */ 1083eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson MockResponse timeout = new MockResponse() 1084eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .setBody("ABC") 1085eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .clearHeaders() 1086eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .addHeader("Content-Length: 4"); 1087eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.enqueue(timeout); 1088eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.play(); 1089eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1090eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 1091eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setReadTimeout(1000); 1092eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson InputStream in = urlConnection.getInputStream(); 1093eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('A', in.read()); 1094eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('B', in.read()); 1095eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('C', in.read()); 1096eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1097eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson in.read(); // if Content-Length was accurate, this would return -1 immediately 1098eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1099eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1100eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1101eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1102eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1103ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson /** 1104deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Encodes the response body using GZIP and adds the corresponding header. 1105deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1106deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public byte[] gzip(byte[] bytes) throws IOException { 1107deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1108deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson OutputStream gzippedOut = new GZIPOutputStream(bytesOut); 1109deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.write(bytes); 1110deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.close(); 1111deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson return bytesOut.toByteArray(); 1112deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1113deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1114c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson /** 1115c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * Reads at most {@code limit} characters from {@code in} and asserts that 1116c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * content equals {@code expected}. 1117c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson */ 1118c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection, int limit) 1119c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson throws IOException { 112051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(expected, readAscii(connection.getInputStream(), limit)); 1121c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ((HttpURLConnection) connection).disconnect(); 1122c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1123c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 1124c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection) throws IOException { 1125c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent(expected, connection, Integer.MAX_VALUE); 1126c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1127c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 112860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson private void assertContains(List<String> headers, String header) { 112960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertTrue(headers.toString(), headers.contains(header)); 113060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 113151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1132ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void assertContainsNoneMatching(List<String> headers, String pattern) { 1133ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (String header : headers) { 1134ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (header.matches(pattern)) { 1135ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail("Header " + header + " matches " + pattern); 1136ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1137ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1138ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1139ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1140eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson private Set<String> newSet(String... elements) { 114183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return new HashSet<String>(Arrays.asList(elements)); 114283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 114383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 114451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson enum TransferKind { 114551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson CHUNKED() { 1146deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) 114751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException { 114851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setChunkedBody(content, chunkSize); 114951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 115051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 115151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson FIXED_LENGTH() { 1152deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 115351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 115451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 115551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 115651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson END_OF_STREAM() { 1157deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 115851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 115951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 116051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (Iterator<String> h = response.getHeaders().iterator(); h.hasNext(); ) { 116151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (h.next().startsWith("Content-Length:")) { 116251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson h.remove(); 116351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 116451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 116551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 116651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 116751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }; 116851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1169deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson abstract void setBody(MockResponse response, byte[] content, int chunkSize) 117051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException; 1171deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1172deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson void setBody(MockResponse response, String content, int chunkSize) throws IOException { 1173deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson setBody(response, content.getBytes("UTF-8"), chunkSize); 1174deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 117551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 1176c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1177c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingTrustManager implements X509TrustManager { 1178c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1179c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1180c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public X509Certificate[] getAcceptedIssuers() { 1181c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("getAcceptedIssuers"); 1182c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return new X509Certificate[] {}; 1183c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1184c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1185c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkClientTrusted(X509Certificate[] chain, String authType) 1186c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1187c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkClientTrusted " + certificatesToString(chain) + " " + authType); 1188c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1189c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1190c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkServerTrusted(X509Certificate[] chain, String authType) 1191c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1192c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkServerTrusted " + certificatesToString(chain) + " " + authType); 1193c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1194c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1195c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private String certificatesToString(X509Certificate[] certificates) { 1196c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson List<String> result = new ArrayList<String>(); 1197c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson for (X509Certificate certificate : certificates) { 1198c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson result.add(certificate.getSubjectDN() + " " + certificate.getSerialNumber()); 1199c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1200c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return result.toString(); 1201c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1202c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1203c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1204c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingHostnameVerifier implements HostnameVerifier { 1205c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1206c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1207c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public boolean verify(String hostname, SSLSession session) { 1208c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("verify " + hostname); 1209c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return true; 1210c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1211c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1212e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes} 1213