SimpleResponseTest.java :  » Wiki-Engine » fitnesse » fitnesse » http » Java Open Source

// Copyright (C) 2003,2004,2005 by Object Mentor, Inc. All rights reserved.
// Released under the terms of the GNU General Public License version 2 or later.
package fitnesse.http;

import fitnesse.testutil.RegexTest;
import java.net.Socket;

public class SimpleResponseTest extends RegexTest implements ResponseSender
{
  private String text;
  private boolean closed = false;

  public void send(byte[] bytes) throws Exception
  {
    text = new String(bytes, "UTF-8");
  }

  public void close()
  {
    closed = true;
  }

  public Socket getSocket() throws Exception
  {
    return null;
  }

  public void setUp() throws Exception
  {
  }

  public void tearDown() throws Exception
  {
  }

  public void testSimpleResponse() throws Exception
  {
    SimpleResponse response = new SimpleResponse();
    response.setContent("some content");
    response.readyToSend(this);
    assertTrue(text.startsWith("HTTP/1.1 200 OK\r\n"));
    assertHasRegexp("Content-Length: 12", text);
    assertHasRegexp("Content-Type: text/html", text);
    assertTrue(text.endsWith("some content"));
    assertTrue(closed);
  }

  public void testPageNotFound() throws Exception
  {
    SimpleResponse response = new SimpleResponse(404);
    response.readyToSend(this);
    assertHasRegexp("404 Not Found", text);
  }

  public void testRedirect() throws Exception
  {
    SimpleResponse response = new SimpleResponse();
    response.redirect("some url");
    response.readyToSend(this);
    assertEquals(303, response.getStatus());
    assertHasRegexp("Location: some url\r\n", text);
  }

  public void testUnicodeCharacters() throws Exception
  {
    SimpleResponse response = new SimpleResponse();
    response.setContent("\uba80\uba81\uba82\uba83");
    response.readyToSend(this);

    assertSubString("\uba80\uba81\uba82\uba83", text);
  }
}