HttpURLConnection.getResponseCode не подбирает HTTP-код ответа 206

Question

У меня есть несколько автоматических тестов, которые используют HttpURLConnection для реализации RESTful API.

Часть моего кода (ниже) утверждает, что ответ возвращает определенный код ответа HTTP. Я ожидаю HTTP 206 Response, но getResponseCode всегда возвращает 200. Однако, если я ударил url напрямую с помощью curl, я получаю «HTTP/1.1 206 Partial Content», как и ожидалось.

URL requestURL = new URL(url); 
    HttpURLConnection connection = (HttpURLConnection) requestURL.openConnection(); 
    try { 
     connection.setRequestProperty("Connection", "close"); 
     connection.setReadTimeout(5000); 

     assertEquals("Request successfully handled", 
       expectedResponseCode, 
       connection.getResponseCode()); 

     InputStream input = connection.getInputStream(); 
     try { 
      return toString(input); 
     } finally { 
      input.close(); 
     } 
    } finally { 
     connection.disconnect(); 
    } 

Любые идеи о том, почему это происходит и как получить поведение, которое я хочу?

Answer 1

Таким образом, проблема заключалась в том, что я вызывал write() перед установкой кода ответа. Хотя это, похоже, работает при использовании с curl, оно не работает с модульными тестами, поскольку код подтверждает код возврата сразу после ввода.

код Проблемы:

String responseMessage = response.getMessage(); 
OutputStreamWriter out = new OutputStreamWriter(httpResponse.getOutputStream()); 
out.write(responseMessage); 
out.close(); 
httpResponse.setContentType("application/json"); 
httpResponse.setContentLength(responseMessage.length()); 
httpResponse.setStatus(response.getResponseCode()); 

Исправлен код:

httpResponse.setStatus(response.getResponseCode()); //Do this first! 
String responseMessage = response.getMessage(); 
OutputStreamWriter out = new OutputStreamWriter(httpResponse.getOutputStream()); 
out.write(responseMessage); 
out.close(); 
httpResponse.setContentType("application/json"); 
httpResponse.setContentLength(responseMessage.length()); 

Answer 2

Пробег: connection.getResponseMessage() Метод? Вероятно, он будет содержать ответ REST, где будет найден фактический код. Ответ на соединение может возникнуть, но фактический ответ операции можно найти в сообщении.

Answer 3

Код, который вы опубликовали, выполняет, как и ожидалось. Вот полный тестовый пример, чтобы доказать это:

import com.sun.net.httpserver.HttpExchange; 
import com.sun.net.httpserver.HttpHandler; 
import com.sun.net.httpserver.HttpServer; 
import org.junit.After; 
import org.junit.Before; 
import org.junit.Test; 

import java.io.IOException; 
import java.io.InputStream; 
import java.net.HttpURLConnection; 
import java.net.InetSocketAddress; 
import java.net.URL; 

import static org.hamcrest.CoreMatchers.equalTo; 
import static org.junit.Assert.assertEquals; 
import static org.junit.Assert.assertThat; 

public class Http206Test { 
    private HttpServer server; 

    @Before 
    public void setUp() throws Exception { 
     server = HttpServer.create(new InetSocketAddress(8080), 0); 
     server.createContext("/", new HttpHandler() { 
      public void handle(HttpExchange t) throws IOException { 
       t.sendResponseHeaders(206, 0); 
       t.getResponseBody().write("I'm a 206 response".getBytes()); 
       t.getResponseBody().close(); 
      } 
     }); 
     server.start(); 
    } 

    @After 
    public void tearDown() throws Exception { 
     server.stop(1); 
    } 

    @Test 
    public void httpUrlConnection206Response() throws Exception { 
     String body = getContent("http://localhost:8080", 206); 
     assertThat(body, equalTo("I'm a 206 response")); 
    } 

    @Test(expected = AssertionError.class) 
    public void httpUrlConnection205Response() throws Exception { 
     getContent("http://localhost:8080", 205); 
    } 

    public String getContent(String url, int expectedResponseCode) throws IOException { 
     URL requestURL = new URL(url); 
     HttpURLConnection connection = (HttpURLConnection) requestURL.openConnection(); 
     try { 
      connection.setRequestProperty("Connection", "close"); 
      connection.setReadTimeout(5000); 
      assertEquals("Request successfully handled", 
        expectedResponseCode, 
        connection.getResponseCode()); 
      InputStream input = connection.getInputStream(); 
      try { 
       return toString(input); 
      } finally { 
       input.close(); 
      } 
     } finally { 
      connection.disconnect(); 
     } 
    } 

    public String toString(InputStream stream) throws IOException { 
     int data; 
     StringBuilder builder = new StringBuilder(); 
     while ((data = stream.read()) != -1) { 
      builder.append((char) data); 
     } 
     return builder.toString(); 
    } 
} 

Это означает, что ваш тест правильный, и есть проблемы с вашим сервисом.