URLConnectionTest.java revision 0c2fd828abec671333b8b88281825fd27a783723
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; 28b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilsonimport java.net.ConnectException; 294557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpRetryException; 304557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.HttpURLConnection; 314557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.PasswordAuthentication; 32f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilsonimport java.net.ProtocolException; 33f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstromimport java.net.Proxy; 344557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.ResponseCache; 3537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilsonimport java.net.SecureCacheResponse; 364557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.SocketTimeoutException; 374557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URI; 384557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URISyntaxException; 394557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URL; 404557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.URLConnection; 4137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilsonimport java.security.Principal; 4237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilsonimport java.security.cert.Certificate; 43c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.CertificateException; 44c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport java.security.cert.X509Certificate; 4551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.ArrayList; 4602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughesimport java.util.Arrays; 4751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Collections; 4883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.HashSet; 4951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Iterator; 506247987eb505a482a67f5f19678260d9e7240a5fElliott Hughesimport java.util.List; 5183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.Map; 5251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport java.util.Set; 53f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilsonimport java.util.concurrent.atomic.AtomicInteger; 54afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilsonimport java.util.concurrent.atomic.AtomicBoolean; 5583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.concurrent.atomic.AtomicReference; 56deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPInputStream; 57deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPOutputStream; 5860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HostnameVerifier; 5960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HttpsURLConnection; 60c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLContext; 61096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilsonimport javax.net.ssl.SSLException; 622915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstromimport javax.net.ssl.SSLHandshakeException; 6360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.SSLSession; 64c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLSocketFactory; 65c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.TrustManager; 66c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.X509TrustManager; 672915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstromimport libcore.java.security.TestKeyStore; 6850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilsonimport libcore.javax.net.ssl.TestSSLContext; 6951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport tests.http.DefaultResponseCache; 7060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockResponse; 7160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockWebServer; 7260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.RecordedRequest; 73e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport tests.http.SocketPolicy; 74e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.DISCONNECT_AT_END; 75e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.DISCONNECT_AT_START; 76e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.SHUTDOWN_INPUT_AT_END; 77e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.SHUTDOWN_OUTPUT_AT_END; 785fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughesimport tests.net.StuckServer; 79e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 80e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughespublic class URLConnectionTest extends junit.framework.TestCase { 81b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 82ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private static final Authenticator SIMPLE_AUTHENTICATOR = new Authenticator() { 83ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson protected PasswordAuthentication getPasswordAuthentication() { 84ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson return new PasswordAuthentication("username", "password".toCharArray()); 85ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 86ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson }; 87ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 8851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockWebServer server = new MockWebServer(); 890c2fd828abec671333b8b88281825fd27a783723Jesse Wilson private String hostName; 9000feece22909b7dc79fc96d666d157390b93858eJesse Wilson 9100feece22909b7dc79fc96d666d157390b93858eJesse Wilson @Override protected void setUp() throws Exception { 9200feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.setUp(); 930c2fd828abec671333b8b88281825fd27a783723Jesse Wilson hostName = server.getHostName(); 9400feece22909b7dc79fc96d666d157390b93858eJesse Wilson } 9551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 9651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson @Override protected void tearDown() throws Exception { 9751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(null); 98ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(null); 99984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyHost"); 100984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyPort"); 101984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyHost"); 102984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyPort"); 103984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyHost"); 104984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyPort"); 10551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); 10600feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.tearDown(); 10751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 10851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 10983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testRequestHeaders() throws IOException, InterruptedException { 11083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse()); 11183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 11283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 11383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 11483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "e"); 11583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "f"); 11683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders = urlConnection.getRequestProperties(); 11783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("e", "f"), new HashSet<String>(requestHeaders.get("D"))); 11883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.put("G", Arrays.asList("h")); 12083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 12183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 12283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 12483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.get("D").add("i"); 12583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 12683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 12783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 12983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty(null, "j"); 13083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 13183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 13283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 13483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty(null, "k"); 13583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 13683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 13783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("NullValue", null); // should fail silently! 13983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("AnotherNullValue", null); // should fail silently! 14083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 14183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.getResponseCode(); 14283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson RecordedRequest request = server.takeRequest(); 14383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: e"); 14483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: f"); 14583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "NullValue.*"); 14683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "AnotherNullValue.*"); 14783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "G:.*"); 14883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "null:.*"); 14983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 15083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 15183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("N", "o"); 15283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 15383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 15483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 15583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 15683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("P", "q"); 15783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 15883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 15983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 16083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 16183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 16283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseHeaders() throws IOException, InterruptedException { 16383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse() 16483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setStatus("HTTP/1.0 200 Fantastic") 16583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: b") 16683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: c") 16783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8)); 16883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 16983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 17083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 17183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 17283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 173c1a675c80c69decadb736b245f0366f93a94a462Jesse Wilson assertEquals("HTTP/1.0 200 Fantastic", urlConnection.getHeaderField(null)); 17483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> responseHeaders = urlConnection.getHeaderFields(); 1758ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(Arrays.asList("HTTP/1.0 200 Fantastic"), responseHeaders.get(null)); 17683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("b", "c"), new HashSet<String>(responseHeaders.get("A"))); 17783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 17883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.put("N", Arrays.asList("o")); 17983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 18083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 18183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 18283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 18383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.get("A").add("d"); 18483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 18583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 18683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 18783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 18883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 189e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // Check that if we don't read to the end of a response, the next request on the 190e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // recycled connection doesn't get the unread tail of the first request's response. 191e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // http://code.google.com/p/android/issues/detail?id=2939 192e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes public void test_2939() throws Exception { 193b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8); 194b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 195b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 196b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 197b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 198b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 199c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 200c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 2018baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson } 2028baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson 203977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // Check that we recognize a few basic mime types by extension. 204977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // http://code.google.com/p/android/issues/detail?id=10100 205977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes public void test_10100() throws Exception { 206977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("image/jpeg", URLConnection.guessContentTypeFromName("someFile.jpg")); 207977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("application/pdf", URLConnection.guessContentTypeFromName("stuff.pdf")); 208977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes } 209977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes 2108baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson public void testConnectionsArePooled() throws Exception { 211b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setBody("ABCDEFGHIJKLMNOPQR"); 212b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 213b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 214b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 215b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 216b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 217b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 21806e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/foo").openConnection()); 219c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 22006e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/bar?baz=quux").openConnection()); 221c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 22206e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/z").openConnection()); 223c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 224c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 225c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 226c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testChunkedConnectionsArePooled() throws Exception { 227c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDEFGHIJKLMNOPQR", 5); 228c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 229c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 230c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 231c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 232c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 233c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 23406e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/foo").openConnection()); 235b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 23606e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/bar?baz=quux").openConnection()); 237b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 23806e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/z").openConnection()); 239b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 240e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes } 24102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 242e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson public void testServerClosesSocket() throws Exception { 243e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson testServerClosesOutput(DISCONNECT_AT_END); 244e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 245e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 246e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson public void testServerShutdownInput() throws Exception { 247e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson testServerClosesOutput(SHUTDOWN_INPUT_AT_END); 248e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 249e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 250e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson public void testServerShutdownOutput() throws Exception { 251e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson testServerClosesOutput(SHUTDOWN_OUTPUT_AT_END); 252e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 253e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 254e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson private void testServerClosesOutput(SocketPolicy socketPolicy) throws Exception { 255e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse() 256e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setBody("This connection won't pool properly") 257e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setSocketPolicy(socketPolicy)); 258e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse() 259e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setBody("This comes after a busted connection")); 260e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.play(); 261e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 262e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertContent("This connection won't pool properly", server.getUrl("/a").openConnection()); 263e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 264e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertContent("This comes after a busted connection", server.getUrl("/b").openConnection()); 265e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson // sequence number 0 means the HTTP socket connection was not reused 266e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 267e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 268e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 269b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson enum WriteKind { BYTE_BY_BYTE, SMALL_BUFFERS, LARGE_BUFFERS } 27002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 27102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_byteByByte() throws Exception { 27251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.BYTE_BY_BYTE); 27302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 27502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_smallBuffers() throws Exception { 27651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.SMALL_BUFFERS); 27702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 27902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_largeBuffers() throws Exception { 28051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.LARGE_BUFFERS); 28102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 28202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 28302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_byteByByte() throws Exception { 28451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.BYTE_BY_BYTE); 28502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 28602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 28702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_smallBuffers() throws Exception { 28851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.SMALL_BUFFERS); 28902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 29002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 29102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_largeBuffers() throws Exception { 29251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.LARGE_BUFFERS); 29302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 29402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 29551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void doUpload(TransferKind uploadKind, WriteKind writeKind) throws Exception { 29602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes int n = 512*1024; 297b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.setBodyLimit(0); 298b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse()); 299b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 300b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 301b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson HttpURLConnection conn = (HttpURLConnection) server.getUrl("/").openConnection(); 30202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setDoOutput(true); 30302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setRequestMethod("POST"); 30451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 30502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setChunkedStreamingMode(-1); 30602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 30702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setFixedLengthStreamingMode(n); 30802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 30902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes OutputStream out = conn.getOutputStream(); 31002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes if (writeKind == WriteKind.BYTE_BY_BYTE) { 31102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; ++i) { 31202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write('x'); 31302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 31402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 31502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes byte[] buf = new byte[writeKind == WriteKind.SMALL_BUFFERS ? 256 : 64*1024]; 31602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes Arrays.fill(buf, (byte) 'x'); 31702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; i += buf.length) { 31802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write(buf, 0, Math.min(buf.length, n - i)); 31902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 32002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 32102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.close(); 3224cb7f05dc68abb23ae54a5891c369062185f2210Elliott Hughes assertEquals(200, conn.getResponseCode()); 323b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson RecordedRequest request = server.takeRequest(); 324b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(n, request.getBodySize()); 32551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 326b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().size() > 0); 327b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } else { 328b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().isEmpty()); 329b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } 33002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 3316247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 33251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 33351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Test that response caching is consistent with the RI and the spec. 33451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.4 33551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 3366247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching() throws Exception { 3376247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // Test each documented HTTP/1.1 code, plus the first unused value in each range. 3386247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html 3396247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 3406247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // We can't test 100 because it's not really a response. 3416247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // assertCached(false, 100); 3426247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 101); 3436247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 102); 3446247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 200); 3456247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 201); 3466247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 202); 3476247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 203); 3486247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 204); 3496247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 205); 3506247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 206); 3516247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 207); 35251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_300.) 3536247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 301); 3546247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 302; i <= 308; ++i) { 3556247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3566247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3576247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 400; i <= 406; ++i) { 3586247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3596247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3606247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // (See test_responseCaching_407.) 3616247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 408); 3626247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 409); 36351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_410.) 3646247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 411; i <= 418; ++i) { 3656247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3666247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3676247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 500; i <= 506; ++i) { 3686247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3696247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3706247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3716247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 37251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_300() throws Exception { 37351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: fix this for android 37451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 300); 37551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 37651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3771f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson /** 3781f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * Response code 407 should only come from proxy servers. Android's client 3791f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * throws if it is sent by an origin server. 3801f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson */ 3811f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson public void testOriginServerSends407() throws Exception { 3821f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.enqueue(new MockResponse().setResponseCode(407)); 3831f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.play(); 3841f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson 3851f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson URL url = server.getUrl("/"); 3861f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3871f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson try { 3881f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson conn.getResponseCode(); 3891f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson fail(); 3901f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } catch (IOException expected) { 3911f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } 3926247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3936247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 39451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_410() throws Exception { 39551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // the HTTP spec permits caching 410s, but the RI doesn't. 39651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 410); 39751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 39851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3996247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes private void assertCached(boolean shouldPut, int responseCode) throws Exception { 40051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server = new MockWebServer(); 401b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse() 402b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .setResponseCode(responseCode) 40351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("ABCDE") 404b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .addHeader("WWW-Authenticate: challenge")); 405b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 406b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 40751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache responseCache = new DefaultResponseCache(); 40851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(responseCache); 40951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 41051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 4116247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertEquals(responseCode, conn.getResponseCode()); 4126247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 41351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // exhaust the content stream 41451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 41551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: remove special case once testUnauthorizedResponseHandling() is fixed 41651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (responseCode != 401) { 41751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(conn.getInputStream(), Integer.MAX_VALUE); 41851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 41938db242d16a5a31fd86c519b817ae94c5fc89417Elliott Hughes } catch (IOException expected) { 42051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 42151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 42251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson Set<URI> expectedCachedUris = shouldPut 42351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ? Collections.singleton(url.toURI()) 42451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson : Collections.<URI>emptySet(); 42551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(Integer.toString(responseCode), 42651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson expectedCachedUris, responseCache.getContents().keySet()); 42751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); // tearDown() isn't sufficient; this test starts multiple servers 4286247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 42960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 430f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson /** 431f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson * Test that we can interrogate the response when the cache is being 432f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson * populated. http://code.google.com/p/android/issues/detail?id=7787 433f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson */ 434f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson public void testResponseCacheCallbackApis() throws Exception { 435f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson final String body = "ABCDE"; 436f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson final AtomicInteger cacheCount = new AtomicInteger(); 437f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson 438f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson server.enqueue(new MockResponse() 439f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson .setStatus("HTTP/1.1 200 Fantastic") 440f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson .addHeader("fgh: ijk") 441f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson .setBody(body)); 442f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson server.play(); 443f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson 444f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 445f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 446f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 447f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson return null; 448f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } 449f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection conn) throws IOException { 450f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson HttpURLConnection httpConnection = (HttpURLConnection) conn; 451f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals("HTTP/1.1 200 Fantastic", httpConnection.getHeaderField(null)); 452fa9f91a486b81b19db13eab59e704179381de22fJesse Wilson assertEquals(Arrays.asList("HTTP/1.1 200 Fantastic"), 453f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson httpConnection.getHeaderFields().get(null)); 454f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals(200, httpConnection.getResponseCode()); 455f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals("Fantastic", httpConnection.getResponseMessage()); 456f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals(body.length(), httpConnection.getContentLength()); 457f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals("ijk", httpConnection.getHeaderField("fgh")); 458f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson try { 459f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson httpConnection.getInputStream(); // the RI doesn't forbid this, but it should 460f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson fail(); 461f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } catch (IOException expected) { 462f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } 463f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson cacheCount.incrementAndGet(); 464f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson return null; 465f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } 466f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson }); 467f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson 468f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson URL url = server.getUrl("/"); 469f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson HttpURLConnection connection = (HttpURLConnection) url.openConnection(); 470f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals(body, readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 471f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals(1, cacheCount.get()); 472f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } 473f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson 474f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson public void testGetResponseCodeNoResponseBody() throws Exception { 475f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson server.enqueue(new MockResponse() 476f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson .addHeader("abc: def")); 477f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson server.play(); 478f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson 479f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson URL url = server.getUrl("/"); 480f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 481f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson conn.setDoInput(false); 482f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals("def", conn.getHeaderField("abc")); 483f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson assertEquals(200, conn.getResponseCode()); 484f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson try { 485f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson conn.getInputStream(); 486f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson fail(); 487f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } catch (ProtocolException expected) { 488f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } 489f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson } 490f14bee4dbf9f724a177a85a7b73f7b0bc6df1ec0Jesse Wilson 49160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttps() throws IOException, InterruptedException { 49260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 49360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 494059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 495c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 49660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson server.play(); 49760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 498096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 4994559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 50060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 501c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via HTTPS", connection); 50260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 50360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson RecordedRequest request = server.takeRequest(); 50460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 50560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 50660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 507096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson public void testConnectViaHttpsReusingConnections() throws IOException, InterruptedException { 508096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 509096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 510059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 511096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 512096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 513096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.play(); 514096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 515b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 516b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 517b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("this response comes via HTTPS", connection); 518b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 519b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 520b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 521b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("another response via HTTPS", connection); 522b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 523b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 524b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 525b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson } 526b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 5278116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson public void testConnectViaHttpsReusingConnectionsDifferentFactories() 528b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson throws IOException, InterruptedException { 529b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 530b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 531b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 532b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 533b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 534b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.play(); 535b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 536096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson // install a custom SSL socket factory so the server can be authorized 537096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 538059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 539096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertContent("this response comes via HTTPS", connection); 540096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 541096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 542096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson try { 543096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 544b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson fail("without an SSL socket factory, the connection should fail"); 545096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } catch (SSLException expected) { 546096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 547096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 548096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 5494559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom public void testConnectViaHttpsWithSSLFallback() throws IOException, InterruptedException { 5504559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 5514559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 5524559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 553e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse().setSocketPolicy(DISCONNECT_AT_START)); 5544559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via SSL")); 5554559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.play(); 5564559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 5574559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 5584559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 5594559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 5604559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertContent("this response comes via SSL", connection); 5614559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 5624559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom RecordedRequest request = server.takeRequest(); 5634559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 5644559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom } 5654559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 5662915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom /** 5672915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom * Verify that we don't retry connections on certificate verification errors. 5682915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom * 5692915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom * http://code.google.com/p/android/issues/detail?id=13178 5702915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom */ 5712915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom public void testConnectViaHttpsToUntrustedServer() throws IOException, InterruptedException { 5722915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(TestKeyStore.getClientCA2(), 5732915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom TestKeyStore.getServer()); 5742915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom 5752915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 5762915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom server.enqueue(new MockResponse()); // unused 5772915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom server.play(); 5782915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom 5792915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 5802915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 5812915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom try { 5822915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom connection.getInputStream(); 5832915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom fail(); 5842915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom } catch (SSLHandshakeException expected) { 5852915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom assertTrue(expected.getCause() instanceof CertificateException); 5862915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom } 5872915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom assertEquals(0, server.getRequestCount()); 5882915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom } 5892915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom 590984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxyArg() throws Exception { 591984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.CREATE_ARG); 592984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 593984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 594984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxySystemProperty() throws Exception { 595984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.PROXY_SYSTEM_PROPERTY); 596984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 597984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 598984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingHttpProxySystemProperty() throws Exception { 599984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 600984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 601984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 602984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaProxy(ProxyConfig proxyConfig) throws Exception { 603c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse mockResponse = new MockResponse().setBody("this response comes via a proxy"); 60451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(mockResponse); 60551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 60660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 607984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson URL url = new URL("http://android.com/foo"); 608984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpURLConnection connection = proxyConfig.connect(server, url); 609c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a proxy", connection); 61060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 61151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest request = server.takeRequest(); 61260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET http://android.com/foo HTTP/1.1", request.getRequestLine()); 61360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(request.getHeaders(), "Host: android.com"); 61460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 61560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 616c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithContentLengthHeader() throws IOException { 617c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(new MockResponse() 618c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .setBody("abc\r\nYOU SHOULD NOT SEE THIS") 619c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .clearHeaders() 620c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .addHeader("Content-Length: 3")); 621c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 622c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 623c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 624c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 625c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 626c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithChunkedHeader() throws IOException { 627c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse mockResponse = new MockResponse(); 628c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setChunkedBody("abc", 3); 629c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 630c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write(mockResponse.getBody()); 631c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write("\r\nYOU SHOULD NOT SEE THIS".getBytes()); 632c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setBody(bytesOut.toByteArray()); 633c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.clearHeaders(); 634c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.addHeader("Transfer-encoding: chunked"); 635c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 636c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(mockResponse); 637c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 638c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 639c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 640c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 641c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 642f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom public void testConnectViaHttpProxyToHttpsUsingProxyArgWithNoProxy() throws Exception { 643f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom testConnectViaDirectProxyToHttps(ProxyConfig.NO_PROXY); 644f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 645f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 646f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom public void testConnectViaHttpProxyToHttpsUsingHttpProxySystemProperty() throws Exception { 647f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom // https should not use http proxy 648f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom testConnectViaDirectProxyToHttps(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 649f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 650f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 651f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom private void testConnectViaDirectProxyToHttps(ProxyConfig proxyConfig) throws Exception { 652f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 653f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 654f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 655f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 656f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.play(); 657f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 658f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom URL url = server.getUrl("/foo"); 659f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) proxyConfig.connect(server, url); 660f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 661f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 662f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom assertContent("this response comes via HTTPS", connection); 663f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 664f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom RecordedRequest request = server.takeRequest(); 665f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 666f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 667f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 668f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 669984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxyArg() throws Exception { 670984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.CREATE_ARG); 671984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 672984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 673984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 674984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We weren't honoring all of the appropriate proxy system properties when 675984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * connecting via HTTPS. http://b/3097518 676984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 677984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxySystemProperty() throws Exception { 678984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.PROXY_SYSTEM_PROPERTY); 679984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 680984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 681984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingHttpsProxySystemProperty() throws Exception { 682984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.HTTPS_PROXY_SYSTEM_PROPERTY); 683984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 684984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 685984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 686984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We were verifying the wrong hostname when connecting to an HTTPS site 687984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * through a proxy. http://b/3097277 688984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 689984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaHttpProxyToHttps(ProxyConfig proxyConfig) throws Exception { 69060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 691984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 69260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 693059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 694c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 695c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via a secure proxy")); 69651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 69760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 69860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URL url = new URL("https://android.com/foo"); 699984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) proxyConfig.connect(server, url); 700059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 701984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 70260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 703c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a secure proxy", connection); 70460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 70551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest connect = server.takeRequest(); 70660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("Connect line failure on proxy", 70760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson "CONNECT android.com:443 HTTP/1.1", connect.getRequestLine()); 70860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 70960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 71051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest get = server.takeRequest(); 71160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", get.getRequestLine()); 71260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(get.getHeaders(), "Host: android.com"); 713984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 71460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 71560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 716d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson /** 717d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson * Test which headers are sent unencrypted to the HTTP proxy. 718d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson */ 719d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testProxyConnectIncludesProxyHeadersOnly() 720d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson throws IOException, InterruptedException { 721984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 722d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 723d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 724d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 725d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 726d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("encrypted response from the origin server")); 727d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 728d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 729d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson URL url = new URL("https://android.com/foo"); 730d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) url.openConnection( 731d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.toProxyAddress()); 732d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Private", "Secret"); 733d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Proxy-Authorization", "bar"); 734d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("User-Agent", "baz"); 735d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 736984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 737d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContent("encrypted response from the origin server", connection); 738d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 739d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest connect = server.takeRequest(); 740d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContainsNoneMatching(connect.getHeaders(), "Private.*"); 741d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Authorization: bar"); 742d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "User-Agent: baz"); 743d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 744d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Connection: Keep-Alive"); 745d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 746d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest get = server.takeRequest(); 747d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(get.getHeaders(), "Private: Secret"); 748984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 749d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 750d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 751d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testDisconnectedConnection() throws IOException { 752d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("ABCDEFGHIJKLMNOPQR")); 753d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 754d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 755d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 756d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson InputStream in = connection.getInputStream(); 757d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertEquals('A', (char) in.read()); 758d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.disconnect(); 759d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson try { 760d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson in.read(); 761d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson fail("Expected a connection closed exception"); 762d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } catch (IOException expected) { 763d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 764d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 765d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 76651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithFixedLength() throws IOException { 76751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.FIXED_LENGTH); 76851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 76951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 77051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithChunkedEncoding() throws IOException { 77151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.CHUNKED); 77251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 77351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 77451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithNoLengthHeaders() throws IOException { 77551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.END_OF_STREAM); 77651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 77751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 77851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 77951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * HttpURLConnection.getInputStream().skip(long) causes ResponseCache corruption 78051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://code.google.com/p/android/issues/detail?id=8175 78151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 78251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testResponseCaching(TransferKind transferKind) throws IOException { 7838ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson MockResponse response = new MockResponse() 7848ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setStatus("HTTP/1.1 200 Fantastic"); 78551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "I love puppies but hate spiders", 1); 78651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 78751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 78851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 78951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 79051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 79151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 79251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // Make sure that calling skip() doesn't omit bytes from the cache. 7938ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 79451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = urlConnection.getInputStream(); 79551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love ", readAscii(in, "I love ".length())); 79651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reliableSkip(in, "puppies but hate ".length()); 79751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("spiders", readAscii(in, "spiders".length())); 79851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 79951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 80051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 801096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 80251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 8038ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); // cached! 80451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in = urlConnection.getInputStream(); 80551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love puppies but hate spiders", 80651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(in, "I love puppies but hate spiders".length())); 8078ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 8088ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 8098ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 81051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 81151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getMissCount()); 81251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 813096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(1, cache.getSuccessCount()); 814096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 81551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 81651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 81737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson public void testSecureResponseCaching() throws IOException { 81837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 81937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 82037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 82137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.play(); 82237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 82337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 82437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache.setDefault(cache); 82537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 82637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 82737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 82837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 82937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 83037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson // OpenJDK 6 fails on this line, complaining that the connection isn't open yet 83137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson String suite = connection.getCipherSuite(); 83237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson List<Certificate> localCerts = toListOrNull(connection.getLocalCertificates()); 83337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson List<Certificate> serverCerts = toListOrNull(connection.getServerCertificates()); 83437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Principal peerPrincipal = connection.getPeerPrincipal(); 83537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Principal localPrincipal = connection.getLocalPrincipal(); 83637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 83737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // cached! 83837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 83937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 84037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 84137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(1, cache.getMissCount()); 84237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(1, cache.getHitCount()); 84337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 84437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(suite, connection.getCipherSuite()); 84537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(localCerts, toListOrNull(connection.getLocalCertificates())); 84637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(serverCerts, toListOrNull(connection.getServerCertificates())); 84737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(peerPrincipal, connection.getPeerPrincipal()); 84837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(localPrincipal, connection.getLocalPrincipal()); 84937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 85037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 85137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson public void testCacheReturnsInsecureResponseForSecureRequest() throws IOException { 85237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 85337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 85437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 85537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 85637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.play(); 85737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 85837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache insecureResponseCache = new InsecureResponseCache(); 85937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache.setDefault(insecureResponseCache); 86037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 86137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 86237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 86337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 86437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 86537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // not cached! 86637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 86737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("DEF", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 86837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 86937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 8708ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson public void testResponseCachingAndRedirects() throws IOException { 8718ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse() 8728ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_PERM) 8738ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .addHeader("Location: /foo")); 8748ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 8758ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 8768ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.play(); 8778ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8788ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 8798ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson ResponseCache.setDefault(cache); 8808ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8818ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 8828ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 8838ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8848ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection = (HttpURLConnection) server.getUrl("/").openConnection(); // cached! 8858ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 8868ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8878ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getMissCount()); // 1 redirect + 1 final response = 2 8888ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getHitCount()); 8898ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson } 8908ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8918ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson public void testSecureResponseCachingAndRedirects() throws IOException { 8928ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 8938ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 8948ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse() 8958ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_PERM) 8968ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .addHeader("Location: /foo")); 8978ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 8988ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 8998ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.play(); 9008ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 9018ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 9028ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson ResponseCache.setDefault(cache); 9038ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 9048ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 9058ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 9068ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 9078ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 9088ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // cached! 9098ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 9108ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 9118ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 9128ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getMissCount()); // 1 redirect + 1 final response = 2 9138ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getHitCount()); 9148ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson } 9158ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 91683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseCacheRequestHeaders() throws IOException, URISyntaxException { 91783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 91883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 91983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 92083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson final AtomicReference<Map<String, List<String>>> requestHeadersRef 92183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson = new AtomicReference<Map<String, List<String>>>(); 92283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 92383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 92483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 92583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeadersRef.set(requestHeaders); 92683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 92783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 92883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection conn) throws IOException { 92983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 93083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 93183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson }); 93283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 93383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URL url = server.getUrl("/"); 93483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URLConnection urlConnection = url.openConnection(); 93583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("A", "android"); 93683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson readAscii(urlConnection.getInputStream(), Integer.MAX_VALUE); 93783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(Arrays.asList("android"), requestHeadersRef.get().get("A")); 93883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 93983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 94051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void reliableSkip(InputStream in, int length) throws IOException { 94151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson while (length > 0) { 94251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson length -= in.skip(length); 94351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 94451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 94551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 94651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 94751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Reads {@code count} characters from the stream. If the stream is 94851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * exhausted before {@code count} characters can be read, the remaining 94951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * characters are returned and the stream is closed. 95051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 95151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private String readAscii(InputStream in, int count) throws IOException { 95251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson StringBuilder result = new StringBuilder(); 95351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (int i = 0; i < count; i++) { 95451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson int value = in.read(); 95551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (value == -1) { 95651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 95751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 95851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 95951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson result.append((char) value); 96051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 96151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return result.toString(); 96251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 96351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 96451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithContentLengthHeader() throws IOException { 96551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.FIXED_LENGTH); 96651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 96751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 96851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithChunkedEncoding() throws IOException { 96951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.CHUNKED); 97051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 97151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 97251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithNoLengthHeaders() throws IOException { 97351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /* 97451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Intentionally empty. This case doesn't make sense because there's no 97551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * such thing as a premature disconnect when the disconnect itself 97651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * indicates the end of the data stream. 97751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 97851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 97951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 98051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testServerPrematureDisconnect(TransferKind transferKind) throws IOException { 98151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 98251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 16); 98351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(truncateViolently(response, 16)); 98451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 98551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 98651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 98751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 98851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 98951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 99051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson BufferedReader reader = new BufferedReader(new InputStreamReader( 99151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.getUrl("/").openConnection().getInputStream())); 99251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", reader.readLine()); 99351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 99451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reader.readLine(); 99551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("This implementation silently ignored a truncated HTTP body."); 99651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 99751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 99851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 99951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 100051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 100151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 100251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 100351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 100451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 100551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 100651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithContentLengthHeader() throws IOException { 100751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.FIXED_LENGTH); 100851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 100951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 101051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithChunkedEncoding() throws IOException { 101151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.CHUNKED); 101251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 101351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 101451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithNoLengthHeaders() throws IOException { 101551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.END_OF_STREAM); 101651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 101751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 101851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testClientPrematureDisconnect(TransferKind transferKind) throws IOException { 101951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 102051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 1024); 102151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 102251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 102351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 102451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 102551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 102651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 102751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 102851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 102951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 103051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 103151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 103251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.read(); 103351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("Expected an IOException because the stream is closed."); 103451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 103551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 103651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 103751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 103851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 103951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 104051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 104151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 104251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 104351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 104451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 104551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Shortens the body of {@code response} but not the corresponding headers. 104651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Only useful to test how clients respond to the premature conclusion of 104751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the HTTP body. 104851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 104951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) { 1050e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson response.setSocketPolicy(DISCONNECT_AT_END); 105151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson List<String> headers = new ArrayList<String>(response.getHeaders()); 105251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(Arrays.copyOfRange(response.getBody(), 0, numBytesToKeep)); 105351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().clear(); 105451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().addAll(headers); 105551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return response; 105651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 105751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 105851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithContentLengthHeader() throws IOException { 105951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.FIXED_LENGTH); 106051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 106151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 106251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithChunkedEncoding() throws IOException { 106351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.CHUNKED); 106451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 106551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 106651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithNoLengthHeaders() throws IOException { 106751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.END_OF_STREAM); 106851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 106951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 10700c2fd828abec671333b8b88281825fd27a783723Jesse Wilson private void testMarkAndReset(TransferKind transferKind) throws IOException { 107151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 107251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDEFGHIJKLMNOPQRSTUVWXYZ", 1024); 107351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 107451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 107551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 107651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 107751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 107851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 107951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 108051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertFalse("This implementation claims to support mark().", in.markSupported()); 108151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.mark(5); 108251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 108351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 108451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.reset(); 108551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail(); 108651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 108751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 108851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("FGHIJKLMNOPQRSTUVWXYZ", readAscii(in, Integer.MAX_VALUE)); 108951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 109051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQRSTUVWXYZ", server.getUrl("/").openConnection()); 109151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 109251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 109351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 109451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 109551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 109651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * We've had a bug where we forget the HTTP response when we see response 109751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * code 401. This causes a new HTTP request to be issued for every call into 109851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the URLConnection. 109951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 110051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testUnauthorizedResponseHandling() throws IOException { 110151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse() 110251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .addHeader("WWW-Authenticate: challenge") 110351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setResponseCode(401) // UNAUTHORIZED 110451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("Unauthorized"); 110551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 110651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 110751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 110851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 110951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 111051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 111151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 111251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 111351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 111451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 111551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 111651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, server.getRequestCount()); 111751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 111851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 11196906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testNonHexChunkSize() throws IOException { 11206906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 11216906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5\r\nABCDE\r\nG\r\nFGHIJKLMNOPQRSTU\r\n0\r\n\r\n") 11226906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 11236906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked")); 11246906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 11256906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 11266906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 11276906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 11286906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 11296906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 11306906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 11316906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 11326906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 11336906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 11346906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testMissingChunkBody() throws IOException { 11356906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 11366906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5") 11376906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 11386906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked") 1139e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setSocketPolicy(DISCONNECT_AT_END)); 11406906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 11416906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 11426906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 11436906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 11446906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 11456906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 11466906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 11476906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 11486906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 11496906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 115050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson /** 115150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * This test checks whether connections are gzipped by default. This 115250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * behavior in not required by the API, so a failure of this test does not 115350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * imply a bug in the implementation. 115450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson */ 115550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testGzipEncodingEnabledByDefault() throws IOException, InterruptedException { 115650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 115750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody(gzip("ABCABCABC".getBytes("UTF-8"))) 115850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: gzip")); 115950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 116050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 116150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 116250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCABCABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 11638116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson assertNull(connection.getContentEncoding()); 116450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 116550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 116650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 116750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 116850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 1169deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testClientConfiguredGzipContentEncoding() throws Exception { 1170deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(new MockResponse() 1171deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .setBody(gzip("ABCDEFGHIJKLMNOPQRSTUVWXYZ".getBytes("UTF-8"))) 1172deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .addHeader("Content-Encoding: gzip")); 1173deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 1174deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1175deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1176deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 1177deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 1178deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("ABCDEFGHIJKLMNOPQRSTUVWXYZ", readAscii(gunzippedIn, Integer.MAX_VALUE)); 1179deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1180deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson RecordedRequest request = server.takeRequest(); 1181deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 1182deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1183deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1184deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithFixedLength() throws Exception { 1185deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.FIXED_LENGTH); 1186deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1187deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1188deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithChunkedEncoding() throws Exception { 1189deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.CHUNKED); 1190deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1191deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 119250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testClientConfiguredCustomContentEncoding() throws Exception { 119350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 119450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody("ABCDE") 119550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: custom")); 119650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 119750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 119850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 119950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson connection.addRequestProperty("Accept-Encoding", "custom"); 120050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCDE", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 120150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 120250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 120350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: custom"); 120450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 120550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 1206deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 1207deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Test a bug where gzip input streams weren't exhausting the input stream, 1208deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * which corrupted the request that followed. 1209deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * http://code.google.com/p/android/issues/detail?id=7059 1210deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1211deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson private void testClientConfiguredGzipContentEncodingAndConnectionReuse( 1212deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson TransferKind transferKind) throws Exception { 1213deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseOne = new MockResponse(); 1214deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson responseOne.addHeader("Content-Encoding: gzip"); 1215deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseOne, gzip("one (gzipped)".getBytes("UTF-8")), 5); 1216deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseOne); 1217deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseTwo = new MockResponse(); 1218deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseTwo, "two (identity)", 5); 1219deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseTwo); 1220deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 1221deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1222deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1223deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 1224deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 1225deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("one (gzipped)", readAscii(gunzippedIn, Integer.MAX_VALUE)); 1226deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1227deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1228deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection = server.getUrl("/").openConnection(); 1229deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("two (identity)", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1230deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 1231deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1232deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1233deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 1234ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * Obnoxiously test that the chunk sizes transmitted exactly equal the 1235ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * requested data+chunk header size. Although setChunkedStreamingMode() 1236ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * isn't specific about whether the size applies to the data or the 1237ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * complete chunk, the RI interprets it as a complete chunk. 1238ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson */ 1239ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testSetChunkedStreamingMode() throws IOException, InterruptedException { 1240ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse()); 1241ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1242ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1243ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1244ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setChunkedStreamingMode(8); 1245ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setDoOutput(true); 1246ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = urlConnection.getOutputStream(); 1247ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write("ABCDEFGHIJKLMNOPQ".getBytes("US-ASCII")); 1248ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1249ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1250ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1251ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("ABCDEFGHIJKLMNOPQ", new String(request.getBody(), "US-ASCII")); 1252ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.asList(3, 3, 3, 3, 3, 2), request.getChunkSizes()); 1253ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1254ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1255ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithFixedLengthStreaming() throws Exception { 1256ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.FIXED_LENGTH); 1257ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1258ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1259ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithChunkedStreaming() throws Exception { 1260ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.CHUNKED); 1261ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1262ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1263ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void testAuthenticateWithStreamingPost(StreamingMode streamingMode) throws Exception { 1264ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1265ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1266ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1267ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1268ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1269ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1270ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1271ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1272ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1273ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1274ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1275ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 1276ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 1277ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 1278ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setChunkedStreamingMode(0); 1279ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1280ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1281ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1282ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1283ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson try { 1284ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.getInputStream(); 1285ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail(); 1286ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } catch (HttpRetryException expected) { 1287ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1288ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1289ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the request... 1290ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1291ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1292ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1293ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1294ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 129535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson public void testSecureFixedLengthStreaming() throws Exception { 129635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson testSecureStreamingPost(StreamingMode.FIXED_LENGTH); 129735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 129835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 129935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson public void testSecureChunkedStreaming() throws Exception { 130035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson testSecureStreamingPost(StreamingMode.CHUNKED); 130135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 130235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 130335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson /** 130435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson * Users have reported problems using HTTPS with streaming request bodies. 130535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson * http://code.google.com/p/android/issues/detail?id=12860 130635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson */ 130735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson private void testSecureStreamingPost(StreamingMode streamingMode) throws Exception { 130835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 130935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 131035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson server.enqueue(new MockResponse().setBody("Success!")); 131135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson server.play(); 131235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 131335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 131435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 131535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setDoOutput(true); 131635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 131735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 131835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 131935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 132035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setChunkedStreamingMode(0); 132135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 132235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson OutputStream outputStream = connection.getOutputStream(); 132335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson outputStream.write(requestBody); 132435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson outputStream.close(); 132535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals("Success!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 132635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 132735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson RecordedRequest request = server.takeRequest(); 132835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 132935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 133035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals(Collections.<Integer>emptyList(), request.getChunkSizes()); 133135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 133235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals(Arrays.asList(4), request.getChunkSizes()); 133335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 133435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 133535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 133635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 1337ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson enum StreamingMode { 1338ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson FIXED_LENGTH, CHUNKED 1339ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1340ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1341ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithPost() throws Exception { 1342ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1343ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1344ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1345ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1346ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1347ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1348ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1349ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1350ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1351ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1352ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1353ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1354ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1355ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1356ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1357ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1358ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1359ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1360ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1361ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1362ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1363ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1364ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1365ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1366ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1367da289bcd0a9e207cc03c752f7c21c9004056e179Jesse Wilson // ...but the three requests that follow include an authorization header 1368ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1369ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1370ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 1371ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1372ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1373ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1374ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1375ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1376ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1377ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithGet() throws Exception { 1378ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1379ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1380ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1381ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1382ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1383ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1384ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1385ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1386ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1387ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1388ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1389ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1390ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1391ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1392ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1393ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1394ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1395ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1396ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1397ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1398ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...but the three requests that follow requests include an authorization header 1399ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1400ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1401ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("GET / HTTP/1.1", request.getRequestLine()); 1402ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1403ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1404ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1405ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1406ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1407c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithChunkedEncoding() throws Exception { 1408c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.CHUNKED, true); 1409c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1410c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1411c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithContentLengthHeader() throws Exception { 1412c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.FIXED_LENGTH, true); 1413c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1414c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1415c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithNoLengthHeaders() throws Exception { 1416c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.END_OF_STREAM, false); 1417c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1418c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1419c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private void testRedirected(TransferKind transferKind, boolean reuse) throws Exception { 1420c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse response = new MockResponse() 1421c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1422c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo"); 1423c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson transferKind.setBody(response, "This page has moved!", 10); 1424c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(response); 1425c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1426c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1427c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1428c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1429c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1430c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1431c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1432c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1433c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1434c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1435c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1436c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson if (reuse) { 1437c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1438c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1439c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1440c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1441c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedOnHttps() throws IOException, InterruptedException { 1442c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1443059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1444c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1445c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1446c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo") 1447c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1448c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1449c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1450c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1451c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1452059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1453c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1454c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1455c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1456c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1457c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1458c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1459c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1460c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1461c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1462c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1463c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpsToHttp() throws IOException, InterruptedException { 1464c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1465059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1466c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1467c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1468c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: http://anyhost/foo") 1469c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1470c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1471c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1472c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1473059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1474c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1475c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1476c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1477c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1478c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpToHttps() throws IOException, InterruptedException { 1479c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1480c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1481c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: https://anyhost/foo") 1482c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1483c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1484c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1485c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1486c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1487c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1488c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1489c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1490c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectToAnotherOriginServer() throws Exception { 1491c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockWebServer server2 = new MockWebServer(); 1492c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.enqueue(new MockResponse().setBody("This is the 2nd server!")); 1493c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.play(); 1494c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1495c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1496c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1497c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: " + server2.getUrl("/").toString()) 1498c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1499c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the first server again!")); 1500c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1501c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1502c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1503c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the 2nd server!", 1504c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1505c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(server2.getUrl("/"), connection.getURL()); 1506c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1507c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson // make sure the first server was careful to recycle the connection 1508c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the first server again!", 1509c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(server.getUrl("/").openStream(), Integer.MAX_VALUE)); 1510c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1511c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 15120c2fd828abec671333b8b88281825fd27a783723Jesse Wilson assertContains(first.getHeaders(), "Host: " + hostName + ":" + server.getPort()); 1513c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest second = server2.takeRequest(); 15140c2fd828abec671333b8b88281825fd27a783723Jesse Wilson assertContains(second.getHeaders(), "Host: " + hostName + ":" + server2.getPort()); 1515c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest third = server.takeRequest(); 1516c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, third.getSequenceNumber()); 1517c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1518c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.shutdown(); 1519c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1520c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1521c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testHttpsWithCustomTrustManager() throws Exception { 1522c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 1523c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingTrustManager trustManager = new RecordingTrustManager(); 1524c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLContext sc = SSLContext.getInstance("TLS"); 1525c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson sc.init(null, new TrustManager[] { trustManager }, new java.security.SecureRandom()); 1526c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1527c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HostnameVerifier defaultHostnameVerifier = HttpsURLConnection.getDefaultHostnameVerifier(); 1528c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(hostnameVerifier); 1529c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLSocketFactory defaultSSLSocketFactory = HttpsURLConnection.getDefaultSSLSocketFactory(); 1530c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory()); 1531c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson try { 1532c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1533059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1534c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1535c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 1536c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("GHI")); 1537c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1538c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1539c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URL url = server.getUrl("/"); 1540c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("ABC", readAscii(url.openStream(), Integer.MAX_VALUE)); 1541c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("DEF", readAscii(url.openStream(), Integer.MAX_VALUE)); 1542c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GHI", readAscii(url.openStream(), Integer.MAX_VALUE)); 1543c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 15440c2fd828abec671333b8b88281825fd27a783723Jesse Wilson assertEquals(Arrays.asList("verify " + hostName), hostnameVerifier.calls); 15454559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals(Arrays.asList("checkServerTrusted [" 15460c2fd828abec671333b8b88281825fd27a783723Jesse Wilson + "CN=" + hostName + " 1, " 1547b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson + "CN=Test Intermediate Certificate Authority 1, " 1548b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson + "CN=Test Root Certificate Authority 1" 1549b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson + "] RSA"), 1550c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson trustManager.calls); 1551c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } finally { 1552c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(defaultHostnameVerifier); 1553c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(defaultSSLSocketFactory); 1554c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1555c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1556c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1557eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testConnectTimeouts() throws IOException { 15585fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes StuckServer ss = new StuckServer(); 15595fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes int serverPort = ss.getLocalPort(); 1560f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom URLConnection urlConnection = new URL("http://localhost:" + serverPort).openConnection(); 1561b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson int timeout = 1000; 1562b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson urlConnection.setConnectTimeout(timeout); 1563b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson long start = System.currentTimeMillis(); 1564eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1565eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.getInputStream(); 1566eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1567eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1568b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson long actual = System.currentTimeMillis() - start; 1569b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson assertTrue(Math.abs(timeout - actual) < 500); 15705fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes } finally { 15715fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes ss.close(); 1572f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 1573eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1574eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1575eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testReadTimeouts() throws IOException { 1576eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson /* 1577eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * This relies on the fact that MockWebServer doesn't close the 1578eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * connection after a response has been sent. This causes the client to 1579eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * try to read more bytes than are sent, which results in a timeout. 1580eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson */ 1581eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson MockResponse timeout = new MockResponse() 1582eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .setBody("ABC") 1583eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .clearHeaders() 1584eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .addHeader("Content-Length: 4"); 1585eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.enqueue(timeout); 1586b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.enqueue(new MockResponse().setBody("unused")); // to keep the server alive 1587eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.play(); 1588eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1589eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 1590eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setReadTimeout(1000); 1591eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson InputStream in = urlConnection.getInputStream(); 1592eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('A', in.read()); 1593eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('B', in.read()); 1594eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('C', in.read()); 1595eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1596eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson in.read(); // if Content-Length was accurate, this would return -1 immediately 1597eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1598eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1599eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1600eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1601eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1602125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson public void testSetChunkedEncodingAsRequestProperty() throws IOException, InterruptedException { 1603125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.enqueue(new MockResponse()); 1604125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.play(); 1605125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1606125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1607125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setRequestProperty("Transfer-encoding", "chunked"); 1608125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setDoOutput(true); 1609125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.getOutputStream().write("ABC".getBytes("UTF-8")); 1610125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1611125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1612125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson RecordedRequest request = server.takeRequest(); 1613125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals("ABC", new String(request.getBody(), "UTF-8")); 1614125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson } 1615125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1616f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInRequest() throws IOException, InterruptedException { 1617f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); // server doesn't honor the connection: close header! 1618f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1619f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1620f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1621f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1622f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson a.setRequestProperty("Connection", "close"); 1623f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1624f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1625f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1626f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1627f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1628f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1629f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1630f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1631f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1632f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1633f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInResponse() throws IOException, InterruptedException { 1634f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().addHeader("Connection: close")); 1635f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1636f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1637f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1638f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1639f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1640f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1641f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1642f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1643f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1644f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1645f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1646f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1647f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1648f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1649f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseWithRedirect() throws IOException, InterruptedException { 1650f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson MockResponse response = new MockResponse() 1651f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1652f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Location: /foo") 1653f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Connection: close"); 1654f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(response); 1655f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1656f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1657f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1658f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1659f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("This is the new location!", 1660f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1661f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1662f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1663f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1664f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1665f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1666f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 166765d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson public void testResponseCodeDisagreesWithHeaders() throws IOException, InterruptedException { 166865d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.enqueue(new MockResponse() 166965d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setResponseCode(HttpURLConnection.HTTP_NO_CONTENT) 167065d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setBody("This body is not allowed!")); 167165d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.play(); 167265d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 167365d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 167465d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson assertEquals("This body is not allowed!", 167565d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 167665d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson } 167765d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 1678ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson public void testSingleByteReadIsSigned() throws IOException { 1679ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.enqueue(new MockResponse().setBody(new byte[] { -2, -1 })); 1680ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.play(); 1681ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1682ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1683ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson InputStream in = connection.getInputStream(); 1684ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(254, in.read()); 1685ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(255, in.read()); 1686ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(-1, in.read()); 1687ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson } 1688ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1689f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithChunkedEncoding() throws IOException { 1690f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.CHUNKED); 1691f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1692f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1693f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithFixedLength() throws IOException { 1694f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.FIXED_LENGTH); 1695f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1696f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1697f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithNoLengthHeaders() throws IOException { 1698f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.END_OF_STREAM); 1699f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1700f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1701f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson /** 1702f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * We explicitly permit apps to close the upload stream even after it has 1703f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * been transmitted. We also permit flush so that buffered streams can 1704f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * do a no-op flush when they are closed. http://b/3038470 1705f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson */ 1706f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson private void testFlushAfterStreamTransmitted(TransferKind transferKind) throws IOException { 1707f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.enqueue(new MockResponse().setBody("abc")); 1708f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.play(); 1709f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1710f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1711f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setDoOutput(true); 1712f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson byte[] upload = "def".getBytes("UTF-8"); 1713f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1714f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson if (transferKind == TransferKind.CHUNKED) { 1715f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setChunkedStreamingMode(0); 1716f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } else if (transferKind == TransferKind.FIXED_LENGTH) { 1717f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setFixedLengthStreamingMode(upload.length); 1718f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1719f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1720f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson OutputStream out = connection.getOutputStream(); 1721f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write(upload); 1722f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson assertEquals("abc", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1723f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1724f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.flush(); // dubious but permitted 1725f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson try { 1726f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write("ghi".getBytes("UTF-8")); 1727f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson fail(); 1728f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } catch (IOException expected) { 1729f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1730f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1731f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1732c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson public void testGetHeadersThrows() throws IOException { 1733e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse().setSocketPolicy(DISCONNECT_AT_START)); 1734c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson server.play(); 1735c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson 1736c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1737c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson try { 1738c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson connection.getInputStream(); 1739c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson fail(); 1740c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson } catch (IOException expected) { 1741c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson } 1742c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson 1743c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson try { 1744c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson connection.getInputStream(); 1745c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson fail(); 1746f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } catch (IOException expected) { 1747f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1748f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1749f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1750b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson public void testGetKeepAlive() throws Exception { 1751b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson MockWebServer server = new MockWebServer(); 1752b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1753b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.play(); 1754b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1755b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson // The request should work once and then fail 1756b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson URLConnection connection = server.getUrl("").openConnection(); 1757b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson InputStream input = connection.getInputStream(); 1758b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson assertEquals("ABC", readAscii(input, Integer.MAX_VALUE)); 1759b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson input.close(); 1760b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson try { 1761b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.getUrl("").openConnection().getInputStream(); 1762b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson fail(); 1763b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } catch (ConnectException expected) { 1764b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } 1765b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } 1766b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1767b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson /** 1768d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * This test goes through the exhaustive set of interesting ASCII characters 1769d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * because most of those characters are interesting in some way according to 1770d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * RFC 2396 and RFC 2732. http://b/1158780 1771b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson */ 1772d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson public void testLenientUrlToUri() throws Exception { 1773d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // alphanum 1774d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("abzABZ09", "abzABZ09", "abzABZ09", "abzABZ09", "abzABZ09"); 1775d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1776d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // control characters 1777d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u0001", "%01", "%01", "%01", "%01"); 1778d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u001f", "%1F", "%1F", "%1F", "%1F"); 1779d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1780d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // ascii characters 1781d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("%20", "%20", "%20", "%20", "%20"); 1782d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("%20", "%20", "%20", "%20", "%20"); 1783d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(" ", "%20", "%20", "%20", "%20"); 1784d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("!", "!", "!", "!", "!"); 1785d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\"", "%22", "%22", "%22", "%22"); 1786d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("#", null, null, null, "%23"); 1787d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("$", "$", "$", "$", "$"); 1788d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("&", "&", "&", "&", "&"); 1789d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("'", "'", "'", "'", "'"); 1790d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("(", "(", "(", "(", "("); 1791d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(")", ")", ")", ")", ")"); 1792d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("*", "*", "*", "*", "*"); 1793d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("+", "+", "+", "+", "+"); 1794d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(",", ",", ",", ",", ","); 1795d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("-", "-", "-", "-", "-"); 1796d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(".", ".", ".", ".", "."); 1797d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("/", null, "/", "/", "/"); 1798d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(":", null, ":", ":", ":"); 1799d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(";", ";", ";", ";", ";"); 1800d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("<", "%3C", "%3C", "%3C", "%3C"); 1801d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("=", "=", "=", "=", "="); 1802d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(">", "%3E", "%3E", "%3E", "%3E"); 1803d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("?", null, null, "?", "?"); 1804d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("@", "@", "@", "@", "@"); 1805d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("[", null, "%5B", null, "%5B"); 1806d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\\", "%5C", "%5C", "%5C", "%5C"); 1807d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("]", null, "%5D", null, "%5D"); 1808d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("^", "%5E", "%5E", "%5E", "%5E"); 1809d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("_", "_", "_", "_", "_"); 1810d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("`", "%60", "%60", "%60", "%60"); 1811d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("{", "%7B", "%7B", "%7B", "%7B"); 1812d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("|", "%7C", "%7C", "%7C", "%7C"); 1813d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("}", "%7D", "%7D", "%7D", "%7D"); 1814d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("~", "~", "~", "~", "~"); 1815d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("~", "~", "~", "~", "~"); 1816d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u007f", "%7F", "%7F", "%7F", "%7F"); 1817d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1818d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // beyond ascii 1819d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u0080", "%C2%80", "%C2%80", "%C2%80", "%C2%80"); 1820d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u20ac", "\u20ac", "\u20ac", "\u20ac", "\u20ac"); 182132559028b14b9b321b10eede050afd554a376569Jesse Wilson testUrlToUriMapping("\ud842\udf9f", 182232559028b14b9b321b10eede050afd554a376569Jesse Wilson "\ud842\udf9f", "\ud842\udf9f", "\ud842\udf9f", "\ud842\udf9f"); 1823d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1824d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1825d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson public void testLenientUrlToUriNul() throws Exception { 1826d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u0000", "%00", "%00", "%00", "%00"); // RI fails this 1827d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1828d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1829d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson private void testUrlToUriMapping(String string, String asAuthority, String asFile, 1830d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson String asQuery, String asFragment) throws Exception { 1831d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson if (asAuthority != null) { 1832d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host" + asAuthority + ".tld/", 1833d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host" + string + ".tld/")).toString()); 1834d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1835d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson if (asFile != null) { 1836d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host.tld/file" + asFile + "/", 1837d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host.tld/file" + string + "/")).toString()); 1838d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1839d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson if (asQuery != null) { 1840d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host.tld/file?q" + asQuery + "=x", 1841d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host.tld/file?q" + string + "=x")).toString()); 1842d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1843d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host.tld/file#" + asFragment + "-x", 1844d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host.tld/file#" + asFragment + "-x")).toString()); 1845d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1846b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1847d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson /** 1848d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * Exercises HttpURLConnection to convert URL to a URI. Unlike URL#toURI, 1849d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * HttpURLConnection recovers from URLs with unescaped but unsupported URI 1850d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * characters like '{' and '|' by escaping these characters. 1851d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson */ 1852d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson private URI backdoorUrlToUri(URL url) throws Exception { 1853d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson final AtomicReference<URI> uriReference = new AtomicReference<URI>(); 1854d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1855d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 1856d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection connection) throws IOException { 1857d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson return null; 1858d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1859d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 1860d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 1861d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson uriReference.set(uri); 1862d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson throw new UnsupportedOperationException(); 1863d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1864d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson }); 1865d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1866d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson try { 1867d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson HttpURLConnection connection = (HttpURLConnection) url.openConnection(); 1868d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson connection.getResponseCode(); 1869d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } catch (Exception expected) { 1870d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1871d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1872d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson return uriReference.get(); 1873b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } 1874b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1875afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson /** 1876afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson * Don't explode if the cache returns a null body. http://b/3373699 1877afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson */ 1878afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson public void testResponseCacheReturnsNullOutputStream() throws Exception { 1879afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson final AtomicBoolean aborted = new AtomicBoolean(); 1880afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 1881afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 1882afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 1883afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson return null; 1884afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1885afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection connection) throws IOException { 1886afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson return new CacheRequest() { 1887afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public void abort() { 1888afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson aborted.set(true); 1889afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1890afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public OutputStream getBody() throws IOException { 1891afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson return null; 1892afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1893afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson }; 1894afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1895afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson }); 1896afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson 1897afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson server.enqueue(new MockResponse().setBody("abcdef")); 1898afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson server.play(); 1899afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson 1900afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1901afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson InputStream in = connection.getInputStream(); 1902afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson assertEquals("abc", readAscii(in, 3)); 1903afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson in.close(); 1904afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson assertFalse(aborted.get()); // The best behavior is ambiguous, but RI 6 doesn't abort here 1905afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1906d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1907ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson /** 1908deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Encodes the response body using GZIP and adds the corresponding header. 1909deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1910deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public byte[] gzip(byte[] bytes) throws IOException { 1911deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1912deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson OutputStream gzippedOut = new GZIPOutputStream(bytesOut); 1913deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.write(bytes); 1914deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.close(); 1915deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson return bytesOut.toByteArray(); 1916deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1917deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 191837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private <T> List<T> toListOrNull(T[] arrayOrNull) { 191937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return arrayOrNull != null ? Arrays.asList(arrayOrNull) : null; 192037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 192137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 1922c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson /** 1923c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * Reads at most {@code limit} characters from {@code in} and asserts that 1924c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * content equals {@code expected}. 1925c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson */ 1926c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection, int limit) 1927c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson throws IOException { 1928f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom connection.connect(); 192951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(expected, readAscii(connection.getInputStream(), limit)); 1930c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ((HttpURLConnection) connection).disconnect(); 1931c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1932c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 1933c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection) throws IOException { 1934c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent(expected, connection, Integer.MAX_VALUE); 1935c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1936c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 193760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson private void assertContains(List<String> headers, String header) { 193860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertTrue(headers.toString(), headers.contains(header)); 193960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 194051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1941ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void assertContainsNoneMatching(List<String> headers, String pattern) { 1942ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (String header : headers) { 1943ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (header.matches(pattern)) { 1944ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail("Header " + header + " matches " + pattern); 1945ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1946ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1947ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1948ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1949eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson private Set<String> newSet(String... elements) { 195083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return new HashSet<String>(Arrays.asList(elements)); 195183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 195283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 195351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson enum TransferKind { 195451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson CHUNKED() { 1955deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) 195651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException { 195751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setChunkedBody(content, chunkSize); 195851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 195951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 196051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson FIXED_LENGTH() { 1961deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 196251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 196351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 196451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 196551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson END_OF_STREAM() { 1966deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 196751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 1968e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson response.setSocketPolicy(DISCONNECT_AT_END); 196951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (Iterator<String> h = response.getHeaders().iterator(); h.hasNext(); ) { 197051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (h.next().startsWith("Content-Length:")) { 197151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson h.remove(); 197251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 197351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 197451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 197551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 197651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }; 197751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1978deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson abstract void setBody(MockResponse response, byte[] content, int chunkSize) 197951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException; 1980deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1981deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson void setBody(MockResponse response, String content, int chunkSize) throws IOException { 1982deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson setBody(response, content.getBytes("UTF-8"), chunkSize); 1983deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 198451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 1985c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1986984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson enum ProxyConfig { 1987f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom NO_PROXY() { 1988f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom @Override public HttpURLConnection connect(MockWebServer server, URL url) 1989f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom throws IOException { 1990f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom return (HttpURLConnection) url.openConnection(Proxy.NO_PROXY); 1991f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 1992f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom }, 1993f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 1994984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson CREATE_ARG() { 1995984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1996984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1997984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(server.toProxyAddress()); 1998984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1999984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 2000984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 2001984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson PROXY_SYSTEM_PROPERTY() { 2002984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 2003984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 2004984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyHost", "localhost"); 2005984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyPort", Integer.toString(server.getPort())); 2006984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 2007984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 2008984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 2009984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 2010984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTP_PROXY_SYSTEM_PROPERTY() { 2011984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 2012984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 2013984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyHost", "localhost"); 2014984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyPort", Integer.toString(server.getPort())); 2015984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 2016984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 2017984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 2018984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 2019984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTPS_PROXY_SYSTEM_PROPERTY() { 2020984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 2021984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 2022984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyHost", "localhost"); 2023984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyPort", Integer.toString(server.getPort())); 2024984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 2025984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 2026984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }; 2027984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 2028984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public abstract HttpURLConnection connect(MockWebServer server, URL url) throws IOException; 2029984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 2030984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 2031c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingTrustManager implements X509TrustManager { 2032c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 2033c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 2034c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public X509Certificate[] getAcceptedIssuers() { 2035c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("getAcceptedIssuers"); 2036c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return new X509Certificate[] {}; 2037c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2038c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 2039c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkClientTrusted(X509Certificate[] chain, String authType) 2040c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 2041c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkClientTrusted " + certificatesToString(chain) + " " + authType); 2042c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2043c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 2044c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkServerTrusted(X509Certificate[] chain, String authType) 2045c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 2046c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkServerTrusted " + certificatesToString(chain) + " " + authType); 2047c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2048c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 2049c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private String certificatesToString(X509Certificate[] certificates) { 2050c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson List<String> result = new ArrayList<String>(); 2051c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson for (X509Certificate certificate : certificates) { 2052c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson result.add(certificate.getSubjectDN() + " " + certificate.getSerialNumber()); 2053c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2054c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return result.toString(); 2055c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2056c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2057c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 2058c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingHostnameVerifier implements HostnameVerifier { 2059c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 2060c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 2061c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public boolean verify(String hostname, SSLSession session) { 2062c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("verify " + hostname); 2063c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return true; 2064c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2065c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 206637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 206737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private static class InsecureResponseCache extends ResponseCache { 206837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private final DefaultResponseCache delegate = new DefaultResponseCache(); 206937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 207037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection connection) throws IOException { 207137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return delegate.put(uri, connection); 207237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 207337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 207437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 207537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 207637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson final CacheResponse response = delegate.get(uri, requestMethod, requestHeaders); 207737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson if (response instanceof SecureCacheResponse) { 207837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return new CacheResponse() { 207937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public InputStream getBody() throws IOException { 208037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response.getBody(); 208137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 208237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public Map<String, List<String>> getHeaders() throws IOException { 208337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response.getHeaders(); 208437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 208537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson }; 208637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 208737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response; 208837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 208937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 2090e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes} 2091