15c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)/* 25c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Copyright (C) 2011 Google Inc. 35c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * 45c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Licensed under the Apache License, Version 2.0 (the "License"); 55c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * you may not use this file except in compliance with the License. 65c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * You may obtain a copy of the License at 75c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * 85c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * http://www.apache.org/licenses/LICENSE-2.0 95c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * 105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Unless required by applicable law or agreed to in writing, software 115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * distributed under the License is distributed on an "AS IS" BASIS, 125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * See the License for the specific language governing permissions and 145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * limitations under the License. 155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)package com.squareup.okhttp.mockwebserver; 175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import com.squareup.okhttp.internal.Util; 195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.io.ByteArrayInputStream; 205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.io.ByteArrayOutputStream; 215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.io.IOException; 225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.io.InputStream; 235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.io.UnsupportedEncodingException; 245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.util.ArrayList; 255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.util.Iterator; 265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.util.List; 275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import java.util.concurrent.TimeUnit; 285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)/** A scripted response to be replayed by the mock web server. */ 3053e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles)public final class MockResponse implements Cloneable { 315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) private static final String CHUNKED_BODY_HEADER = "Transfer-encoding: chunked"; 325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) private String status = "HTTP/1.1 200 OK"; 3493ac45cfc74041c8ae536ce58a9534d46db2024eTorne (Richard Coles) private List<String> headers = new ArrayList<String>(); 3553e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) 3653e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) /** The response body content, or null if {@code bodyStream} is set. */ 3753e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) private byte[] body; 3853e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) /** The response body content, or null if {@code body} is set. */ 3953e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) private InputStream bodyStream; 4053e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) 4153e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) private int throttleBytesPerPeriod = Integer.MAX_VALUE; 4253e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) private long throttlePeriod = 1; 4353e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) private TimeUnit throttleUnit = TimeUnit.SECONDS; 4453e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) 4553e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) private SocketPolicy socketPolicy = SocketPolicy.KEEP_OPEN; 46e52495584422c5edb5b2944981473a2e208da323Torne (Richard Coles) 47e52495584422c5edb5b2944981473a2e208da323Torne (Richard Coles) private int bodyDelayTimeMs = 0; 48e52495584422c5edb5b2944981473a2e208da323Torne (Richard Coles) 4953e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) private List<PushPromise> promises = new ArrayList<PushPromise>(); 5053e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) 5153e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) /** Creates a new mock response with an empty body. */ 525267f701546148b83dfbe1d151cb184385bb5c22Torne (Richard Coles) public MockResponse() { 5353e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) setBody(new byte[0]); 5453e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) } 5553e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) 5653e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) @Override public MockResponse clone() { 5753e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) try { 5853e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) MockResponse result = (MockResponse) super.clone(); 5953e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) result.headers = new ArrayList<String>(headers); 6053e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) result.promises = new ArrayList<PushPromise>(promises); 6153e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) return result; 6253e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) } catch (CloneNotSupportedException e) { 6353e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) throw new AssertionError(); 6453e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) } 6553e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) } 6653e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) 6753e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) /** Returns the HTTP response line, such as "HTTP/1.1 200 OK". */ 6853e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) public String getStatus() { 6953e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) return status; 7053e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) } 7153e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) 72591b958dee2cf159d33a0b931e6231072eaf38d5Ben Murdoch public MockResponse setResponseCode(int code) { 735c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.status = "HTTP/1.1 " + code + " OK"; 745c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 755c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 765c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 775c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setStatus(String status) { 785c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.status = status; 795c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 805c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 815c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 825c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** Returns the HTTP headers, such as "Content-Length: 0". */ 835c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public List<String> getHeaders() { 845c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return headers; 855c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 865c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 875c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 885c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Removes all HTTP headers including any "Content-Length" and 895c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * "Transfer-encoding" headers that were added by default. 905c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 915c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse clearHeaders() { 925c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) headers.clear(); 935c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 945c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 955c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 965c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 975c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Adds {@code header} as an HTTP header. For well-formed HTTP {@code header} 985c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * should contain a name followed by a colon and a value. 995c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 1005c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse addHeader(String header) { 1015c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) headers.add(header); 1025c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 1035c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1045c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1055c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 1065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Adds a new header with the name and value. This may be used to add multiple 1075c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * headers with the same name. 1085c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 1095c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse addHeader(String name, Object value) { 1105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return addHeader(name + ": " + String.valueOf(value)); 1115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 1145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Removes all headers named {@code name}, then adds a new header with the 1155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * name and value. 1165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 1175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setHeader(String name, Object value) { 1185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) removeHeader(name); 1195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return addHeader(name, value); 1205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 122926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) /** Removes all headers named {@code name}. */ 123926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) public MockResponse removeHeader(String name) { 124926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) name += ":"; 12553e740f4a82e17f3ae59772501622dc354e42336Torne (Richard Coles) for (Iterator<String> i = headers.iterator(); i.hasNext(); ) { 1265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) String header = i.next(); 1275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if (name.regionMatches(true, 0, header, 0, name.length())) { 1285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) i.remove(); 1295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 131926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) return this; 1325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** Returns the raw HTTP payload, or null if this response is streamed. */ 135926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) public byte[] getBody() { 1365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return body; 137926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) } 138926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 139926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) /** Returns an input stream containing the raw HTTP payload. */ 140926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) InputStream getBodyStream() { 1415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return bodyStream != null ? bodyStream : new ByteArrayInputStream(body); 142926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) } 143926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 1445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setBody(byte[] body) { 1455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) setHeader("Content-Length", body.length); 1465c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.body = body; 1475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.bodyStream = null; 1485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 1495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setBody(InputStream bodyStream, long bodyLength) { 1525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) setHeader("Content-Length", bodyLength); 1535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.body = null; 1545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.bodyStream = bodyStream; 1555c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 1565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 157926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 1585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** Sets the response body to the UTF-8 encoded bytes of {@code body}. */ 1595c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setBody(String body) { 1605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) try { 1615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return setBody(body.getBytes("UTF-8")); 1625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } catch (UnsupportedEncodingException e) { 1635c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) throw new AssertionError(); 1645c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1655c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1665c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1675c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 1685c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Sets the response body to {@code body}, chunked every {@code maxChunkSize} 1695c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * bytes. 1705c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 1715c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setChunkedBody(byte[] body, int maxChunkSize) { 1725c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) removeHeader("Content-Length"); 1735c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) headers.add(CHUNKED_BODY_HEADER); 1745c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1755c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) try { 1765c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) ByteArrayOutputStream bytesOut = new ByteArrayOutputStream(); 1775c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) int pos = 0; 1785c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) while (pos < body.length) { 1795c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) int chunkSize = Math.min(body.length - pos, maxChunkSize); 1805c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) bytesOut.write(Integer.toHexString(chunkSize).getBytes(Util.US_ASCII)); 1815c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) bytesOut.write("\r\n".getBytes(Util.US_ASCII)); 1825c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) bytesOut.write(body, pos, chunkSize); 1835c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) bytesOut.write("\r\n".getBytes(Util.US_ASCII)); 1845c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) pos += chunkSize; 1855c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1865c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) bytesOut.write("0\r\n\r\n".getBytes(Util.US_ASCII)); // Last chunk + empty trailer + crlf. 1875c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1885c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.body = bytesOut.toByteArray(); 1895c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 19081a5157921f1d2a7ff6aae115bfe3c139b38a5c8Torne (Richard Coles) } catch (IOException e) { 1915c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) throw new AssertionError(); // In-memory I/O doesn't throw IOExceptions. 19281a5157921f1d2a7ff6aae115bfe3c139b38a5c8Torne (Richard Coles) } 1935c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 1945c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1955c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 1965c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Sets the response body to the UTF-8 encoded bytes of {@code body}, chunked 1975c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * every {@code maxChunkSize} bytes. 1985c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 1995c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setChunkedBody(String body, int maxChunkSize) { 2005c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) try { 2015c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return setChunkedBody(body.getBytes("UTF-8"), maxChunkSize); 2025c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } catch (UnsupportedEncodingException e) { 2035c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) throw new AssertionError(); 2045c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2055c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 207e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch public SocketPolicy getSocketPolicy() { 208e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch return socketPolicy; 209e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch } 210e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch 211e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch public MockResponse setSocketPolicy(SocketPolicy socketPolicy) { 2125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.socketPolicy = socketPolicy; 2135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 2145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 2175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Throttles the response body writer to sleep for the given period after each 2185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * series of {@code bytesPerPeriod} bytes are written. Use this to simulate 2195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * network behavior. 2205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 2215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse throttleBody(int bytesPerPeriod, long period, TimeUnit unit) { 2225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.throttleBytesPerPeriod = bytesPerPeriod; 2235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.throttlePeriod = period; 2245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.throttleUnit = unit; 2255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 2265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public int getThrottleBytesPerPeriod() { 2295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return throttleBytesPerPeriod; 2305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public long getThrottlePeriod() { 2335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return throttlePeriod; 2345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public TimeUnit getThrottleUnit() { 2375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return throttleUnit; 2385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2395c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 2415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * Set the delayed time of the response body to {@code delay}. This applies to the 2425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * response body only; response headers are not affected. 2435c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 2445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse setBodyDelayTimeMs(int delay) { 2455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) bodyDelayTimeMs = delay; 2465c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 2475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public int getBodyDelayTimeMs() { 2505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return bodyDelayTimeMs; 2515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** 2545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * When {@link MockWebServer#setNpnProtocols(java.util.List) protocols} 2555c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) * include a SPDY variant, this attaches a pushed stream to this response. 2565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) */ 2575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public MockResponse withPush(PushPromise promise) { 2585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) this.promises.add(promise); 2595c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return this; 2605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) /** Returns the streams the server will push with this response. */ 2635c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) public List<PushPromise> getPushPromises() { 264926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) return promises; 265926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) } 266926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 267926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) @Override public String toString() { 268926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) return status; 2695c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) } 2705c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)} 2715c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)