URLConnectionTest.java revision 984fcff696380abd6ea14e80030f9fd2d09dbad0
1e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes/* 2e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * Copyright (C) 2009 The Android Open Source Project 3f33eae7e84eb6d3b0f4e86b59605bb3de73009f3Elliott Hughes * 4e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * Licensed under the Apache License, Version 2.0 (the "License"); 5e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * you may not use this file except in compliance with the License. 6e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * You may obtain a copy of the License at 7f33eae7e84eb6d3b0f4e86b59605bb3de73009f3Elliott Hughes * 8e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * http://www.apache.org/licenses/LICENSE-2.0 9f33eae7e84eb6d3b0f4e86b59605bb3de73009f3Elliott Hughes * 10e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * Unless required by applicable law or agreed to in writing, software 11e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * distributed under the License is distributed on an "AS IS" BASIS, 12e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * See the License for the specific language governing permissions and 14e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes * limitations under the License. 15e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes */ 16e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 174557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonpackage libcore.java.net; 18e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 1951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.io.BufferedReader; 20c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilsonimport java.io.ByteArrayOutputStream; 216247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.io.IOException; 2251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.io.InputStream; 23e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughesimport java.io.InputStreamReader; 2402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.io.OutputStream; 254557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.Authenticator; 264557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.CacheRequest; 274557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.CacheResponse; 284557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpRetryException; 294557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpURLConnection; 3000feece22909b7dc79fc96d666d157390b93858eJesse Wilsonimport java.net.InetAddress; 314557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.PasswordAuthentication; 324557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.ResponseCache; 334557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.SocketTimeoutException; 344557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URI; 354557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URISyntaxException; 364557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URL; 374557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URLConnection; 38c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.CertificateException; 39c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.X509Certificate; 4051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.ArrayList; 4102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.util.Arrays; 4251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Collections; 4383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.HashSet; 4451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Iterator; 456247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.util.List; 4683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.Map; 4751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Set; 4883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.concurrent.atomic.AtomicReference; 49deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPInputStream; 50deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPOutputStream; 5160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HostnameVerifier; 5260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HttpsURLConnection; 53c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLContext; 54096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilsonimport javax.net.ssl.SSLException; 5560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.SSLSession; 56c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLSocketFactory; 57c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.TrustManager; 58c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.X509TrustManager; 5950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilsonimport libcore.javax.net.ssl.TestSSLContext; 6051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport tests.http.DefaultResponseCache; 6160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockResponse; 6260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockWebServer; 6360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.RecordedRequest; 64e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 65e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughespublic class URLConnectionTest extends junit.framework.TestCase { 66b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 67ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private static final Authenticator SIMPLE_AUTHENTICATOR = new Authenticator() { 68ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson protected PasswordAuthentication getPasswordAuthentication() { 69ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson return new PasswordAuthentication("username", "password".toCharArray()); 70ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 71ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson }; 72ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 7351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockWebServer server = new MockWebServer(); 7400feece22909b7dc79fc96d666d157390b93858eJesse Wilson private String hostname; 7500feece22909b7dc79fc96d666d157390b93858eJesse Wilson 7600feece22909b7dc79fc96d666d157390b93858eJesse Wilson @Override protected void setUp() throws Exception { 7700feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.setUp(); 7800feece22909b7dc79fc96d666d157390b93858eJesse Wilson hostname = InetAddress.getLocalHost().getHostName(); 7900feece22909b7dc79fc96d666d157390b93858eJesse Wilson } 8051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 8151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson @Override protected void tearDown() throws Exception { 8251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(null); 83ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(null); 84984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyHost"); 85984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyPort"); 86984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyHost"); 87984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyPort"); 88984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyHost"); 89984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyPort"); 9051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); 9100feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.tearDown(); 9251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 9351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 9483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testRequestHeaders() throws IOException, InterruptedException { 9583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse()); 9683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 9783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 9883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 9983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "e"); 10083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "f"); 10183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders = urlConnection.getRequestProperties(); 10283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("e", "f"), new HashSet<String>(requestHeaders.get("D"))); 10383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 10483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.put("G", Arrays.asList("h")); 10583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 10683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 10783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 10883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 10983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.get("D").add("i"); 11083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 11183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 11283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 11383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty(null, "j"); 11583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 11683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 11783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 11883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty(null, "k"); 12083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 12183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 12283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("NullValue", null); // should fail silently! 12483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("AnotherNullValue", null); // should fail silently! 12583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 12683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.getResponseCode(); 12783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson RecordedRequest request = server.takeRequest(); 12883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: e"); 12983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: f"); 13083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "NullValue.*"); 13183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "AnotherNullValue.*"); 13283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "G:.*"); 13383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "null:.*"); 13483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 13583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 13683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("N", "o"); 13783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 13883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 13983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 14083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 14183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("P", "q"); 14283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 14383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 14483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 14583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 14683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 14783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseHeaders() throws IOException, InterruptedException { 14883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse() 14983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setStatus("HTTP/1.0 200 Fantastic") 15083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: b") 15183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: c") 15283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8)); 15383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 15483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 15583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 15683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 15783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 15883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> responseHeaders = urlConnection.getHeaderFields(); 15983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("b", "c"), new HashSet<String>(responseHeaders.get("A"))); 16083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 16183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.put("N", Arrays.asList("o")); 16283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 16383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 16483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 16583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 16683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.get("A").add("d"); 16783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 16883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 16983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 17083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 17183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 172e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // Check that if we don't read to the end of a response, the next request on the 173e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // recycled connection doesn't get the unread tail of the first request's response. 174e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // http://code.google.com/p/android/issues/detail?id=2939 175e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes public void test_2939() throws Exception { 176b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8); 177b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 178b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 179b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 180b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 181b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 182c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 183c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 1848baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson } 1858baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson 186977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // Check that we recognize a few basic mime types by extension. 187977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // http://code.google.com/p/android/issues/detail?id=10100 188977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes public void test_10100() throws Exception { 189977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("image/jpeg", URLConnection.guessContentTypeFromName("someFile.jpg")); 190977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("application/pdf", URLConnection.guessContentTypeFromName("stuff.pdf")); 191977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes } 192977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes 1938baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson public void testConnectionsArePooled() throws Exception { 194b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setBody("ABCDEFGHIJKLMNOPQR"); 195b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 196b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 197b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 198b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 199b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 200b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 201c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 202c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 203c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 204c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 205c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 206c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 207c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 208c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 209c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testChunkedConnectionsArePooled() throws Exception { 210c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDEFGHIJKLMNOPQR", 5); 211c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 212c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 213c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 214c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 215c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 216c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 217c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 218b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 219c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 220b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 221c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/").openConnection()); 222b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 223e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes } 22402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 225b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson enum WriteKind { BYTE_BY_BYTE, SMALL_BUFFERS, LARGE_BUFFERS } 22602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 22702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_byteByByte() throws Exception { 22851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.BYTE_BY_BYTE); 22902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 23102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_smallBuffers() throws Exception { 23251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.SMALL_BUFFERS); 23302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 23502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_largeBuffers() throws Exception { 23651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.LARGE_BUFFERS); 23702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 23802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 23902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_byteByByte() throws Exception { 24051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.BYTE_BY_BYTE); 24102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 24202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 24302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_smallBuffers() throws Exception { 24451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.SMALL_BUFFERS); 24502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 24602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 24702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_largeBuffers() throws Exception { 24851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.LARGE_BUFFERS); 24902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 25002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 25151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void doUpload(TransferKind uploadKind, WriteKind writeKind) throws Exception { 25202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes int n = 512*1024; 253b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.setBodyLimit(0); 254b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse()); 255b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 256b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 257b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson HttpURLConnection conn = (HttpURLConnection) server.getUrl("/").openConnection(); 25802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setDoOutput(true); 25902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setRequestMethod("POST"); 26051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 26102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setChunkedStreamingMode(-1); 26202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 26302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setFixedLengthStreamingMode(n); 26402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 26502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes OutputStream out = conn.getOutputStream(); 26602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes if (writeKind == WriteKind.BYTE_BY_BYTE) { 26702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; ++i) { 26802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write('x'); 26902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 27102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes byte[] buf = new byte[writeKind == WriteKind.SMALL_BUFFERS ? 256 : 64*1024]; 27202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes Arrays.fill(buf, (byte) 'x'); 27302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; i += buf.length) { 27402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write(buf, 0, Math.min(buf.length, n - i)); 27502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.close(); 2784cb7f05dc68abb23ae54a5891c369062185f2210Elliott Hughes assertEquals(200, conn.getResponseCode()); 279b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson RecordedRequest request = server.takeRequest(); 280b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(n, request.getBodySize()); 28151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 282b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().size() > 0); 283b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } else { 284b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().isEmpty()); 285b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } 28602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 2876247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 28851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 28951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Test that response caching is consistent with the RI and the spec. 29051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.4 29151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 2926247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching() throws Exception { 2936247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // Test each documented HTTP/1.1 code, plus the first unused value in each range. 2946247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html 2956247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 2966247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // We can't test 100 because it's not really a response. 2976247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // assertCached(false, 100); 2986247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 101); 2996247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 102); 3006247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 200); 3016247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 201); 3026247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 202); 3036247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 203); 3046247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 204); 3056247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 205); 3066247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 206); 3076247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 207); 30851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_300.) 3096247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 301); 3106247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 302; i <= 308; ++i) { 3116247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3126247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3136247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 400; i <= 406; ++i) { 3146247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3156247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3166247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // (See test_responseCaching_407.) 3176247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 408); 3186247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 409); 31951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_410.) 3206247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 411; i <= 418; ++i) { 3216247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3226247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3236247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 500; i <= 506; ++i) { 3246247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3256247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3266247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3276247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 32851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_300() throws Exception { 32951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: fix this for android 33051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 300); 33151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 33251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3331f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson /** 3341f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * Response code 407 should only come from proxy servers. Android's client 3351f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * throws if it is sent by an origin server. 3361f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson */ 3371f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson public void testOriginServerSends407() throws Exception { 3381f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.enqueue(new MockResponse().setResponseCode(407)); 3391f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.play(); 3401f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson 3411f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson URL url = server.getUrl("/"); 3421f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3431f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson try { 3441f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson conn.getResponseCode(); 3451f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson fail(); 3461f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } catch (IOException expected) { 3471f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } 3486247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3496247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 35051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_410() throws Exception { 35151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // the HTTP spec permits caching 410s, but the RI doesn't. 35251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 410); 35351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 35451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3556247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes private void assertCached(boolean shouldPut, int responseCode) throws Exception { 35651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server = new MockWebServer(); 357b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse() 358b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .setResponseCode(responseCode) 35951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("ABCDE") 360b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .addHeader("WWW-Authenticate: challenge")); 361b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 362b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 36351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache responseCache = new DefaultResponseCache(); 36451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(responseCache); 36551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 36651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3676247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertEquals(responseCode, conn.getResponseCode()); 3686247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 36951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // exhaust the content stream 37051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 37151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: remove special case once testUnauthorizedResponseHandling() is fixed 37251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (responseCode != 401) { 37351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(conn.getInputStream(), Integer.MAX_VALUE); 37451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 37551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException ignored) { 37651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 37751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 37851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson Set<URI> expectedCachedUris = shouldPut 37951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ? Collections.singleton(url.toURI()) 38051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson : Collections.<URI>emptySet(); 38151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(Integer.toString(responseCode), 38251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson expectedCachedUris, responseCache.getContents().keySet()); 38351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); // tearDown() isn't sufficient; this test starts multiple servers 3846247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 38560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 38660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttps() throws IOException, InterruptedException { 38760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 38860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 389059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 390c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 39160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson server.play(); 39260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 393096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 3944559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 39560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 396c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via HTTPS", connection); 39760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 39860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson RecordedRequest request = server.takeRequest(); 39960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 40060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 40160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 402096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson public void testConnectViaHttpsReusingConnections() throws IOException, InterruptedException { 403096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 404096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 405059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 406096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 407096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 408096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.play(); 409096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 410b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 411b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 412b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("this response comes via HTTPS", connection); 413b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 414b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 415b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 416b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("another response via HTTPS", connection); 417b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 418b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 419b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 420b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson } 421b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 4228116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson public void testConnectViaHttpsReusingConnectionsDifferentFactories() 423b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson throws IOException, InterruptedException { 424b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 425b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 426b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 427b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 428b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 429b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.play(); 430b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 431096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson // install a custom SSL socket factory so the server can be authorized 432096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 433059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 434096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertContent("this response comes via HTTPS", connection); 435096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 436096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 437096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson try { 438096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 439b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson fail("without an SSL socket factory, the connection should fail"); 440096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } catch (SSLException expected) { 441096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 442096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 443096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 4444559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom public void testConnectViaHttpsWithSSLFallback() throws IOException, InterruptedException { 4454559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 4464559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4474559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 4484559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setDisconnectAtStart(true)); 4494559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via SSL")); 4504559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.play(); 4514559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4524559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 4534559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 4544559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4554559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertContent("this response comes via SSL", connection); 4564559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4574559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom RecordedRequest request = server.takeRequest(); 4584559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 4594559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom } 4604559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 461984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxyArg() throws Exception { 462984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.CREATE_ARG); 463984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 464984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 465984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxySystemProperty() throws Exception { 466984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.PROXY_SYSTEM_PROPERTY); 467984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 468984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 469984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingHttpProxySystemProperty() throws Exception { 470984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 471984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 472984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 473984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaProxy(ProxyConfig proxyConfig) throws Exception { 474c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse mockResponse = new MockResponse().setBody("this response comes via a proxy"); 47551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(mockResponse); 47651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 47760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 478984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson URL url = new URL("http://android.com/foo"); 479984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpURLConnection connection = proxyConfig.connect(server, url); 480c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a proxy", connection); 48160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 48251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest request = server.takeRequest(); 48360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET http://android.com/foo HTTP/1.1", request.getRequestLine()); 48460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(request.getHeaders(), "Host: android.com"); 48560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 48660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 487c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithContentLengthHeader() throws IOException { 488c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(new MockResponse() 489c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .setBody("abc\r\nYOU SHOULD NOT SEE THIS") 490c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .clearHeaders() 491c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .addHeader("Content-Length: 3")); 492c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 493c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 494c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 495c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 496c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 497c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithChunkedHeader() throws IOException { 498c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse mockResponse = new MockResponse(); 499c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setChunkedBody("abc", 3); 500c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 501c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write(mockResponse.getBody()); 502c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write("\r\nYOU SHOULD NOT SEE THIS".getBytes()); 503c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setBody(bytesOut.toByteArray()); 504c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.clearHeaders(); 505c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.addHeader("Transfer-encoding: chunked"); 506c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 507c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(mockResponse); 508c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 509c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 510c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 511c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 512c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 513984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxyArg() throws Exception { 514984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.CREATE_ARG); 515984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 516984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 517984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 518984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We weren't honoring all of the appropriate proxy system properties when 519984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * connecting via HTTPS. http://b/3097518 520984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 521984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxySystemProperty() throws Exception { 522984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.PROXY_SYSTEM_PROPERTY); 523984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 524984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 525984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingHttpProxySystemProperty() throws Exception { 526984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 527984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 528984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 529984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingHttpsProxySystemProperty() throws Exception { 530984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.HTTPS_PROXY_SYSTEM_PROPERTY); 531984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 532984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 533984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 534984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We were verifying the wrong hostname when connecting to an HTTPS site 535984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * through a proxy. http://b/3097277 536984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 537984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaHttpProxyToHttps(ProxyConfig proxyConfig) throws Exception { 53860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 539984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 54060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 541059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 542c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 543c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via a secure proxy")); 54451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 54560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 54660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URL url = new URL("https://android.com/foo"); 547984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) proxyConfig.connect(server, url); 548059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 549984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 55060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 551c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a secure proxy", connection); 55260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 55351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest connect = server.takeRequest(); 55460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("Connect line failure on proxy", 55560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson "CONNECT android.com:443 HTTP/1.1", connect.getRequestLine()); 55660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 55760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 55851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest get = server.takeRequest(); 55960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", get.getRequestLine()); 56060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(get.getHeaders(), "Host: android.com"); 561984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 56260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 56360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 564d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson /** 565d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson * Test which headers are sent unencrypted to the HTTP proxy. 566d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson */ 567d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testProxyConnectIncludesProxyHeadersOnly() 568d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson throws IOException, InterruptedException { 569984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 570d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 571d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 572d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 573d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 574d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("encrypted response from the origin server")); 575d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 576d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 577d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson URL url = new URL("https://android.com/foo"); 578d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) url.openConnection( 579d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.toProxyAddress()); 580d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Private", "Secret"); 581d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Proxy-Authorization", "bar"); 582d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("User-Agent", "baz"); 583d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 584984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 585d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContent("encrypted response from the origin server", connection); 586d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 587d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest connect = server.takeRequest(); 588d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContainsNoneMatching(connect.getHeaders(), "Private.*"); 589d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Authorization: bar"); 590d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "User-Agent: baz"); 591d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 592d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Connection: Keep-Alive"); 593d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 594d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest get = server.takeRequest(); 595d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(get.getHeaders(), "Private: Secret"); 596984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 597d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 598d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 599d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testDisconnectedConnection() throws IOException { 600d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("ABCDEFGHIJKLMNOPQR")); 601d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 602d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 603d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 604d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson InputStream in = connection.getInputStream(); 605d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertEquals('A', (char) in.read()); 606d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.disconnect(); 607d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson try { 608d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson in.read(); 609d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson fail("Expected a connection closed exception"); 610d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } catch (IOException expected) { 611d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 612d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 613d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 61451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithFixedLength() throws IOException { 61551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.FIXED_LENGTH); 61651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 61751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 61851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithChunkedEncoding() throws IOException { 61951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.CHUNKED); 62051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 62151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 62251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithNoLengthHeaders() throws IOException { 62351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.END_OF_STREAM); 62451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 62551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 62651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 62751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * HttpURLConnection.getInputStream().skip(long) causes ResponseCache corruption 62851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://code.google.com/p/android/issues/detail?id=8175 62951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 63051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testResponseCaching(TransferKind transferKind) throws IOException { 63151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 63251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "I love puppies but hate spiders", 1); 63351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 63451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 63551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 63651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 63751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 63851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 63951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // Make sure that calling skip() doesn't omit bytes from the cache. 64051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 64151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = urlConnection.getInputStream(); 64251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love ", readAscii(in, "I love ".length())); 64351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reliableSkip(in, "puppies but hate ".length()); 64451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("spiders", readAscii(in, "spiders".length())); 64551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 64651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 64751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 648096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 64951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 65051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson urlConnection = server.getUrl("/").openConnection(); // this response is cached! 65151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in = urlConnection.getInputStream(); 65251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love puppies but hate spiders", 65351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(in, "I love puppies but hate spiders".length())); 65451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 65551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getMissCount()); 65651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 657096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(1, cache.getSuccessCount()); 658096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 65951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 66051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 66183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseCacheRequestHeaders() throws IOException, URISyntaxException { 66283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 66383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 66483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 66583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson final AtomicReference<Map<String, List<String>>> requestHeadersRef 66683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson = new AtomicReference<Map<String, List<String>>>(); 66783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 66883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 66983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 67083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeadersRef.set(requestHeaders); 67183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 67283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 67383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection conn) throws IOException { 67483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 67583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 67683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson }); 67783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 67883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URL url = server.getUrl("/"); 67983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URLConnection urlConnection = url.openConnection(); 68083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("A", "android"); 68183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson readAscii(urlConnection.getInputStream(), Integer.MAX_VALUE); 68283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(Arrays.asList("android"), requestHeadersRef.get().get("A")); 68383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 68483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 68551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void reliableSkip(InputStream in, int length) throws IOException { 68651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson while (length > 0) { 68751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson length -= in.skip(length); 68851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 68951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 69051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 69151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 69251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Reads {@code count} characters from the stream. If the stream is 69351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * exhausted before {@code count} characters can be read, the remaining 69451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * characters are returned and the stream is closed. 69551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 69651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private String readAscii(InputStream in, int count) throws IOException { 69751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson StringBuilder result = new StringBuilder(); 69851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (int i = 0; i < count; i++) { 69951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson int value = in.read(); 70051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (value == -1) { 70151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 70251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 70351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 70451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson result.append((char) value); 70551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 70651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return result.toString(); 70751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 70851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 70951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithContentLengthHeader() throws IOException { 71051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.FIXED_LENGTH); 71151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 71251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 71351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithChunkedEncoding() throws IOException { 71451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.CHUNKED); 71551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 71651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 71751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithNoLengthHeaders() throws IOException { 71851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /* 71951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Intentionally empty. This case doesn't make sense because there's no 72051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * such thing as a premature disconnect when the disconnect itself 72151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * indicates the end of the data stream. 72251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 72351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 72451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 72551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testServerPrematureDisconnect(TransferKind transferKind) throws IOException { 72651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 72751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 16); 72851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(truncateViolently(response, 16)); 72951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 73051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 73151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 73251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 73351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 73451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 73551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson BufferedReader reader = new BufferedReader(new InputStreamReader( 73651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.getUrl("/").openConnection().getInputStream())); 73751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", reader.readLine()); 73851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 73951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reader.readLine(); 74051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("This implementation silently ignored a truncated HTTP body."); 74151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 74251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 74351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 74451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 74551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 74651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 74751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 74851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 74951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 75051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 75151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithContentLengthHeader() throws IOException { 75251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.FIXED_LENGTH); 75351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 75451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 75551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithChunkedEncoding() throws IOException { 75651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.CHUNKED); 75751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 75851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 75951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithNoLengthHeaders() throws IOException { 76051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.END_OF_STREAM); 76151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 76251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 76351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testClientPrematureDisconnect(TransferKind transferKind) throws IOException { 76451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 76551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 1024); 76651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 76751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 76851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 76951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 77051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 77151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 77251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 77351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 77451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 77551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 77651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 77751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.read(); 77851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("Expected an IOException because the stream is closed."); 77951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 78051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 78151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 78251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 78351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 78451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 78551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 78651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 78751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 78851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 78951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 79051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Shortens the body of {@code response} but not the corresponding headers. 79151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Only useful to test how clients respond to the premature conclusion of 79251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the HTTP body. 79351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 79451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) { 79551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 79651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson List<String> headers = new ArrayList<String>(response.getHeaders()); 79751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(Arrays.copyOfRange(response.getBody(), 0, numBytesToKeep)); 79851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().clear(); 79951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().addAll(headers); 80051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return response; 80151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 80251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 80351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithContentLengthHeader() throws IOException { 80451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.FIXED_LENGTH); 80551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 80651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 80751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithChunkedEncoding() throws IOException { 80851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.CHUNKED); 80951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 81051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 81151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithNoLengthHeaders() throws IOException { 81251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.END_OF_STREAM); 81351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 81451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 81551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndReset(TransferKind transferKind) throws IOException { 81651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 81751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDEFGHIJKLMNOPQRSTUVWXYZ", 1024); 81851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 81951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 82051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 82151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 82251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 82351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 82451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 82551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertFalse("This implementation claims to support mark().", in.markSupported()); 82651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.mark(5); 82751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 82851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 82951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.reset(); 83051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail(); 83151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 83251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 83351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("FGHIJKLMNOPQRSTUVWXYZ", readAscii(in, Integer.MAX_VALUE)); 83451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 83551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQRSTUVWXYZ", server.getUrl("/").openConnection()); 83651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 83751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 83851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 83951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 84051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 84151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * We've had a bug where we forget the HTTP response when we see response 84251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * code 401. This causes a new HTTP request to be issued for every call into 84351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the URLConnection. 84451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 84551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testUnauthorizedResponseHandling() throws IOException { 84651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse() 84751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .addHeader("WWW-Authenticate: challenge") 84851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setResponseCode(401) // UNAUTHORIZED 84951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("Unauthorized"); 85051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 85151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 85251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 85351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 85451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 85551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 85651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 85751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 85851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 85951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 86051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 86151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, server.getRequestCount()); 86251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 86351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 8646906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testNonHexChunkSize() throws IOException { 8656906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 8666906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5\r\nABCDE\r\nG\r\nFGHIJKLMNOPQRSTU\r\n0\r\n\r\n") 8676906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 8686906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked")); 8696906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 8706906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 8716906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 8726906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 8736906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 8746906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 8756906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 8766906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8776906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8786906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 8796906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testMissingChunkBody() throws IOException { 8806906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 8816906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5") 8826906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 8836906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked") 8846906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setDisconnectAtEnd(true)); 8856906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 8866906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 8876906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 8886906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 8896906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 8906906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 8916906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 8926906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8936906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 8946906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 89550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson /** 89650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * This test checks whether connections are gzipped by default. This 89750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * behavior in not required by the API, so a failure of this test does not 89850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * imply a bug in the implementation. 89950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson */ 90050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testGzipEncodingEnabledByDefault() throws IOException, InterruptedException { 90150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 90250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody(gzip("ABCABCABC".getBytes("UTF-8"))) 90350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: gzip")); 90450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 90550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 90650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 90750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCABCABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 9088116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson assertNull(connection.getContentEncoding()); 90950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 91050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 91150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 91250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 91350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 914deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testClientConfiguredGzipContentEncoding() throws Exception { 915deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(new MockResponse() 916deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .setBody(gzip("ABCDEFGHIJKLMNOPQRSTUVWXYZ".getBytes("UTF-8"))) 917deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .addHeader("Content-Encoding: gzip")); 918deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 919deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 920deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 921deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 922deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 923deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("ABCDEFGHIJKLMNOPQRSTUVWXYZ", readAscii(gunzippedIn, Integer.MAX_VALUE)); 924deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 925deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson RecordedRequest request = server.takeRequest(); 926deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 927deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 928deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 929deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithFixedLength() throws Exception { 930deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.FIXED_LENGTH); 931deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 932deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 933deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithChunkedEncoding() throws Exception { 934deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.CHUNKED); 935deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 936deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 93750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testClientConfiguredCustomContentEncoding() throws Exception { 93850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 93950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody("ABCDE") 94050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: custom")); 94150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 94250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 94350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 94450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson connection.addRequestProperty("Accept-Encoding", "custom"); 94550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCDE", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 94650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 94750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 94850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: custom"); 94950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 95050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 951deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 952deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Test a bug where gzip input streams weren't exhausting the input stream, 953deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * which corrupted the request that followed. 954deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * http://code.google.com/p/android/issues/detail?id=7059 955deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 956deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson private void testClientConfiguredGzipContentEncodingAndConnectionReuse( 957deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson TransferKind transferKind) throws Exception { 958deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseOne = new MockResponse(); 959deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson responseOne.addHeader("Content-Encoding: gzip"); 960deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseOne, gzip("one (gzipped)".getBytes("UTF-8")), 5); 961deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseOne); 962deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseTwo = new MockResponse(); 963deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseTwo, "two (identity)", 5); 964deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseTwo); 965deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 966deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 967deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 968deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 969deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 970deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("one (gzipped)", readAscii(gunzippedIn, Integer.MAX_VALUE)); 971deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 972deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 973deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection = server.getUrl("/").openConnection(); 974deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("two (identity)", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 975deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 976deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 977deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 978deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 979ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * Obnoxiously test that the chunk sizes transmitted exactly equal the 980ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * requested data+chunk header size. Although setChunkedStreamingMode() 981ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * isn't specific about whether the size applies to the data or the 982ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * complete chunk, the RI interprets it as a complete chunk. 983ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson */ 984ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testSetChunkedStreamingMode() throws IOException, InterruptedException { 985ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse()); 986ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 987ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 988ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 989ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setChunkedStreamingMode(8); 990ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setDoOutput(true); 991ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = urlConnection.getOutputStream(); 992ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write("ABCDEFGHIJKLMNOPQ".getBytes("US-ASCII")); 993ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 994ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 995ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 996ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("ABCDEFGHIJKLMNOPQ", new String(request.getBody(), "US-ASCII")); 997ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.asList(3, 3, 3, 3, 3, 2), request.getChunkSizes()); 998ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 999ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1000ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithFixedLengthStreaming() throws Exception { 1001ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.FIXED_LENGTH); 1002ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1003ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1004ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithChunkedStreaming() throws Exception { 1005ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.CHUNKED); 1006ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1007ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1008ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void testAuthenticateWithStreamingPost(StreamingMode streamingMode) throws Exception { 1009ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1010ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1011ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1012ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1013ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1014ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1015ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1016ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1017ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1018ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1019ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1020ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 1021ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 1022ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 1023ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setChunkedStreamingMode(0); 1024ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1025ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1026ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1027ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1028ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson try { 1029ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.getInputStream(); 1030ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail(); 1031ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } catch (HttpRetryException expected) { 1032ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1033ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1034ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the request... 1035ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1036ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1037ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1038ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1039ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1040ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson enum StreamingMode { 1041ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson FIXED_LENGTH, CHUNKED 1042ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1043ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1044ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithPost() throws Exception { 1045ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1046ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1047ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1048ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1049ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1050ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1051ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1052ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1053ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1054ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1055ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1056ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1057ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1058ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1059ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1060ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1061ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1062ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1063ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1064ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1065ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1066ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1067ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1068ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1069ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1070da289bcd0a9e207cc03c752f7c21c9004056e179Jesse Wilson // ...but the three requests that follow include an authorization header 1071ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1072ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1073ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 1074ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1075ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1076ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1077ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1078ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1079ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1080ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithGet() throws Exception { 1081ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1082ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1083ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1084ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1085ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1086ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1087ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1088ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1089ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1090ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1091ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1092ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1093ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1094ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1095ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1096ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1097ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1098ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1099ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1100ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1101ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...but the three requests that follow requests include an authorization header 1102ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1103ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1104ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("GET / HTTP/1.1", request.getRequestLine()); 1105ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1106ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1107ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1108ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1109ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1110c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithChunkedEncoding() throws Exception { 1111c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.CHUNKED, true); 1112c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1113c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1114c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithContentLengthHeader() throws Exception { 1115c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.FIXED_LENGTH, true); 1116c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1117c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1118c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithNoLengthHeaders() throws Exception { 1119c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.END_OF_STREAM, false); 1120c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1121c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1122c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private void testRedirected(TransferKind transferKind, boolean reuse) throws Exception { 1123c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse response = new MockResponse() 1124c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1125c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo"); 1126c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson transferKind.setBody(response, "This page has moved!", 10); 1127c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(response); 1128c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1129c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1130c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1131c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1132c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1133c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1134c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1135c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1136c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1137c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1138c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1139c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson if (reuse) { 1140c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1141c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1142c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1143c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1144c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedOnHttps() throws IOException, InterruptedException { 1145c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1146059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1147c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1148c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1149c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo") 1150c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1151c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1152c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1153c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1154c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1155059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1156c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1157c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1158c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1159c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1160c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1161c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1162c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1163c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1164c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1165c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1166c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpsToHttp() throws IOException, InterruptedException { 1167c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1168059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1169c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1170c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1171c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: http://anyhost/foo") 1172c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1173c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1174c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1175c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1176059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1177c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1178c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1179c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1180c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1181c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpToHttps() throws IOException, InterruptedException { 1182c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1183c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1184c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: https://anyhost/foo") 1185c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1186c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1187c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1188c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1189c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1190c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1191c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1192c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1193c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectToAnotherOriginServer() throws Exception { 1194c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockWebServer server2 = new MockWebServer(); 1195c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.enqueue(new MockResponse().setBody("This is the 2nd server!")); 1196c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.play(); 1197c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1198c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1199c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1200c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: " + server2.getUrl("/").toString()) 1201c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1202c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the first server again!")); 1203c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1204c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1205c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1206c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the 2nd server!", 1207c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1208c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(server2.getUrl("/"), connection.getURL()); 1209c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1210c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson // make sure the first server was careful to recycle the connection 1211c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the first server again!", 1212c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(server.getUrl("/").openStream(), Integer.MAX_VALUE)); 1213c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1214c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 121500feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertContains(first.getHeaders(), "Host: " + hostname + ":" + server.getPort()); 1216c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest second = server2.takeRequest(); 121700feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertContains(second.getHeaders(), "Host: " + hostname + ":" + server2.getPort()); 1218c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest third = server.takeRequest(); 1219c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, third.getSequenceNumber()); 1220c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1221c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.shutdown(); 1222c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1223c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1224c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testHttpsWithCustomTrustManager() throws Exception { 1225c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 1226c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingTrustManager trustManager = new RecordingTrustManager(); 1227c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLContext sc = SSLContext.getInstance("TLS"); 1228c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson sc.init(null, new TrustManager[] { trustManager }, new java.security.SecureRandom()); 1229c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1230c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HostnameVerifier defaultHostnameVerifier = HttpsURLConnection.getDefaultHostnameVerifier(); 1231c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(hostnameVerifier); 1232c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLSocketFactory defaultSSLSocketFactory = HttpsURLConnection.getDefaultSSLSocketFactory(); 1233c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory()); 1234c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson try { 1235c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1236059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1237c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1238c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 1239c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("GHI")); 1240c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1241c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1242c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URL url = server.getUrl("/"); 1243c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("ABC", readAscii(url.openStream(), Integer.MAX_VALUE)); 1244c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("DEF", readAscii(url.openStream(), Integer.MAX_VALUE)); 1245c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GHI", readAscii(url.openStream(), Integer.MAX_VALUE)); 1246c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 124700feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertEquals(Arrays.asList("verify " + hostname), hostnameVerifier.calls); 12484559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals(Arrays.asList("checkServerTrusted [" 124900feece22909b7dc79fc96d666d157390b93858eJesse Wilson + "CN=" + hostname + " 1, " 12504559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "CN=Test Intermediate Certificate Authority 1, " 12514559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "CN=Test Root Certificate Authority 1" 12524559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom + "] RSA"), 1253c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson trustManager.calls); 1254c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } finally { 1255c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(defaultHostnameVerifier); 1256c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(defaultSSLSocketFactory); 1257c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1258c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1259c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1260eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testConnectTimeouts() throws IOException { 1261eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson // 10.0.0.0 is non-routable and will time out on every network 1262eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = new URL("http://10.0.0.0/").openConnection(); 1263eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setConnectTimeout(1000); 1264eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1265eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.getInputStream(); 1266eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1267eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1268eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1269eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1270eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1271eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testReadTimeouts() throws IOException { 1272eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson /* 1273eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * This relies on the fact that MockWebServer doesn't close the 1274eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * connection after a response has been sent. This causes the client to 1275eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * try to read more bytes than are sent, which results in a timeout. 1276eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson */ 1277eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson MockResponse timeout = new MockResponse() 1278eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .setBody("ABC") 1279eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .clearHeaders() 1280eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .addHeader("Content-Length: 4"); 1281eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.enqueue(timeout); 1282eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.play(); 1283eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1284eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 1285eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setReadTimeout(1000); 1286eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson InputStream in = urlConnection.getInputStream(); 1287eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('A', in.read()); 1288eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('B', in.read()); 1289eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('C', in.read()); 1290eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1291eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson in.read(); // if Content-Length was accurate, this would return -1 immediately 1292eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1293eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1294eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1295eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1296eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1297125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson public void testSetChunkedEncodingAsRequestProperty() throws IOException, InterruptedException { 1298125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.enqueue(new MockResponse()); 1299125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.play(); 1300125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1301125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1302125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setRequestProperty("Transfer-encoding", "chunked"); 1303125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setDoOutput(true); 1304125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.getOutputStream().write("ABC".getBytes("UTF-8")); 1305125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1306125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1307125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson RecordedRequest request = server.takeRequest(); 1308125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals("ABC", new String(request.getBody(), "UTF-8")); 1309125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson } 1310125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1311f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInRequest() throws IOException, InterruptedException { 1312f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); // server doesn't honor the connection: close header! 1313f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1314f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1315f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1316f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1317f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson a.setRequestProperty("Connection", "close"); 1318f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1319f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1320f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1321f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1322f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1323f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1324f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1325f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1326f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1327f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1328f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInResponse() throws IOException, InterruptedException { 1329f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().addHeader("Connection: close")); 1330f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1331f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1332f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1333f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1334f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1335f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1336f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1337f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1338f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1339f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1340f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1341f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1342f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1343f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1344f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseWithRedirect() throws IOException, InterruptedException { 1345f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson MockResponse response = new MockResponse() 1346f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1347f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Location: /foo") 1348f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Connection: close"); 1349f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(response); 1350f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1351f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1352f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1353f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1354f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("This is the new location!", 1355f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1356f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1357f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1358f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1359f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1360f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1361f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 136265d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson public void testResponseCodeDisagreesWithHeaders() throws IOException, InterruptedException { 136365d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.enqueue(new MockResponse() 136465d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setResponseCode(HttpURLConnection.HTTP_NO_CONTENT) 136565d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setBody("This body is not allowed!")); 136665d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.play(); 136765d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 136865d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 136965d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson assertEquals("This body is not allowed!", 137065d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 137165d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson } 137265d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 1373ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson public void testSingleByteReadIsSigned() throws IOException { 1374ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.enqueue(new MockResponse().setBody(new byte[] { -2, -1 })); 1375ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.play(); 1376ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1377ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1378ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson InputStream in = connection.getInputStream(); 1379ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(254, in.read()); 1380ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(255, in.read()); 1381ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(-1, in.read()); 1382ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson } 1383ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1384f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithChunkedEncoding() throws IOException { 1385f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.CHUNKED); 1386f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1387f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1388f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithFixedLength() throws IOException { 1389f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.FIXED_LENGTH); 1390f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1391f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1392f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithNoLengthHeaders() throws IOException { 1393f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.END_OF_STREAM); 1394f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1395f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1396f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson /** 1397f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * We explicitly permit apps to close the upload stream even after it has 1398f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * been transmitted. We also permit flush so that buffered streams can 1399f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * do a no-op flush when they are closed. http://b/3038470 1400f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson */ 1401f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson private void testFlushAfterStreamTransmitted(TransferKind transferKind) throws IOException { 1402f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.enqueue(new MockResponse().setBody("abc")); 1403f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.play(); 1404f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1405f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1406f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setDoOutput(true); 1407f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson byte[] upload = "def".getBytes("UTF-8"); 1408f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1409f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson if (transferKind == TransferKind.CHUNKED) { 1410f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setChunkedStreamingMode(0); 1411f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } else if (transferKind == TransferKind.FIXED_LENGTH) { 1412f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setFixedLengthStreamingMode(upload.length); 1413f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1414f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1415f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson OutputStream out = connection.getOutputStream(); 1416f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write(upload); 1417f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson assertEquals("abc", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1418f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1419f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.flush(); // dubious but permitted 1420f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson try { 1421f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write("ghi".getBytes("UTF-8")); 1422f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson fail(); 1423f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } catch (IOException expected) { 1424f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1425f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1426f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1427ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson /** 1428deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Encodes the response body using GZIP and adds the corresponding header. 1429deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1430deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public byte[] gzip(byte[] bytes) throws IOException { 1431deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1432deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson OutputStream gzippedOut = new GZIPOutputStream(bytesOut); 1433deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.write(bytes); 1434deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.close(); 1435deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson return bytesOut.toByteArray(); 1436deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1437deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1438c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson /** 1439c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * Reads at most {@code limit} characters from {@code in} and asserts that 1440c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * content equals {@code expected}. 1441c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson */ 1442c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection, int limit) 1443c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson throws IOException { 144451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(expected, readAscii(connection.getInputStream(), limit)); 1445c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ((HttpURLConnection) connection).disconnect(); 1446c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1447c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 1448c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection) throws IOException { 1449c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent(expected, connection, Integer.MAX_VALUE); 1450c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1451c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 145260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson private void assertContains(List<String> headers, String header) { 145360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertTrue(headers.toString(), headers.contains(header)); 145460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 145551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1456ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void assertContainsNoneMatching(List<String> headers, String pattern) { 1457ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (String header : headers) { 1458ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (header.matches(pattern)) { 1459ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail("Header " + header + " matches " + pattern); 1460ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1461ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1462ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1463ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1464eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson private Set<String> newSet(String... elements) { 146583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return new HashSet<String>(Arrays.asList(elements)); 146683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 146783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 146851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson enum TransferKind { 146951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson CHUNKED() { 1470deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) 147151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException { 147251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setChunkedBody(content, chunkSize); 147351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 147451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 147551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson FIXED_LENGTH() { 1476deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 147751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 147851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 147951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 148051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson END_OF_STREAM() { 1481deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 148251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 148351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setDisconnectAtEnd(true); 148451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (Iterator<String> h = response.getHeaders().iterator(); h.hasNext(); ) { 148551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (h.next().startsWith("Content-Length:")) { 148651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson h.remove(); 148751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 148851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 148951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 149051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 149151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }; 149251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1493deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson abstract void setBody(MockResponse response, byte[] content, int chunkSize) 149451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException; 1495deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1496deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson void setBody(MockResponse response, String content, int chunkSize) throws IOException { 1497deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson setBody(response, content.getBytes("UTF-8"), chunkSize); 1498deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 149951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 1500c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1501984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson enum ProxyConfig { 1502984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson CREATE_ARG() { 1503984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1504984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1505984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(server.toProxyAddress()); 1506984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1507984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1508984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1509984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson PROXY_SYSTEM_PROPERTY() { 1510984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1511984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1512984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyHost", "localhost"); 1513984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyPort", Integer.toString(server.getPort())); 1514984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1515984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1516984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1517984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1518984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTP_PROXY_SYSTEM_PROPERTY() { 1519984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1520984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1521984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyHost", "localhost"); 1522984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyPort", Integer.toString(server.getPort())); 1523984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1524984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1525984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1526984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1527984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTPS_PROXY_SYSTEM_PROPERTY() { 1528984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1529984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1530984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyHost", "localhost"); 1531984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyPort", Integer.toString(server.getPort())); 1532984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1533984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1534984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }; 1535984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1536984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public abstract HttpURLConnection connect(MockWebServer server, URL url) throws IOException; 1537984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1538984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1539c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingTrustManager implements X509TrustManager { 1540c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1541c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1542c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public X509Certificate[] getAcceptedIssuers() { 1543c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("getAcceptedIssuers"); 1544c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return new X509Certificate[] {}; 1545c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1546c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1547c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkClientTrusted(X509Certificate[] chain, String authType) 1548c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1549c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkClientTrusted " + certificatesToString(chain) + " " + authType); 1550c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1551c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1552c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkServerTrusted(X509Certificate[] chain, String authType) 1553c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1554c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkServerTrusted " + certificatesToString(chain) + " " + authType); 1555c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1556c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1557c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private String certificatesToString(X509Certificate[] certificates) { 1558c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson List<String> result = new ArrayList<String>(); 1559c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson for (X509Certificate certificate : certificates) { 1560c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson result.add(certificate.getSubjectDN() + " " + certificate.getSerialNumber()); 1561c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1562c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return result.toString(); 1563c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1564c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1565c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1566c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingHostnameVerifier implements HostnameVerifier { 1567c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1568c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1569c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public boolean verify(String hostname, SSLSession session) { 1570c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("verify " + hostname); 1571c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return true; 1572c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1573c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1574e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes} 1575