Java MockResponse.setBody方法代碼示例

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
@Test public void contentDisagreesWithChunkedHeaderBodyTooShort() throws IOException {
  MockResponse mockResponse = new MockResponse();
  mockResponse.setChunkedBody("abcde", 5);

  Buffer truncatedBody = new Buffer();
  Buffer fullBody = mockResponse.getBody();
  truncatedBody.write(fullBody, fullBody.indexOf((byte) 'e'));
  mockResponse.setBody(truncatedBody);

  mockResponse.clearHeaders();
  mockResponse.addHeader("Transfer-encoding: chunked");
  mockResponse.setSocketPolicy(DISCONNECT_AT_END);

  server.enqueue(mockResponse);

  try {
    readAscii(urlFactory.open(server.url("/").url()).getInputStream(), 5);
    fail();
  } catch (ProtocolException expected) {
  }
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
@Before
public void before() throws IOException {
    metrics = new MetricRegistry();
    server = new MockWebServer();
    server.start();

    MockResponse response = new MockResponse();
    response.setBody(RESPONSE_BODY);
    server.enqueue(response);

    retrofit = new Retrofit.Builder()
            .addConverterFactory(GsonConverterFactory.create())
            .addCallAdapterFactory(new TimedCallAdapterFactory(metrics))
            .addCallAdapterFactory(RxJavaCallAdapterFactory.create())
            .baseUrl(server.url("/").toString())
            .build();
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
/**
 * Shortens the body of {@code response} but not the corresponding headers. Only useful to test
 * how clients respond to the premature conclusion of the HTTP body.
 */
private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) {
  response.setSocketPolicy(SocketPolicy.DISCONNECT_AT_END);
  Headers headers = response.getHeaders();
  Buffer truncatedBody = new Buffer();
  truncatedBody.write(response.getBody(), numBytesToKeep);
  response.setBody(truncatedBody);
  response.setHeaders(headers);
  return response;
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
/**
 * Shortens the body of {@code response} but not the corresponding headers. Only useful to test
 * how clients respond to the premature conclusion of the HTTP body.
 */
private MockResponse truncateViolently(MockResponse response, int numBytesToKeep) {
  response.setSocketPolicy(DISCONNECT_AT_END);
  Headers headers = response.getHeaders();
  Buffer truncatedBody = new Buffer();
  truncatedBody.write(response.getBody(), numBytesToKeep);
  response.setBody(truncatedBody);
  response.setHeaders(headers);
  return response;
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
@Test public void contentDisagreesWithChunkedHeaderBodyTooLong() throws IOException {
  MockResponse mockResponse = new MockResponse();
  mockResponse.setChunkedBody("abc", 3);
  Buffer buffer = mockResponse.getBody();
  buffer.writeUtf8("\r\nYOU SHOULD NOT SEE THIS");
  mockResponse.setBody(buffer);
  mockResponse.clearHeaders();
  mockResponse.addHeader("Transfer-encoding: chunked");

  server.enqueue(mockResponse);

  assertContent("abc", urlFactory.open(server.url("/").url()));
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
private MockResponse newResponse() throws IOException {
  byte[] bytes = new byte[bodyByteCount];
  random.nextBytes(bytes);
  Buffer body = new Buffer().write(bytes);

  MockResponse result = new MockResponse();

  if (gzip) {
    Buffer gzipBody = new Buffer();
    GzipSink gzipSink = new GzipSink(gzipBody);
    gzipSink.write(body, body.size());
    gzipSink.close();
    body = gzipBody;
    result.addHeader("Content-Encoding: gzip");
  }

  if (chunked) {
    result.setChunkedBody(body, 1024);
  } else {
    result.setBody(body);
  }

  for (int i = 0; i < headerCount; i++) {
    result.addHeader(randomString(12), randomString(20));
  }

  return result;
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
@Override void setBody(MockResponse response, Buffer content, int chunkSize) {
  response.setBody(content);
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
@Override void setBody(MockResponse response, Buffer content, int chunkSize) {
  response.setBody(content);
  response.setSocketPolicy(SocketPolicy.DISCONNECT_AT_END);
  response.removeHeader("Content-Length");
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
public void enqueue(String body) {
    MockResponse response = new MockResponse();
    response.setBody(body);
    enqueue(response);
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
private void testRedirect(boolean temporary, String method) throws Exception {
  MockResponse response1 = new MockResponse()
      .setResponseCode(temporary ? HTTP_TEMP_REDIRECT : HTTP_PERM_REDIRECT)
      .addHeader("Location: /page2");
  if (!method.equals("HEAD")) {
    response1.setBody("This page has moved!");
  }
  server.enqueue(response1);
  server.enqueue(new MockResponse().setBody("Page 2"));

  connection = urlFactory.open(server.url("/page1").url());
  connection.setRequestMethod(method);
  byte[] requestBody = {'A', 'B', 'C', 'D'};
  if (method.equals("POST")) {
    connection.setDoOutput(true);
    OutputStream outputStream = connection.getOutputStream();
    outputStream.write(requestBody);
    outputStream.close();
  }

  String response = readAscii(connection.getInputStream(), Integer.MAX_VALUE);

  RecordedRequest page1 = server.takeRequest();
  assertEquals(method + " /page1 HTTP/1.1", page1.getRequestLine());

  if (method.equals("GET")) {
    assertEquals("Page 2", response);
  } else if (method.equals("HEAD")) {
    assertEquals("", response);
  } else {
    // Methods other than GET/HEAD shouldn't follow the redirect
    if (method.equals("POST")) {
      assertTrue(connection.getDoOutput());
      assertEquals("ABCD", page1.getBody().readUtf8());
    }
    assertEquals(1, server.getRequestCount());
    assertEquals("This page has moved!", response);
    return;
  }

  // GET/HEAD requests should have followed the redirect with the same method
  assertFalse(connection.getDoOutput());
  assertEquals(2, server.getRequestCount());
  RecordedRequest page2 = server.takeRequest();
  assertEquals(method + " /page2 HTTP/1.1", page2.getRequestLine());
}
 

import okhttp3.mockwebserver.MockResponse; //導入方法依賴的package包/類
@Override void setBody(MockResponse response, Buffer content, int chunkSize) {
  response.setBody(content);
  response.setSocketPolicy(DISCONNECT_AT_END);
  response.removeHeader("Content-Length");
}