ResponseParserTest.java

// Copyright (C) 2003-2009 by Object Mentor, Inc. All rights reserved.
// Released under the terms of the CPL Common Public License version 1.0.
package fitnesse.http;

import static org.junit.Assert.assertEquals;

import java.io.ByteArrayInputStream;
import java.io.InputStream;

import org.junit.Test;

public class ResponseParserTest {
  private String response;
  private InputStream input;

  @Test
  public void testParsing() throws Exception {
    response = "HTTP/1.1 200 OK\r\n" +
      "Content-Type: text/html\r\n" +
      "Content-Length: 12\r\n" +
      "Cache-Control: max-age=0\r\n" +
      "\r\n" +
      "some content";
    input = new ByteArrayInputStream(response.getBytes());

    ResponseParser parser = new ResponseParser(input);
    assertEquals(200, parser.getStatus());
    assertEquals("text/html", parser.getHeader("Content-Type"));
    assertEquals("some content", parser.getBody());
  }

  @Test
  public void testChunkedResponse() throws Exception {
    response = "HTTP/1.1 200 OK\r\n" +
      "Content-Type: text/html\r\n" +
      "Transfer-Encoding: chunked\r\n" +
      "\r\n" +
      "3\r\n" +
      "123\r\n" +
      "7\r\n" +
      "4567890\r\n" +
      "0\r\n" +
      "Tail-Header: TheEnd!\r\n";
    input = new ByteArrayInputStream(response.getBytes());

    ResponseParser parser = new ResponseParser(input);
    assertEquals(200, parser.getStatus());
    assertEquals("text/html", parser.getHeader("Content-Type"));
    assertEquals("1234567890", parser.getBody());
    assertEquals("TheEnd!", parser.getHeader("Tail-Header"));
  }
}