URLConnectionTest.java revision afd9b157f467b7c4f2f0b5592dca72f18d844602
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; 3100feece22909b7dc79fc96d666d157390b93858eJesse Wilsonimport java.net.InetAddress; 324557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.PasswordAuthentication; 33f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstromimport java.net.Proxy; 344557728efb66c455a52b7669a8eefef7a9e54854Jesse Wilsonimport java.net.ResponseCache; 3537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilsonimport java.net.SecureCacheResponse; 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; 53afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilsonimport java.util.concurrent.atomic.AtomicBoolean; 5483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilsonimport java.util.concurrent.atomic.AtomicReference; 55deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPInputStream; 56deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilsonimport java.util.zip.GZIPOutputStream; 5760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HostnameVerifier; 5860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.HttpsURLConnection; 59c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLContext; 60096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilsonimport javax.net.ssl.SSLException; 612915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstromimport javax.net.ssl.SSLHandshakeException; 6260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport javax.net.ssl.SSLSession; 63c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.SSLSocketFactory; 64c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.TrustManager; 65c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilsonimport javax.net.ssl.X509TrustManager; 662915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstromimport libcore.java.security.TestKeyStore; 6750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilsonimport libcore.javax.net.ssl.TestSSLContext; 6851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilsonimport tests.http.DefaultResponseCache; 6960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockResponse; 7060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.MockWebServer; 7160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilsonimport tests.http.RecordedRequest; 72e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport tests.http.SocketPolicy; 73e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.DISCONNECT_AT_END; 74e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.DISCONNECT_AT_START; 75e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.SHUTDOWN_INPUT_AT_END; 76e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilsonimport static tests.http.SocketPolicy.SHUTDOWN_OUTPUT_AT_END; 775fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughesimport tests.net.StuckServer; 78e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes 79e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughespublic class URLConnectionTest extends junit.framework.TestCase { 80b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 81ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private static final Authenticator SIMPLE_AUTHENTICATOR = new Authenticator() { 82ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson protected PasswordAuthentication getPasswordAuthentication() { 83ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson return new PasswordAuthentication("username", "password".toCharArray()); 84ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 85ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson }; 86ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 8751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockWebServer server = new MockWebServer(); 8800feece22909b7dc79fc96d666d157390b93858eJesse Wilson private String hostname; 8900feece22909b7dc79fc96d666d157390b93858eJesse Wilson 9000feece22909b7dc79fc96d666d157390b93858eJesse Wilson @Override protected void setUp() throws Exception { 9100feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.setUp(); 9200feece22909b7dc79fc96d666d157390b93858eJesse Wilson hostname = InetAddress.getLocalHost().getHostName(); 9300feece22909b7dc79fc96d666d157390b93858eJesse Wilson } 9451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 9551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson @Override protected void tearDown() throws Exception { 9651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(null); 97ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(null); 98984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyHost"); 99984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("proxyPort"); 100984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyHost"); 101984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("http.proxyPort"); 102984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyHost"); 103984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.clearProperty("https.proxyPort"); 10451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); 10500feece22909b7dc79fc96d666d157390b93858eJesse Wilson super.tearDown(); 10651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 10751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 10883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testRequestHeaders() throws IOException, InterruptedException { 10983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse()); 11083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 11183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 11283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 11383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "e"); 11483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("D", "f"); 11583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders = urlConnection.getRequestProperties(); 11683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("e", "f"), new HashSet<String>(requestHeaders.get("D"))); 11783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 11883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.put("G", Arrays.asList("h")); 11983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 12083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 12183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 12383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeaders.get("D").add("i"); 12483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 12583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 12683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 12783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 12883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty(null, "j"); 12983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 13083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 13183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 13383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty(null, "k"); 13483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail(); 13583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (NullPointerException expected) { 13683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 13783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("NullValue", null); // should fail silently! 13883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("AnotherNullValue", null); // should fail silently! 13983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 14083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.getResponseCode(); 14183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson RecordedRequest request = server.takeRequest(); 14283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: e"); 14383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContains(request.getHeaders(), "D: f"); 14483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "NullValue.*"); 14583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "AnotherNullValue.*"); 14683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "G:.*"); 14783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertContainsNoneMatching(request.getHeaders(), "null:.*"); 14883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 14983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 15083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("N", "o"); 15183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 15283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 15383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 15483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 15583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.setRequestProperty("P", "q"); 15683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Set header after connect"); 15783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (IllegalStateException expected) { 15883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 15983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 16083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 16183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseHeaders() throws IOException, InterruptedException { 16283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse() 16383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setStatus("HTTP/1.0 200 Fantastic") 16483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: b") 16583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .addHeader("A: c") 16683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson .setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8)); 16783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 16883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 16983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 17083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 17183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 172c1a675c80c69decadb736b245f0366f93a94a462Jesse Wilson assertEquals("HTTP/1.0 200 Fantastic", urlConnection.getHeaderField(null)); 17383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> responseHeaders = urlConnection.getHeaderFields(); 1748ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(Arrays.asList("HTTP/1.0 200 Fantastic"), responseHeaders.get(null)); 17583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(newSet("b", "c"), new HashSet<String>(responseHeaders.get("A"))); 17683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 17783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.put("N", Arrays.asList("o")); 17883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 17983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 18083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 18183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson try { 18283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson responseHeaders.get("A").add("d"); 18383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson fail("Modified an unmodifiable view."); 18483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } catch (UnsupportedOperationException expected) { 18583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 18683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 18783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 188e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // Check that if we don't read to the end of a response, the next request on the 189e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // recycled connection doesn't get the unread tail of the first request's response. 190e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes // http://code.google.com/p/android/issues/detail?id=2939 191e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes public void test_2939() throws Exception { 192b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDE\nFGHIJ\nKLMNO\nPQR", 8); 193b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 194b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 195b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 196b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 197b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 198c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 199c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("ABCDE", server.getUrl("/").openConnection(), 5); 2008baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson } 2018baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson 202977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // Check that we recognize a few basic mime types by extension. 203977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes // http://code.google.com/p/android/issues/detail?id=10100 204977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes public void test_10100() throws Exception { 205977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("image/jpeg", URLConnection.guessContentTypeFromName("someFile.jpg")); 206977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes assertEquals("application/pdf", URLConnection.guessContentTypeFromName("stuff.pdf")); 207977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes } 208977a9954414ec41256b218e6278a8544ea135d45Elliott Hughes 2098baf143a7c8921d07b54adbc66ac1e5b42de5fe6Jesse Wilson public void testConnectionsArePooled() throws Exception { 210b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson MockResponse response = new MockResponse().setBody("ABCDEFGHIJKLMNOPQR"); 211b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 212b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 213b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 214b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(response); 215b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 216b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 21706e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/foo").openConnection()); 218c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 21906e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/bar?baz=quux").openConnection()); 220c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 22106e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/z").openConnection()); 222c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 223c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 224c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 225c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testChunkedConnectionsArePooled() throws Exception { 226c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse response = new MockResponse().setChunkedBody("ABCDEFGHIJKLMNOPQR", 5); 227c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 228c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 229c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 230c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(response); 231c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 232c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 23306e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/foo").openConnection()); 234b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 23506e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/bar?baz=quux").openConnection()); 236b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 23706e15e6c528fcb773bedb43e34b0577312570927Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQR", server.getUrl("/z").openConnection()); 238b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(2, server.takeRequest().getSequenceNumber()); 239e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes } 24002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 241e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson public void testServerClosesSocket() throws Exception { 242e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson testServerClosesOutput(DISCONNECT_AT_END); 243e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 244e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 245e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson public void testServerShutdownInput() throws Exception { 246e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson testServerClosesOutput(SHUTDOWN_INPUT_AT_END); 247e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 248e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 249e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson public void testServerShutdownOutput() throws Exception { 250e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson testServerClosesOutput(SHUTDOWN_OUTPUT_AT_END); 251e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 252e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 253e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson private void testServerClosesOutput(SocketPolicy socketPolicy) throws Exception { 254e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse() 255e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setBody("This connection won't pool properly") 256e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setSocketPolicy(socketPolicy)); 257e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse() 258e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setBody("This comes after a busted connection")); 259e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.play(); 260e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 261e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertContent("This connection won't pool properly", server.getUrl("/a").openConnection()); 262e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 263e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertContent("This comes after a busted connection", server.getUrl("/b").openConnection()); 264e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson // sequence number 0 means the HTTP socket connection was not reused 265e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 266e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson } 267e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson 268b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson enum WriteKind { BYTE_BY_BYTE, SMALL_BUFFERS, LARGE_BUFFERS } 26902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 27002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_byteByByte() throws Exception { 27151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.BYTE_BY_BYTE); 27202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 27402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_smallBuffers() throws Exception { 27551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.SMALL_BUFFERS); 27602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 27702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 27802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_chunkedUpload_largeBuffers() throws Exception { 27951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.CHUNKED, WriteKind.LARGE_BUFFERS); 28002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 28102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 28202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_byteByByte() throws Exception { 28351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.BYTE_BY_BYTE); 28402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 28502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 28602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_smallBuffers() throws Exception { 28751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.SMALL_BUFFERS); 28802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 28902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 29002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes public void test_fixedLengthUpload_largeBuffers() throws Exception { 29151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson doUpload(TransferKind.FIXED_LENGTH, WriteKind.LARGE_BUFFERS); 29202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 29302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes 29451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void doUpload(TransferKind uploadKind, WriteKind writeKind) throws Exception { 29502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes int n = 512*1024; 296b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.setBodyLimit(0); 297b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse()); 298b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 299b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 300b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson HttpURLConnection conn = (HttpURLConnection) server.getUrl("/").openConnection(); 30102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setDoOutput(true); 30202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setRequestMethod("POST"); 30351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 30402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setChunkedStreamingMode(-1); 30502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 30602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes conn.setFixedLengthStreamingMode(n); 30702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 30802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes OutputStream out = conn.getOutputStream(); 30902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes if (writeKind == WriteKind.BYTE_BY_BYTE) { 31002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; ++i) { 31102f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write('x'); 31202f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 31302f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } else { 31402f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes byte[] buf = new byte[writeKind == WriteKind.SMALL_BUFFERS ? 256 : 64*1024]; 31502f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes Arrays.fill(buf, (byte) 'x'); 31602f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes for (int i = 0; i < n; i += buf.length) { 31702f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.write(buf, 0, Math.min(buf.length, n - i)); 31802f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 31902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 32002f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes out.close(); 3214cb7f05dc68abb23ae54a5891c369062185f2210Elliott Hughes assertEquals(200, conn.getResponseCode()); 322b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson RecordedRequest request = server.takeRequest(); 323b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertEquals(n, request.getBodySize()); 32451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (uploadKind == TransferKind.CHUNKED) { 325b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().size() > 0); 326b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } else { 327b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson assertTrue(request.getChunkSizes().isEmpty()); 328b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson } 32902f0cb2eb84a112fcf644d7d1fd0b5f94ea2f03bElliott Hughes } 3306247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 33151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 33251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Test that response caching is consistent with the RI and the spec. 33351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://www.w3.org/Protocols/rfc2616/rfc2616-sec13.html#sec13.4 33451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 3356247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes public void test_responseCaching() throws Exception { 3366247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // Test each documented HTTP/1.1 code, plus the first unused value in each range. 3376247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html 3386247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 3396247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // We can't test 100 because it's not really a response. 3406247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // assertCached(false, 100); 3416247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 101); 3426247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 102); 3436247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 200); 3446247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 201); 3456247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 202); 3466247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 203); 3476247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 204); 3486247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 205); 3496247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 206); 3506247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 207); 35151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_300.) 3526247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(true, 301); 3536247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 302; i <= 308; ++i) { 3546247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3556247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3566247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 400; i <= 406; ++i) { 3576247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3586247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3596247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes // (See test_responseCaching_407.) 3606247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 408); 3616247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, 409); 36251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // (See test_responseCaching_410.) 3636247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 411; i <= 418; ++i) { 3646247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3656247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3666247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes for (int i = 500; i <= 506; ++i) { 3676247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertCached(false, i); 3686247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3696247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3706247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 37151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_300() throws Exception { 37251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: fix this for android 37351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 300); 37451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 37551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3761f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson /** 3771f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * Response code 407 should only come from proxy servers. Android's client 3781f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson * throws if it is sent by an origin server. 3791f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson */ 3801f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson public void testOriginServerSends407() throws Exception { 3811f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.enqueue(new MockResponse().setResponseCode(407)); 3821f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson server.play(); 3831f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson 3841f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson URL url = server.getUrl("/"); 3851f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 3861f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson try { 3871f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson conn.getResponseCode(); 3881f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson fail(); 3891f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } catch (IOException expected) { 3901f8243e3d2b5a3f8e0398c304d1dea0395cbc368Jesse Wilson } 3916247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 3926247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 39351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void test_responseCaching_410() throws Exception { 39451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // the HTTP spec permits caching 410s, but the RI doesn't. 39551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertCached(false, 410); 39651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 39751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 3986247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes private void assertCached(boolean shouldPut, int responseCode) throws Exception { 39951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server = new MockWebServer(); 400b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.enqueue(new MockResponse() 401b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .setResponseCode(responseCode) 40251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("ABCDE") 403b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson .addHeader("WWW-Authenticate: challenge")); 404b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson server.play(); 405b1b5baac449d2725002338735f4db34bec8fd001Jesse Wilson 40651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache responseCache = new DefaultResponseCache(); 40751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(responseCache); 40851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 40951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 4106247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes assertEquals(responseCode, conn.getResponseCode()); 4116247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes 41251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // exhaust the content stream 41351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 41451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // TODO: remove special case once testUnauthorizedResponseHandling() is fixed 41551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (responseCode != 401) { 41651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(conn.getInputStream(), Integer.MAX_VALUE); 41751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 41851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException ignored) { 41951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 42051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 42151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson Set<URI> expectedCachedUris = shouldPut 42251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ? Collections.singleton(url.toURI()) 42351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson : Collections.<URI>emptySet(); 42451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(Integer.toString(responseCode), 42551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson expectedCachedUris, responseCache.getContents().keySet()); 42651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.shutdown(); // tearDown() isn't sufficient; this test starts multiple servers 4276247987eb505a482a67f5f19678260d9e7240a5fElliott Hughes } 42860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 42960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson public void testConnectViaHttps() throws IOException, InterruptedException { 43060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 43160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 432059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 433c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 43460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson server.play(); 43560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 436096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 4374559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 43860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 439c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via HTTPS", connection); 44060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 44160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson RecordedRequest request = server.takeRequest(); 44260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 44360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 44460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 445096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson public void testConnectViaHttpsReusingConnections() throws IOException, InterruptedException { 446096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 447096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 448059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 449096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 450096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 451096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson server.play(); 452096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 453b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 454b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 455b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("this response comes via HTTPS", connection); 456b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 457b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 458b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 459b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertContent("another response via HTTPS", connection); 460b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 461b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 462b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 463b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson } 464b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 4658116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson public void testConnectViaHttpsReusingConnectionsDifferentFactories() 466b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson throws IOException, InterruptedException { 467b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 468b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 469b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 470b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 471b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.enqueue(new MockResponse().setBody("another response via HTTPS")); 472b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson server.play(); 473b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson 474096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson // install a custom SSL socket factory so the server can be authorized 475096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 476059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 477096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertContent("this response comes via HTTPS", connection); 478096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 479096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 480096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson try { 481096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 482b7155fd57239e986bbaba254a91aeb9600d60305Jesse Wilson fail("without an SSL socket factory, the connection should fail"); 483096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } catch (SSLException expected) { 484096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 485096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson } 486096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson 4874559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom public void testConnectViaHttpsWithSSLFallback() throws IOException, InterruptedException { 4884559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 4894559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4904559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 491e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse().setSocketPolicy(DISCONNECT_AT_START)); 4924559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via SSL")); 4934559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom server.play(); 4944559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4954559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 4964559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 4974559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 4984559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertContent("this response comes via SSL", connection); 4994559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 5004559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom RecordedRequest request = server.takeRequest(); 5014559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 5024559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom } 5034559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom 5042915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom /** 5052915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom * Verify that we don't retry connections on certificate verification errors. 5062915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom * 5072915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom * http://code.google.com/p/android/issues/detail?id=13178 5082915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom */ 5092915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom public void testConnectViaHttpsToUntrustedServer() throws IOException, InterruptedException { 5102915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(TestKeyStore.getClientCA2(), 5112915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom TestKeyStore.getServer()); 5122915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom 5132915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 5142915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom server.enqueue(new MockResponse()); // unused 5152915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom server.play(); 5162915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom 5172915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/foo").openConnection(); 5182915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 5192915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom try { 5202915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom connection.getInputStream(); 5212915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom fail(); 5222915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom } catch (SSLHandshakeException expected) { 5232915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom assertTrue(expected.getCause() instanceof CertificateException); 5242915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom } 5252915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom assertEquals(0, server.getRequestCount()); 5262915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom } 5272915378e253f08e47fe5a9bfd026cd1ca7c6c351Brian Carlstrom 528984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxyArg() throws Exception { 529984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.CREATE_ARG); 530984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 531984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 532984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingProxySystemProperty() throws Exception { 533984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.PROXY_SYSTEM_PROPERTY); 534984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 535984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 536984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaProxyUsingHttpProxySystemProperty() throws Exception { 537984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaProxy(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 538984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 539984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 540984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaProxy(ProxyConfig proxyConfig) throws Exception { 541c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse mockResponse = new MockResponse().setBody("this response comes via a proxy"); 54251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(mockResponse); 54351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 54460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 545984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson URL url = new URL("http://android.com/foo"); 546984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpURLConnection connection = proxyConfig.connect(server, url); 547c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a proxy", connection); 54860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 54951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest request = server.takeRequest(); 55060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET http://android.com/foo HTTP/1.1", request.getRequestLine()); 55160476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(request.getHeaders(), "Host: android.com"); 55260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 55360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 554c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithContentLengthHeader() throws IOException { 555c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(new MockResponse() 556c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .setBody("abc\r\nYOU SHOULD NOT SEE THIS") 557c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .clearHeaders() 558c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson .addHeader("Content-Length: 3")); 559c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 560c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 561c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 562c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 563c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 564c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson public void testContentDisagreesWithChunkedHeader() throws IOException { 565c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson MockResponse mockResponse = new MockResponse(); 566c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setChunkedBody("abc", 3); 567c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 568c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write(mockResponse.getBody()); 569c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson bytesOut.write("\r\nYOU SHOULD NOT SEE THIS".getBytes()); 570c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.setBody(bytesOut.toByteArray()); 571c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.clearHeaders(); 572c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson mockResponse.addHeader("Transfer-encoding: chunked"); 573c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 574c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.enqueue(mockResponse); 575c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson server.play(); 576c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 577c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("abc", server.getUrl("/").openConnection()); 578c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 579c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 580f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom public void testConnectViaHttpProxyToHttpsUsingProxyArgWithNoProxy() throws Exception { 581f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom testConnectViaDirectProxyToHttps(ProxyConfig.NO_PROXY); 582f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 583f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 584f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom public void testConnectViaHttpProxyToHttpsUsingHttpProxySystemProperty() throws Exception { 585f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom // https should not use http proxy 586f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom testConnectViaDirectProxyToHttps(ProxyConfig.HTTP_PROXY_SYSTEM_PROPERTY); 587f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 588f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 589f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom private void testConnectViaDirectProxyToHttps(ProxyConfig proxyConfig) throws Exception { 590f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom TestSSLContext testSSLContext = TestSSLContext.create(); 591f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 592f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 593f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.enqueue(new MockResponse().setBody("this response comes via HTTPS")); 594f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom server.play(); 595f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 596f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom URL url = server.getUrl("/foo"); 597f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom HttpsURLConnection connection = (HttpsURLConnection) proxyConfig.connect(server, url); 598f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 599f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 600f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom assertContent("this response comes via HTTPS", connection); 601f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 602f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom RecordedRequest request = server.takeRequest(); 603f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom assertEquals("GET /foo HTTP/1.1", request.getRequestLine()); 604f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 605f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 606f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 607984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxyArg() throws Exception { 608984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.CREATE_ARG); 609984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 610984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 611984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 612984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We weren't honoring all of the appropriate proxy system properties when 613984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * connecting via HTTPS. http://b/3097518 614984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 615984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingProxySystemProperty() throws Exception { 616984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.PROXY_SYSTEM_PROPERTY); 617984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 618984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 619984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public void testConnectViaHttpProxyToHttpsUsingHttpsProxySystemProperty() throws Exception { 620984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson testConnectViaHttpProxyToHttps(ProxyConfig.HTTPS_PROXY_SYSTEM_PROPERTY); 621984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 622984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 623984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson /** 624984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * We were verifying the wrong hostname when connecting to an HTTPS site 625984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson * through a proxy. http://b/3097277 626984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson */ 627984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson private void testConnectViaHttpProxyToHttps(ProxyConfig proxyConfig) throws Exception { 62860476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 629984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 63060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 631059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 632c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 633c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("this response comes via a secure proxy")); 63451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 63560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 63660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson URL url = new URL("https://android.com/foo"); 637984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) proxyConfig.connect(server, url); 638059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 639984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 64060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 641c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent("this response comes via a secure proxy", connection); 64260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 64351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest connect = server.takeRequest(); 64460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("Connect line failure on proxy", 64560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson "CONNECT android.com:443 HTTP/1.1", connect.getRequestLine()); 64660476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 64760476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 64851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson RecordedRequest get = server.takeRequest(); 64960476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertEquals("GET /foo HTTP/1.1", get.getRequestLine()); 65060476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertContains(get.getHeaders(), "Host: android.com"); 651984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 65260476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 65360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson 654d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson /** 655d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson * Test which headers are sent unencrypted to the HTTP proxy. 656d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson */ 657d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testProxyConnectIncludesProxyHeadersOnly() 658d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson throws IOException, InterruptedException { 659984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 660d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 661d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 662d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), true); 663d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().clearHeaders()); // for CONNECT 664d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("encrypted response from the origin server")); 665d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 666d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 667d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson URL url = new URL("https://android.com/foo"); 668d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) url.openConnection( 669d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.toProxyAddress()); 670d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Private", "Secret"); 671d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("Proxy-Authorization", "bar"); 672d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.addRequestProperty("User-Agent", "baz"); 673d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 674984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson connection.setHostnameVerifier(hostnameVerifier); 675d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContent("encrypted response from the origin server", connection); 676d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 677d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest connect = server.takeRequest(); 678d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContainsNoneMatching(connect.getHeaders(), "Private.*"); 679d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Authorization: bar"); 680d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "User-Agent: baz"); 681d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Host: android.com"); 682d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(connect.getHeaders(), "Proxy-Connection: Keep-Alive"); 683d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 684d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson RecordedRequest get = server.takeRequest(); 685d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertContains(get.getHeaders(), "Private: Secret"); 686984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson assertEquals(Arrays.asList("verify android.com"), hostnameVerifier.calls); 687d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 688d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 689d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson public void testDisconnectedConnection() throws IOException { 690d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.enqueue(new MockResponse().setBody("ABCDEFGHIJKLMNOPQR")); 691d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson server.play(); 692d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 693d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 694d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson InputStream in = connection.getInputStream(); 695d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson assertEquals('A', (char) in.read()); 696d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson connection.disconnect(); 697d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson try { 698d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson in.read(); 699d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson fail("Expected a connection closed exception"); 700d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } catch (IOException expected) { 701d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 702d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson } 703d4bddd7d1fb7b1b7f0836648228235c6e4b56a18Jesse Wilson 70451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithFixedLength() throws IOException { 70551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.FIXED_LENGTH); 70651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 70751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 70851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithChunkedEncoding() throws IOException { 70951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.CHUNKED); 71051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 71151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 71251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testResponseCachingAndInputStreamSkipWithNoLengthHeaders() throws IOException { 71351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testResponseCaching(TransferKind.END_OF_STREAM); 71451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 71551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 71651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 71751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * HttpURLConnection.getInputStream().skip(long) causes ResponseCache corruption 71851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * http://code.google.com/p/android/issues/detail?id=8175 71951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 72051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testResponseCaching(TransferKind transferKind) throws IOException { 7218ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson MockResponse response = new MockResponse() 7228ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setStatus("HTTP/1.1 200 Fantastic"); 72351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "I love puppies but hate spiders", 1); 72451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 72551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 72651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 72751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 72851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 72951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 73051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson // Make sure that calling skip() doesn't omit bytes from the cache. 7318ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 73251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = urlConnection.getInputStream(); 73351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love ", readAscii(in, "I love ".length())); 73451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reliableSkip(in, "puppies but hate ".length()); 73551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("spiders", readAscii(in, "spiders".length())); 73651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 73751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 73851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 739096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 74051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 7418ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); // cached! 74251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in = urlConnection.getInputStream(); 74351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("I love puppies but hate spiders", 74451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson readAscii(in, "I love puppies but hate spiders".length())); 7458ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 7468ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("Fantastic", urlConnection.getResponseMessage()); 7478ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 74851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(-1, in.read()); 74951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getMissCount()); 75051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 751096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(1, cache.getSuccessCount()); 752096aac7b8a607d3da237900f52cab1c5066bf992Jesse Wilson assertEquals(0, cache.getAbortCount()); 75351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 75451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 75537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson public void testSecureResponseCaching() throws IOException { 75637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 75737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 75837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 75937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.play(); 76037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 76137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 76237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache.setDefault(cache); 76337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 76437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 76537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 76637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 76737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 76837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson // OpenJDK 6 fails on this line, complaining that the connection isn't open yet 76937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson String suite = connection.getCipherSuite(); 77037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson List<Certificate> localCerts = toListOrNull(connection.getLocalCertificates()); 77137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson List<Certificate> serverCerts = toListOrNull(connection.getServerCertificates()); 77237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Principal peerPrincipal = connection.getPeerPrincipal(); 77337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Principal localPrincipal = connection.getLocalPrincipal(); 77437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 77537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // cached! 77637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 77737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 77837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 77937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(1, cache.getMissCount()); 78037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(1, cache.getHitCount()); 78137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 78237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(suite, connection.getCipherSuite()); 78337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(localCerts, toListOrNull(connection.getLocalCertificates())); 78437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(serverCerts, toListOrNull(connection.getServerCertificates())); 78537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(peerPrincipal, connection.getPeerPrincipal()); 78637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals(localPrincipal, connection.getLocalPrincipal()); 78737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 78837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 78937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson public void testCacheReturnsInsecureResponseForSecureRequest() throws IOException { 79037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 79137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 79237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 79337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 79437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson server.play(); 79537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 79637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache insecureResponseCache = new InsecureResponseCache(); 79737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson ResponseCache.setDefault(insecureResponseCache); 79837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 79937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 80037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 80137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 80237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 80337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // not cached! 80437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 80537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson assertEquals("DEF", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 80637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 80737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 8088ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson public void testResponseCachingAndRedirects() throws IOException { 8098ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse() 8108ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_PERM) 8118ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .addHeader("Location: /foo")); 8128ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 8138ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 8148ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.play(); 8158ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8168ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 8178ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson ResponseCache.setDefault(cache); 8188ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8198ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 8208ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 8218ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8228ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection = (HttpURLConnection) server.getUrl("/").openConnection(); // cached! 8238ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 8248ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8258ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getMissCount()); // 1 redirect + 1 final response = 2 8268ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getHitCount()); 8278ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson } 8288ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8298ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson public void testSecureResponseCachingAndRedirects() throws IOException { 8308ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 8318ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 8328ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse() 8338ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_PERM) 8348ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson .addHeader("Location: /foo")); 8358ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 8368ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 8378ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson server.play(); 8388ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8398ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 8408ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson ResponseCache.setDefault(cache); 8418ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8428ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 8438ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 8448ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 8458ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8468ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection = (HttpsURLConnection) server.getUrl("/").openConnection(); // cached! 8478ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 8488ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals("ABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 8498ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 8508ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getMissCount()); // 1 redirect + 1 final response = 2 8518ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson assertEquals(2, cache.getHitCount()); 8528ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson } 8538ac847a52e72f0cefbb20a6850ae04468d433a9eJesse Wilson 85483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson public void testResponseCacheRequestHeaders() throws IOException, URISyntaxException { 85583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 85683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson server.play(); 85783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 85883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson final AtomicReference<Map<String, List<String>>> requestHeadersRef 85983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson = new AtomicReference<Map<String, List<String>>>(); 86083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 86183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 86283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 86383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson requestHeadersRef.set(requestHeaders); 86483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 86583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 86683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection conn) throws IOException { 86783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return null; 86883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 86983a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson }); 87083a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 87183a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URL url = server.getUrl("/"); 87283a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson URLConnection urlConnection = url.openConnection(); 87383a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson urlConnection.addRequestProperty("A", "android"); 87483a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson readAscii(urlConnection.getInputStream(), Integer.MAX_VALUE); 87583a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson assertEquals(Arrays.asList("android"), requestHeadersRef.get().get("A")); 87683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 87783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 87851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void reliableSkip(InputStream in, int length) throws IOException { 87951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson while (length > 0) { 88051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson length -= in.skip(length); 88151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 88251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 88351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 88451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 88551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Reads {@code count} characters from the stream. If the stream is 88651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * exhausted before {@code count} characters can be read, the remaining 88751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * characters are returned and the stream is closed. 88851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 88951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private String readAscii(InputStream in, int count) throws IOException { 89051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson StringBuilder result = new StringBuilder(); 89151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (int i = 0; i < count; i++) { 89251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson int value = in.read(); 89351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (value == -1) { 89451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 89551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 89651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 89751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson result.append((char) value); 89851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 89951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return result.toString(); 90051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 90151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 90251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithContentLengthHeader() throws IOException { 90351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.FIXED_LENGTH); 90451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 90551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 90651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithChunkedEncoding() throws IOException { 90751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testServerPrematureDisconnect(TransferKind.CHUNKED); 90851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 90951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 91051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testServerDisconnectsPrematurelyWithNoLengthHeaders() throws IOException { 91151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /* 91251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Intentionally empty. This case doesn't make sense because there's no 91351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * such thing as a premature disconnect when the disconnect itself 91451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * indicates the end of the data stream. 91551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 91651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 91751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 91851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testServerPrematureDisconnect(TransferKind transferKind) throws IOException { 91951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 92051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 16); 92151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(truncateViolently(response, 16)); 92251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 92351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 92451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 92551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 92651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 92751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 92851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson BufferedReader reader = new BufferedReader(new InputStreamReader( 92951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.getUrl("/").openConnection().getInputStream())); 93051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", reader.readLine()); 93151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 93251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson reader.readLine(); 93351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("This implementation silently ignored a truncated HTTP body."); 93451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 93551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 93651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 93751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 93851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 93951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 94051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 94151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 94251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 94351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 94451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithContentLengthHeader() throws IOException { 94551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.FIXED_LENGTH); 94651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 94751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 94851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithChunkedEncoding() throws IOException { 94951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.CHUNKED); 95051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 95151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 95251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testClientPrematureDisconnectWithNoLengthHeaders() throws IOException { 95351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testClientPrematureDisconnect(TransferKind.END_OF_STREAM); 95451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 95551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 95651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private void testClientPrematureDisconnect(TransferKind transferKind) throws IOException { 95751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 95851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDE\nFGHIJKLMNOPQRSTUVWXYZ", 1024); 95951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 96051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(new MockResponse().setBody("Request #2")); 96151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 96251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 96351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 96451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 96551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 96651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 96751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 96851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.close(); 96951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 97051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.read(); 97151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail("Expected an IOException because the stream is closed."); 97251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 97351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 97451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 97551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 97651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(0, cache.getSuccessCount()); 97751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("Request #2", server.getUrl("/").openConnection()); 97851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getAbortCount()); 97951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 98051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 98151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 98251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 98351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Shortens the body of {@code response} but not the corresponding headers. 98451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * Only useful to test how clients respond to the premature conclusion of 98551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the HTTP body. 98651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 98751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) { 988e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson response.setSocketPolicy(DISCONNECT_AT_END); 98951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson List<String> headers = new ArrayList<String>(response.getHeaders()); 99051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(Arrays.copyOfRange(response.getBody(), 0, numBytesToKeep)); 99151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().clear(); 99251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.getHeaders().addAll(headers); 99351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson return response; 99451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 99551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 99651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithContentLengthHeader() throws IOException { 99751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.FIXED_LENGTH); 99851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 99951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 100051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithChunkedEncoding() throws IOException { 100151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.CHUNKED); 100251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 100351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 100451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndResetWithNoLengthHeaders() throws IOException { 100551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson testMarkAndReset(TransferKind.END_OF_STREAM); 100651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 100751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 100851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testMarkAndReset(TransferKind transferKind) throws IOException { 100951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse(); 101051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson transferKind.setBody(response, "ABCDEFGHIJKLMNOPQRSTUVWXYZ", 1024); 101151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 101251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 101351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 101451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson DefaultResponseCache cache = new DefaultResponseCache(); 101551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson ResponseCache.setDefault(cache); 101651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 101751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson InputStream in = server.getUrl("/").openConnection().getInputStream(); 101851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertFalse("This implementation claims to support mark().", in.markSupported()); 101951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.mark(5); 102051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("ABCDE", readAscii(in, 5)); 102151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson try { 102251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson in.reset(); 102351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson fail(); 102451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } catch (IOException expected) { 102551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 102651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals("FGHIJKLMNOPQRSTUVWXYZ", readAscii(in, Integer.MAX_VALUE)); 102751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 102851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertContent("ABCDEFGHIJKLMNOPQRSTUVWXYZ", server.getUrl("/").openConnection()); 102951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getSuccessCount()); 103051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, cache.getHitCount()); 103151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 103251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 103351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson /** 103451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * We've had a bug where we forget the HTTP response when we see response 103551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * code 401. This causes a new HTTP request to be issued for every call into 103651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson * the URLConnection. 103751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson */ 103851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson public void testUnauthorizedResponseHandling() throws IOException { 103951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson MockResponse response = new MockResponse() 104051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .addHeader("WWW-Authenticate: challenge") 104151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setResponseCode(401) // UNAUTHORIZED 104251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson .setBody("Unauthorized"); 104351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 104451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 104551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.enqueue(response); 104651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson server.play(); 104751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 104851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson URL url = server.getUrl("/"); 104951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson HttpURLConnection conn = (HttpURLConnection) url.openConnection(); 105051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 105151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 105251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 105351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(401, conn.getResponseCode()); 105451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(1, server.getRequestCount()); 105551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 105651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 10576906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testNonHexChunkSize() throws IOException { 10586906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 10596906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5\r\nABCDE\r\nG\r\nFGHIJKLMNOPQRSTU\r\n0\r\n\r\n") 10606906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 10616906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked")); 10626906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 10636906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 10646906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 10656906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 10666906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 10676906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 10686906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 10696906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10706906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10716906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 10726906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson public void testMissingChunkBody() throws IOException { 10736906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.enqueue(new MockResponse() 10746906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .setBody("5") 10756906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .clearHeaders() 10766906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson .addHeader("Transfer-encoding: chunked") 1077e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson .setSocketPolicy(DISCONNECT_AT_END)); 10786906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson server.play(); 10796906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 10806906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 10816906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson try { 10826906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE); 10836906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson fail(); 10846906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } catch (IOException e) { 10856906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10866906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson } 10876906b0c12dcf3216883d0373973a252812a20d32Jesse Wilson 108850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson /** 108950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * This test checks whether connections are gzipped by default. This 109050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * behavior in not required by the API, so a failure of this test does not 109150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson * imply a bug in the implementation. 109250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson */ 109350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testGzipEncodingEnabledByDefault() throws IOException, InterruptedException { 109450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 109550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody(gzip("ABCABCABC".getBytes("UTF-8"))) 109650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: gzip")); 109750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 109850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 109950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 110050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCABCABC", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 11018116f7e97e00d223e7fbe5c950c9a5e3277de124Jesse Wilson assertNull(connection.getContentEncoding()); 110250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 110350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 110450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 110550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 110650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 1107deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testClientConfiguredGzipContentEncoding() throws Exception { 1108deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(new MockResponse() 1109deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .setBody(gzip("ABCDEFGHIJKLMNOPQRSTUVWXYZ".getBytes("UTF-8"))) 1110deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson .addHeader("Content-Encoding: gzip")); 1111deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 1112deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1113deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1114deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 1115deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 1116deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("ABCDEFGHIJKLMNOPQRSTUVWXYZ", readAscii(gunzippedIn, Integer.MAX_VALUE)); 1117deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1118deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson RecordedRequest request = server.takeRequest(); 1119deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: gzip"); 1120deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1121deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1122deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithFixedLength() throws Exception { 1123deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.FIXED_LENGTH); 1124deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1125deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1126deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public void testGzipAndConnectionReuseWithChunkedEncoding() throws Exception { 1127deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson testClientConfiguredGzipContentEncodingAndConnectionReuse(TransferKind.CHUNKED); 1128deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1129deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 113050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson public void testClientConfiguredCustomContentEncoding() throws Exception { 113150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.enqueue(new MockResponse() 113250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .setBody("ABCDE") 113350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson .addHeader("Content-Encoding: custom")); 113450ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson server.play(); 113550ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 113650ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 113750ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson connection.addRequestProperty("Accept-Encoding", "custom"); 113850ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertEquals("ABCDE", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 113950ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 114050ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson RecordedRequest request = server.takeRequest(); 114150ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson assertContains(request.getHeaders(), "Accept-Encoding: custom"); 114250ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson } 114350ae32218918eae80298bd1ab8e4f588bbbabdb2Jesse Wilson 1144deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 1145deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Test a bug where gzip input streams weren't exhausting the input stream, 1146deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * which corrupted the request that followed. 1147deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * http://code.google.com/p/android/issues/detail?id=7059 1148deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1149deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson private void testClientConfiguredGzipContentEncodingAndConnectionReuse( 1150deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson TransferKind transferKind) throws Exception { 1151deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseOne = new MockResponse(); 1152deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson responseOne.addHeader("Content-Encoding: gzip"); 1153deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseOne, gzip("one (gzipped)".getBytes("UTF-8")), 5); 1154deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseOne); 1155deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson MockResponse responseTwo = new MockResponse(); 1156deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson transferKind.setBody(responseTwo, "two (identity)", 5); 1157deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.enqueue(responseTwo); 1158deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson server.play(); 1159deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1160deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1161deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection.addRequestProperty("Accept-Encoding", "gzip"); 1162deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson InputStream gunzippedIn = new GZIPInputStream(connection.getInputStream()); 1163deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("one (gzipped)", readAscii(gunzippedIn, Integer.MAX_VALUE)); 1164deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1165deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1166deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson connection = server.getUrl("/").openConnection(); 1167deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals("two (identity)", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1168deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson assertEquals(1, server.takeRequest().getSequenceNumber()); 1169deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1170deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1171deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson /** 1172ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * Obnoxiously test that the chunk sizes transmitted exactly equal the 1173ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * requested data+chunk header size. Although setChunkedStreamingMode() 1174ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * isn't specific about whether the size applies to the data or the 1175ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson * complete chunk, the RI interprets it as a complete chunk. 1176ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson */ 1177ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testSetChunkedStreamingMode() throws IOException, InterruptedException { 1178ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse()); 1179ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1180ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1181ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1182ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setChunkedStreamingMode(8); 1183ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson urlConnection.setDoOutput(true); 1184ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = urlConnection.getOutputStream(); 1185ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write("ABCDEFGHIJKLMNOPQ".getBytes("US-ASCII")); 1186ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1187ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1188ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1189ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("ABCDEFGHIJKLMNOPQ", new String(request.getBody(), "US-ASCII")); 1190ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.asList(3, 3, 3, 3, 3, 2), request.getChunkSizes()); 1191ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1192ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1193ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithFixedLengthStreaming() throws Exception { 1194ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.FIXED_LENGTH); 1195ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1196ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1197ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithChunkedStreaming() throws Exception { 1198ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson testAuthenticateWithStreamingPost(StreamingMode.CHUNKED); 1199ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1200ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1201ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void testAuthenticateWithStreamingPost(StreamingMode streamingMode) throws Exception { 1202ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1203ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1204ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1205ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1206ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1207ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1208ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1209ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1210ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1211ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1212ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1213ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 1214ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 1215ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 1216ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setChunkedStreamingMode(0); 1217ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1218ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1219ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1220ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1221ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson try { 1222ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.getInputStream(); 1223ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail(); 1224ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } catch (HttpRetryException expected) { 1225ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1226ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1227ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the request... 1228ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1229ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1230ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1231ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1232ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 123335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson public void testSecureFixedLengthStreaming() throws Exception { 123435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson testSecureStreamingPost(StreamingMode.FIXED_LENGTH); 123535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 123635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 123735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson public void testSecureChunkedStreaming() throws Exception { 123835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson testSecureStreamingPost(StreamingMode.CHUNKED); 123935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 124035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 124135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson /** 124235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson * Users have reported problems using HTTPS with streaming request bodies. 124335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson * http://code.google.com/p/android/issues/detail?id=12860 124435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson */ 124535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson private void testSecureStreamingPost(StreamingMode streamingMode) throws Exception { 124635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 124735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 124835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson server.enqueue(new MockResponse().setBody("Success!")); 124935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson server.play(); 125035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 125135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 125235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 125335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setDoOutput(true); 125435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 125535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 125635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setFixedLengthStreamingMode(requestBody.length); 125735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 125835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson connection.setChunkedStreamingMode(0); 125935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 126035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson OutputStream outputStream = connection.getOutputStream(); 126135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson outputStream.write(requestBody); 126235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson outputStream.close(); 126335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals("Success!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 126435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 126535eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson RecordedRequest request = server.takeRequest(); 126635eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 126735eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson if (streamingMode == StreamingMode.FIXED_LENGTH) { 126835eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals(Collections.<Integer>emptyList(), request.getChunkSizes()); 126935eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } else if (streamingMode == StreamingMode.CHUNKED) { 127035eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals(Arrays.asList(4), request.getChunkSizes()); 127135eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 127235eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 127335eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson } 127435eef71e8ce721c4199c525890ecc1a263054596Jesse Wilson 1275ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson enum StreamingMode { 1276ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson FIXED_LENGTH, CHUNKED 1277ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1278ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1279ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithPost() throws Exception { 1280ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1281ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1282ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1283ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1284ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1285ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1286ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1287ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1288ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1289ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1290ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1291ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1292ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1293ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1294ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson connection.setDoOutput(true); 1295ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson byte[] requestBody = { 'A', 'B', 'C', 'D' }; 1296ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson OutputStream outputStream = connection.getOutputStream(); 1297ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.write(requestBody); 1298ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson outputStream.close(); 1299ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1300ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1301ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1302ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1303ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1304ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1305da289bcd0a9e207cc03c752f7c21c9004056e179Jesse Wilson // ...but the three requests that follow include an authorization header 1306ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1307ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1308ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("POST / HTTP/1.1", request.getRequestLine()); 1309ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1310ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1311ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals(Arrays.toString(requestBody), Arrays.toString(request.getBody())); 1312ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1313ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1314ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1315ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson public void testAuthenticateWithGet() throws Exception { 1316ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson MockResponse pleaseAuthenticate = new MockResponse() 1317ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setResponseCode(401) 1318ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .addHeader("WWW-Authenticate: Basic realm=\"protected area\"") 1319ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson .setBody("Please authenticate."); 1320ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // fail auth three times... 1321ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1322ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1323ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(pleaseAuthenticate); 1324ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...then succeed the fourth time 1325ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.enqueue(new MockResponse().setBody("Successful auth!")); 1326ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson server.play(); 1327ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1328ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson Authenticator.setDefault(SIMPLE_AUTHENTICATOR); 1329ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1330ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("Successful auth!", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1331ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1332ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // no authorization header for the first request... 1333ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson RecordedRequest request = server.takeRequest(); 1334ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContainsNoneMatching(request.getHeaders(), "Authorization: Basic .*"); 1335ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1336ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson // ...but the three requests that follow requests include an authorization header 1337ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (int i = 0; i < 3; i++) { 1338ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson request = server.takeRequest(); 1339ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertEquals("GET / HTTP/1.1", request.getRequestLine()); 1340ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson assertContains(request.getHeaders(), "Authorization: Basic " 1341ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson + "dXNlcm5hbWU6cGFzc3dvcmQ="); // "dXNl..." == base64("username:password") 1342ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1343ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1344ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1345c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithChunkedEncoding() throws Exception { 1346c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.CHUNKED, true); 1347c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1348c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1349c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithContentLengthHeader() throws Exception { 1350c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.FIXED_LENGTH, true); 1351c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1352c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1353c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedWithNoLengthHeaders() throws Exception { 1354c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson testRedirected(TransferKind.END_OF_STREAM, false); 1355c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1356c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1357c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private void testRedirected(TransferKind transferKind, boolean reuse) throws Exception { 1358c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockResponse response = new MockResponse() 1359c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1360c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo"); 1361c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson transferKind.setBody(response, "This page has moved!", 10); 1362c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(response); 1363c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1364c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1365c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1366c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1367c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1368c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1369c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1370c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1371c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1372c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1373c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1374c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson if (reuse) { 1375c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1376c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1377c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1378c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1379c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectedOnHttps() throws IOException, InterruptedException { 1380c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1381059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1382c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1383c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1384c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: /foo") 1385c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1386c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1387c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1388c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1389c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1390059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1391c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the new location!", 1392c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1393c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1394c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 1395c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET / HTTP/1.1", first.getRequestLine()); 1396c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest retry = server.takeRequest(); 1397c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GET /foo HTTP/1.1", retry.getRequestLine()); 1398c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, retry.getSequenceNumber()); 1399c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1400c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1401c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpsToHttp() throws IOException, InterruptedException { 1402c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1403059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1404c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1405c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1406c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: http://anyhost/foo") 1407c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1408c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1409c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1410c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection connection = (HttpsURLConnection) server.getUrl("/").openConnection(); 1411059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom connection.setSSLSocketFactory(testSSLContext.clientContext.getSocketFactory()); 1412c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1413c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1414c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1415c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1416c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testNotRedirectedFromHttpToHttps() throws IOException, InterruptedException { 1417c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1418c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1419c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: https://anyhost/foo") 1420c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1421c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1422c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1423c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1424c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This page has moved!", 1425c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1426c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1427c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1428c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testRedirectToAnotherOriginServer() throws Exception { 1429c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson MockWebServer server2 = new MockWebServer(); 1430c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.enqueue(new MockResponse().setBody("This is the 2nd server!")); 1431c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.play(); 1432c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1433c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse() 1434c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1435c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .addHeader("Location: " + server2.getUrl("/").toString()) 1436c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson .setBody("This page has moved!")); 1437c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("This is the first server again!")); 1438c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1439c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1440c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1441c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the 2nd server!", 1442c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1443c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals(server2.getUrl("/"), connection.getURL()); 1444c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1445c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson // make sure the first server was careful to recycle the connection 1446c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("This is the first server again!", 1447c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson readAscii(server.getUrl("/").openStream(), Integer.MAX_VALUE)); 1448c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1449c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest first = server.takeRequest(); 145000feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertContains(first.getHeaders(), "Host: " + hostname + ":" + server.getPort()); 1451c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest second = server2.takeRequest(); 145200feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertContains(second.getHeaders(), "Host: " + hostname + ":" + server2.getPort()); 1453c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordedRequest third = server.takeRequest(); 1454c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("Expected connection reuse", 1, third.getSequenceNumber()); 1455c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1456c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server2.shutdown(); 1457c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1458c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1459c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void testHttpsWithCustomTrustManager() throws Exception { 1460c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingHostnameVerifier hostnameVerifier = new RecordingHostnameVerifier(); 1461c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson RecordingTrustManager trustManager = new RecordingTrustManager(); 1462c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLContext sc = SSLContext.getInstance("TLS"); 1463c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson sc.init(null, new TrustManager[] { trustManager }, new java.security.SecureRandom()); 1464c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1465c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HostnameVerifier defaultHostnameVerifier = HttpsURLConnection.getDefaultHostnameVerifier(); 1466c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(hostnameVerifier); 1467c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson SSLSocketFactory defaultSSLSocketFactory = HttpsURLConnection.getDefaultSSLSocketFactory(); 1468c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(sc.getSocketFactory()); 1469c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson try { 1470c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson TestSSLContext testSSLContext = TestSSLContext.create(); 1471059dbc04218144f985b20a228bbe98139d400d0cBrian Carlstrom server.useHttps(testSSLContext.serverContext.getSocketFactory(), false); 1472c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1473c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("DEF")); 1474c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.enqueue(new MockResponse().setBody("GHI")); 1475c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson server.play(); 1476c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1477c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson URL url = server.getUrl("/"); 1478c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("ABC", readAscii(url.openStream(), Integer.MAX_VALUE)); 1479c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("DEF", readAscii(url.openStream(), Integer.MAX_VALUE)); 1480c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson assertEquals("GHI", readAscii(url.openStream(), Integer.MAX_VALUE)); 1481c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 148200feece22909b7dc79fc96d666d157390b93858eJesse Wilson assertEquals(Arrays.asList("verify " + hostname), hostnameVerifier.calls); 14834559b1d37edcb5d7f1da086cf2e3290388d74f46Brian Carlstrom assertEquals(Arrays.asList("checkServerTrusted [" 1484b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson + "CN=" + hostname + " 1, " 1485b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson + "CN=Test Intermediate Certificate Authority 1, " 1486b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson + "CN=Test Root Certificate Authority 1" 1487b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson + "] RSA"), 1488c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson trustManager.calls); 1489c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } finally { 1490c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultHostnameVerifier(defaultHostnameVerifier); 1491c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson HttpsURLConnection.setDefaultSSLSocketFactory(defaultSSLSocketFactory); 1492c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1493c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1494c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1495eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testConnectTimeouts() throws IOException { 14965fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes StuckServer ss = new StuckServer(); 14975fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes int serverPort = ss.getLocalPort(); 1498f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom URLConnection urlConnection = new URL("http://localhost:" + serverPort).openConnection(); 1499b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson int timeout = 1000; 1500b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson urlConnection.setConnectTimeout(timeout); 1501b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson long start = System.currentTimeMillis(); 1502eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1503eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.getInputStream(); 1504eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1505eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1506b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson long actual = System.currentTimeMillis() - start; 1507b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson assertTrue(Math.abs(timeout - actual) < 500); 15085fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes } finally { 15095fc5dde4c719c1dfdac46b67d5d2e4884d07721eElliott Hughes ss.close(); 1510f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 1511eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1512eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1513eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson public void testReadTimeouts() throws IOException { 1514eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson /* 1515eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * This relies on the fact that MockWebServer doesn't close the 1516eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * connection after a response has been sent. This causes the client to 1517eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson * try to read more bytes than are sent, which results in a timeout. 1518eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson */ 1519eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson MockResponse timeout = new MockResponse() 1520eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .setBody("ABC") 1521eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .clearHeaders() 1522eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson .addHeader("Content-Length: 4"); 1523eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.enqueue(timeout); 1524b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.enqueue(new MockResponse().setBody("unused")); // to keep the server alive 1525eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson server.play(); 1526eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1527eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson URLConnection urlConnection = server.getUrl("/").openConnection(); 1528eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson urlConnection.setReadTimeout(1000); 1529eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson InputStream in = urlConnection.getInputStream(); 1530eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('A', in.read()); 1531eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('B', in.read()); 1532eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson assertEquals('C', in.read()); 1533eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson try { 1534eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson in.read(); // if Content-Length was accurate, this would return -1 immediately 1535eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson fail(); 1536eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } catch (SocketTimeoutException expected) { 1537eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1538eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson } 1539eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson 1540125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson public void testSetChunkedEncodingAsRequestProperty() throws IOException, InterruptedException { 1541125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.enqueue(new MockResponse()); 1542125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson server.play(); 1543125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1544125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson HttpURLConnection urlConnection = (HttpURLConnection) server.getUrl("/").openConnection(); 1545125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setRequestProperty("Transfer-encoding", "chunked"); 1546125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.setDoOutput(true); 1547125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson urlConnection.getOutputStream().write("ABC".getBytes("UTF-8")); 1548125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals(200, urlConnection.getResponseCode()); 1549125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1550125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson RecordedRequest request = server.takeRequest(); 1551125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson assertEquals("ABC", new String(request.getBody(), "UTF-8")); 1552125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson } 1553125f068f0a6cd739beac97821c9421cf8317cc87Jesse Wilson 1554f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInRequest() throws IOException, InterruptedException { 1555f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); // server doesn't honor the connection: close header! 1556f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1557f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1558f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1559f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1560f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson a.setRequestProperty("Connection", "close"); 1561f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1562f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1563f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1564f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1565f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1566f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1567f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1568f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1569f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1570f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1571f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseInResponse() throws IOException, InterruptedException { 1572f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().addHeader("Connection: close")); 1573f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse()); 1574f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1575f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1576f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection a = (HttpURLConnection) server.getUrl("/").openConnection(); 1577f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, a.getResponseCode()); 1578f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1579f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson HttpURLConnection b = (HttpURLConnection) server.getUrl("/").openConnection(); 1580f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(200, b.getResponseCode()); 1581f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1582f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1583f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1584f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1585f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1586f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1587f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson public void testConnectionCloseWithRedirect() throws IOException, InterruptedException { 1588f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson MockResponse response = new MockResponse() 1589f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP) 1590f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Location: /foo") 1591f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson .addHeader("Connection: close"); 1592f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(response); 1593f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.enqueue(new MockResponse().setBody("This is the new location!")); 1594f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson server.play(); 1595f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1596f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1597f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("This is the new location!", 1598f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1599f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 1600f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals(0, server.takeRequest().getSequenceNumber()); 1601f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson assertEquals("When connection: close is used, each request should get its own connection", 1602f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 0, server.takeRequest().getSequenceNumber()); 1603f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson } 1604f418bf447fd007cd2ec2d45b4b0399a11904e9b4Jesse Wilson 160565d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson public void testResponseCodeDisagreesWithHeaders() throws IOException, InterruptedException { 160665d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.enqueue(new MockResponse() 160765d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setResponseCode(HttpURLConnection.HTTP_NO_CONTENT) 160865d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson .setBody("This body is not allowed!")); 160965d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson server.play(); 161065d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 161165d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 161265d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson assertEquals("This body is not allowed!", 161365d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 161465d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson } 161565d890eb22aeba9b009ee642ffd4fff48a6f98aeJesse Wilson 1616ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson public void testSingleByteReadIsSigned() throws IOException { 1617ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.enqueue(new MockResponse().setBody(new byte[] { -2, -1 })); 1618ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson server.play(); 1619ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1620ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson URLConnection connection = server.getUrl("/").openConnection(); 1621ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson InputStream in = connection.getInputStream(); 1622ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(254, in.read()); 1623ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(255, in.read()); 1624ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson assertEquals(-1, in.read()); 1625ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson } 1626ef66494dce45a0b7ec22ec3fb20c60096517a4e3Jesse Wilson 1627f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithChunkedEncoding() throws IOException { 1628f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.CHUNKED); 1629f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1630f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1631f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithFixedLength() throws IOException { 1632f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.FIXED_LENGTH); 1633f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1634f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1635f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson public void testFlushAfterStreamTransmittedWithNoLengthHeaders() throws IOException { 1636f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson testFlushAfterStreamTransmitted(TransferKind.END_OF_STREAM); 1637f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1638f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1639f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson /** 1640f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * We explicitly permit apps to close the upload stream even after it has 1641f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * been transmitted. We also permit flush so that buffered streams can 1642f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson * do a no-op flush when they are closed. http://b/3038470 1643f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson */ 1644f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson private void testFlushAfterStreamTransmitted(TransferKind transferKind) throws IOException { 1645f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.enqueue(new MockResponse().setBody("abc")); 1646f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson server.play(); 1647f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1648f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1649f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setDoOutput(true); 1650f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson byte[] upload = "def".getBytes("UTF-8"); 1651f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1652f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson if (transferKind == TransferKind.CHUNKED) { 1653f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setChunkedStreamingMode(0); 1654f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } else if (transferKind == TransferKind.FIXED_LENGTH) { 1655f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson connection.setFixedLengthStreamingMode(upload.length); 1656f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1657f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1658f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson OutputStream out = connection.getOutputStream(); 1659f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write(upload); 1660f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson assertEquals("abc", readAscii(connection.getInputStream(), Integer.MAX_VALUE)); 1661f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1662f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.flush(); // dubious but permitted 1663f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson try { 1664f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson out.write("ghi".getBytes("UTF-8")); 1665f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson fail(); 1666f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } catch (IOException expected) { 1667f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1668f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1669f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1670c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson public void testGetHeadersThrows() throws IOException { 1671e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson server.enqueue(new MockResponse().setSocketPolicy(DISCONNECT_AT_START)); 1672c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson server.play(); 1673c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson 1674c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1675c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson try { 1676c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson connection.getInputStream(); 1677c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson fail(); 1678c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson } catch (IOException expected) { 1679c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson } 1680c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson 1681c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson try { 1682c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson connection.getInputStream(); 1683c6dae581716b9362a5c7f166c80a7f2b46ed1124Jesse Wilson fail(); 1684f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } catch (IOException expected) { 1685f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1686f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson } 1687f29ad8a60254345d1943d1b3836482395a7c916fJesse Wilson 1688b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson public void testGetKeepAlive() throws Exception { 1689b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson MockWebServer server = new MockWebServer(); 1690b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.enqueue(new MockResponse().setBody("ABC")); 1691b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.play(); 1692b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1693b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson // The request should work once and then fail 1694b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson URLConnection connection = server.getUrl("").openConnection(); 1695b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson InputStream input = connection.getInputStream(); 1696b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson assertEquals("ABC", readAscii(input, Integer.MAX_VALUE)); 1697b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson input.close(); 1698b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson try { 1699b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson server.getUrl("").openConnection().getInputStream(); 1700b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson fail(); 1701b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } catch (ConnectException expected) { 1702b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } 1703b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } 1704b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1705b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson /** 1706d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * This test goes through the exhaustive set of interesting ASCII characters 1707d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * because most of those characters are interesting in some way according to 1708d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * RFC 2396 and RFC 2732. http://b/1158780 1709b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson */ 1710d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson public void testLenientUrlToUri() throws Exception { 1711d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // alphanum 1712d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("abzABZ09", "abzABZ09", "abzABZ09", "abzABZ09", "abzABZ09"); 1713d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1714d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // control characters 1715d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u0001", "%01", "%01", "%01", "%01"); 1716d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u001f", "%1F", "%1F", "%1F", "%1F"); 1717d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1718d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // ascii characters 1719d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("%20", "%20", "%20", "%20", "%20"); 1720d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("%20", "%20", "%20", "%20", "%20"); 1721d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(" ", "%20", "%20", "%20", "%20"); 1722d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("!", "!", "!", "!", "!"); 1723d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\"", "%22", "%22", "%22", "%22"); 1724d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("#", null, null, null, "%23"); 1725d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("$", "$", "$", "$", "$"); 1726d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("&", "&", "&", "&", "&"); 1727d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("'", "'", "'", "'", "'"); 1728d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("(", "(", "(", "(", "("); 1729d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(")", ")", ")", ")", ")"); 1730d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("*", "*", "*", "*", "*"); 1731d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("+", "+", "+", "+", "+"); 1732d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(",", ",", ",", ",", ","); 1733d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("-", "-", "-", "-", "-"); 1734d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(".", ".", ".", ".", "."); 1735d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("/", null, "/", "/", "/"); 1736d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(":", null, ":", ":", ":"); 1737d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(";", ";", ";", ";", ";"); 1738d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("<", "%3C", "%3C", "%3C", "%3C"); 1739d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("=", "=", "=", "=", "="); 1740d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping(">", "%3E", "%3E", "%3E", "%3E"); 1741d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("?", null, null, "?", "?"); 1742d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("@", "@", "@", "@", "@"); 1743d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("[", null, "%5B", null, "%5B"); 1744d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\\", "%5C", "%5C", "%5C", "%5C"); 1745d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("]", null, "%5D", null, "%5D"); 1746d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("^", "%5E", "%5E", "%5E", "%5E"); 1747d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("_", "_", "_", "_", "_"); 1748d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("`", "%60", "%60", "%60", "%60"); 1749d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("{", "%7B", "%7B", "%7B", "%7B"); 1750d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("|", "%7C", "%7C", "%7C", "%7C"); 1751d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("}", "%7D", "%7D", "%7D", "%7D"); 1752d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("~", "~", "~", "~", "~"); 1753d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("~", "~", "~", "~", "~"); 1754d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u007f", "%7F", "%7F", "%7F", "%7F"); 1755d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1756d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson // beyond ascii 1757d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u0080", "%C2%80", "%C2%80", "%C2%80", "%C2%80"); 1758d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u20ac", "\u20ac", "\u20ac", "\u20ac", "\u20ac"); 1759d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1760d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1761d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson public void testLenientUrlToUriNul() throws Exception { 1762d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson testUrlToUriMapping("\u0000", "%00", "%00", "%00", "%00"); // RI fails this 1763d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1764d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1765d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson private void testUrlToUriMapping(String string, String asAuthority, String asFile, 1766d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson String asQuery, String asFragment) throws Exception { 1767d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson if (asAuthority != null) { 1768d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host" + asAuthority + ".tld/", 1769d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host" + string + ".tld/")).toString()); 1770d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1771d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson if (asFile != null) { 1772d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host.tld/file" + asFile + "/", 1773d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host.tld/file" + string + "/")).toString()); 1774d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1775d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson if (asQuery != null) { 1776d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host.tld/file?q" + asQuery + "=x", 1777d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host.tld/file?q" + string + "=x")).toString()); 1778d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1779d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson assertEquals("http://host.tld/file#" + asFragment + "-x", 1780d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson backdoorUrlToUri(new URL("http://host.tld/file#" + asFragment + "-x")).toString()); 1781d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1782b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1783d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson /** 1784d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * Exercises HttpURLConnection to convert URL to a URI. Unlike URL#toURI, 1785d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * HttpURLConnection recovers from URLs with unescaped but unsupported URI 1786d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson * characters like '{' and '|' by escaping these characters. 1787d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson */ 1788d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson private URI backdoorUrlToUri(URL url) throws Exception { 1789d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson final AtomicReference<URI> uriReference = new AtomicReference<URI>(); 1790d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1791d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 1792d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection connection) throws IOException { 1793d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson return null; 1794d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1795d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 1796d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 1797d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson uriReference.set(uri); 1798d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson throw new UnsupportedOperationException(); 1799d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1800d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson }); 1801d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1802d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson try { 1803d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson HttpURLConnection connection = (HttpURLConnection) url.openConnection(); 1804d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson connection.getResponseCode(); 1805d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } catch (Exception expected) { 1806d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson } 1807d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1808d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson return uriReference.get(); 1809b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson } 1810b7f4d6c3968c372767b2510f38a3d506067aced6Jesse Wilson 1811afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson /** 1812afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson * Don't explode if the cache returns a null body. http://b/3373699 1813afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson */ 1814afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson public void testResponseCacheReturnsNullOutputStream() throws Exception { 1815afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson final AtomicBoolean aborted = new AtomicBoolean(); 1816afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson ResponseCache.setDefault(new ResponseCache() { 1817afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 1818afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 1819afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson return null; 1820afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1821afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection connection) throws IOException { 1822afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson return new CacheRequest() { 1823afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public void abort() { 1824afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson aborted.set(true); 1825afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1826afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson @Override public OutputStream getBody() throws IOException { 1827afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson return null; 1828afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1829afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson }; 1830afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1831afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson }); 1832afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson 1833afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson server.enqueue(new MockResponse().setBody("abcdef")); 1834afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson server.play(); 1835afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson 1836afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson HttpURLConnection connection = (HttpURLConnection) server.getUrl("/").openConnection(); 1837afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson InputStream in = connection.getInputStream(); 1838afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson assertEquals("abc", readAscii(in, 3)); 1839afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson in.close(); 1840afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson assertFalse(aborted.get()); // The best behavior is ambiguous, but RI 6 doesn't abort here 1841afd9b157f467b7c4f2f0b5592dca72f18d844602Jesse Wilson } 1842d1b5e5da828434388e486a388710d21e4306dae0Jesse Wilson 1843ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson /** 1844deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson * Encodes the response body using GZIP and adds the corresponding header. 1845deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson */ 1846deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson public byte[] gzip(byte[] bytes) throws IOException { 1847deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1848deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson OutputStream gzippedOut = new GZIPOutputStream(bytesOut); 1849deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.write(bytes); 1850deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson gzippedOut.close(); 1851deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson return bytesOut.toByteArray(); 1852deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 1853deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 185437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private <T> List<T> toListOrNull(T[] arrayOrNull) { 185537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return arrayOrNull != null ? Arrays.asList(arrayOrNull) : null; 185637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 185737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 1858c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson /** 1859c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * Reads at most {@code limit} characters from {@code in} and asserts that 1860c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson * content equals {@code expected}. 1861c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson */ 1862c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection, int limit) 1863c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson throws IOException { 1864f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom connection.connect(); 186551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson assertEquals(expected, readAscii(connection.getInputStream(), limit)); 1866c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson ((HttpURLConnection) connection).disconnect(); 1867c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1868c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 1869c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson private void assertContent(String expected, URLConnection connection) throws IOException { 1870c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson assertContent(expected, connection, Integer.MAX_VALUE); 1871c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson } 1872c8977f474b30c5f3807398859a6b16687af6fc7bJesse Wilson 187360476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson private void assertContains(List<String> headers, String header) { 187460476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson assertTrue(headers.toString(), headers.contains(header)); 187560476787f0e0f052366d8031c74e507ffd3d16a3Jesse Wilson } 187651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1877ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson private void assertContainsNoneMatching(List<String> headers, String pattern) { 1878ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson for (String header : headers) { 1879ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson if (header.matches(pattern)) { 1880ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson fail("Header " + header + " matches " + pattern); 1881ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1882ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1883ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson } 1884ffd579b668428272b78f5c6c64f9c89766f37c1aJesse Wilson 1885eafede536f2059bb6c869e7a5f07fd7ad9758e28Jesse Wilson private Set<String> newSet(String... elements) { 188683a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson return new HashSet<String>(Arrays.asList(elements)); 188783a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson } 188883a47d4d0c536e06fc53eda9d5a1a5d93f9accc6Jesse Wilson 188951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson enum TransferKind { 189051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson CHUNKED() { 1891deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) 189251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException { 189351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setChunkedBody(content, chunkSize); 189451e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 189551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 189651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson FIXED_LENGTH() { 1897deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 189851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 189951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 190051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }, 190151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson END_OF_STREAM() { 1902deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson @Override void setBody(MockResponse response, byte[] content, int chunkSize) { 190351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson response.setBody(content); 1904e942f46f10bb9384a1b186b3d7b74f9704c57090Jesse Wilson response.setSocketPolicy(DISCONNECT_AT_END); 190551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson for (Iterator<String> h = response.getHeaders().iterator(); h.hasNext(); ) { 190651e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson if (h.next().startsWith("Content-Length:")) { 190751e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson h.remove(); 190851e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson break; 190951e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 191051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 191151e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 191251e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson }; 191351e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson 1914deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson abstract void setBody(MockResponse response, byte[] content, int chunkSize) 191551e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson throws IOException; 1916deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson 1917deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson void setBody(MockResponse response, String content, int chunkSize) throws IOException { 1918deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson setBody(response, content.getBytes("UTF-8"), chunkSize); 1919deb236fb06f2a14861e7d40dea959f181cd5cf28Jesse Wilson } 192051e468abf2628ce964d3657042f3ac8f2c947504Jesse Wilson } 1921c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1922984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson enum ProxyConfig { 1923f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom NO_PROXY() { 1924f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom @Override public HttpURLConnection connect(MockWebServer server, URL url) 1925f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom throws IOException { 1926f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom return (HttpURLConnection) url.openConnection(Proxy.NO_PROXY); 1927f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom } 1928f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom }, 1929f02c695ed03e708623d9365dec26d533356ef2d0Brian Carlstrom 1930984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson CREATE_ARG() { 1931984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1932984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1933984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(server.toProxyAddress()); 1934984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1935984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1936984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1937984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson PROXY_SYSTEM_PROPERTY() { 1938984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1939984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1940984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyHost", "localhost"); 1941984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("proxyPort", Integer.toString(server.getPort())); 1942984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1943984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1944984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1945984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1946984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTP_PROXY_SYSTEM_PROPERTY() { 1947984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1948984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1949984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyHost", "localhost"); 1950984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("http.proxyPort", Integer.toString(server.getPort())); 1951984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1952984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1953984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }, 1954984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1955984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson HTTPS_PROXY_SYSTEM_PROPERTY() { 1956984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson @Override public HttpURLConnection connect(MockWebServer server, URL url) 1957984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson throws IOException { 1958984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyHost", "localhost"); 1959984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson System.setProperty("https.proxyPort", Integer.toString(server.getPort())); 1960984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson return (HttpURLConnection) url.openConnection(); 1961984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1962984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson }; 1963984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1964984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson public abstract HttpURLConnection connect(MockWebServer server, URL url) throws IOException; 1965984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson } 1966984fcff696380abd6ea14e80030f9fd2d09dbad0Jesse Wilson 1967c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingTrustManager implements X509TrustManager { 1968c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1969c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1970c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public X509Certificate[] getAcceptedIssuers() { 1971c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("getAcceptedIssuers"); 1972c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return new X509Certificate[] {}; 1973c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1974c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1975c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkClientTrusted(X509Certificate[] chain, String authType) 1976c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1977c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkClientTrusted " + certificatesToString(chain) + " " + authType); 1978c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1979c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1980c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public void checkServerTrusted(X509Certificate[] chain, String authType) 1981c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson throws CertificateException { 1982c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("checkServerTrusted " + certificatesToString(chain) + " " + authType); 1983c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1984c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1985c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private String certificatesToString(X509Certificate[] certificates) { 1986c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson List<String> result = new ArrayList<String>(); 1987c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson for (X509Certificate certificate : certificates) { 1988c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson result.add(certificate.getSubjectDN() + " " + certificate.getSerialNumber()); 1989c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1990c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return result.toString(); 1991c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1992c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 1993c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1994c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private static class RecordingHostnameVerifier implements HostnameVerifier { 1995c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson private final List<String> calls = new ArrayList<String>(); 1996c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson 1997c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson public boolean verify(String hostname, SSLSession session) { 1998c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson calls.add("verify " + hostname); 1999c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson return true; 2000c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 2001c0372d90016d241ac979faa6fa1731f30b6f2a03Jesse Wilson } 200237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 200337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private static class InsecureResponseCache extends ResponseCache { 200437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson private final DefaultResponseCache delegate = new DefaultResponseCache(); 200537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 200637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public CacheRequest put(URI uri, URLConnection connection) throws IOException { 200737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return delegate.put(uri, connection); 200837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 200937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson 201037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public CacheResponse get(URI uri, String requestMethod, 201137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson Map<String, List<String>> requestHeaders) throws IOException { 201237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson final CacheResponse response = delegate.get(uri, requestMethod, requestHeaders); 201337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson if (response instanceof SecureCacheResponse) { 201437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return new CacheResponse() { 201537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public InputStream getBody() throws IOException { 201637dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response.getBody(); 201737dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 201837dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson @Override public Map<String, List<String>> getHeaders() throws IOException { 201937dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response.getHeaders(); 202037dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 202137dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson }; 202237dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 202337dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson return response; 202437dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 202537dcf5581f177229ca6c8e7d0d640361640bfb00Jesse Wilson } 2026e40c9e3935a5024c0f3ebfb3f1441fcd5c48ed86Elliott Hughes} 2027