ResponseTest.java

package org.jooby;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

import java.io.File;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.LinkedList;
import java.util.Optional;

import org.jooby.Cookie.Definition;
import org.jooby.Route.After;
import org.jooby.Route.Complete;
import org.junit.Test;

public class ResponseTest {
  public static class ResponseMock implements Response {

    @Override
    public void download(final String filename, final InputStream stream) throws Exception {
      throw new UnsupportedOperationException();
    }

    @Override
    public void download(final String filename, final String location) throws Exception {
    }

    @Override
    public Response cookie(final Definition cookie) {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response cookie(final Cookie cookie) {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response clearCookie(final String name) {
      throw new UnsupportedOperationException();
    }

    @Override
    public Mutant header(final String name) {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response header(final String name, final Object value) {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response header(final String name, final Iterable<Object> values) {
      throw new UnsupportedOperationException();
    }

    @Override
    public Charset charset() {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response charset(final Charset charset) {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response length(final long length) {
      throw new UnsupportedOperationException();
    }

    @Override
    public void end() {
      throw new UnsupportedOperationException();
    }

    @Override
    public Optional<MediaType> type() {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response type(final MediaType type) {
      throw new UnsupportedOperationException();
    }

    @Override
    public void send(final Result result) throws Exception {
      throw new UnsupportedOperationException();
    }

    @Override
    public void redirect(final Status status, final String location) throws Exception {
      throw new UnsupportedOperationException();
    }

    @Override
    public Optional<Status> status() {
      throw new UnsupportedOperationException();
    }

    @Override
    public Response status(final Status status) {
      throw new UnsupportedOperationException();
    }

    @Override
    public boolean committed() {
      throw new UnsupportedOperationException();
    }

    @Override
    public void after(final After handler) {
      throw new UnsupportedOperationException();
    }

     @Override
    public void complete(final Complete handler) {
       throw new UnsupportedOperationException();
    }
  }

  @Test
  public void type() {
    LinkedList<MediaType> types = new LinkedList<>();
    new ResponseMock() {
      @Override
      public Response type(final MediaType type) {
        types.add(type);
        return this;
      }
    }.type("json");
    assertEquals(MediaType.json, types.getFirst());
  }

  @Test
  public void sendObject() throws Throwable {
    Object data = new Object();
    LinkedList<Object> dataList = new LinkedList<>();
    new ResponseMock() {
      @Override
      public void send(final Result result) {
        assertNotNull(result);
        assertEquals(Status.OK, result.status().get());
        assertEquals(MediaType.json, result.type().get());
        dataList.add(result.ifGet().get());
      }

      @Override
      public Optional<Status> status() {
        return Optional.of(Status.OK);
      }

      @Override
      public Optional<MediaType> type() {
        return Optional.of(MediaType.json);
      }

    }.send(data);

    assertEquals(data, dataList.getFirst());
  }

  @Test
  public void sendBody() throws Throwable {
    Object data = Results.noContent();
    LinkedList<Object> dataList = new LinkedList<>();
    new ResponseMock() {
      @Override
      public void send(final Result body) throws Exception {
        assertNotNull(body);
        dataList.add(body);
      }

      @Override
      public Optional<Status> status() {
        return Optional.of(Status.OK);
      }

      @Override
      public Optional<MediaType> type() {
        return Optional.of(MediaType.json);
      }

    }.send(data);

    assertEquals(data, dataList.getFirst());
  }

  @Test
  public void redirect() throws Throwable {
    LinkedList<Object> dataList = new LinkedList<>();
    new ResponseMock() {
      @Override
      public void redirect(final Status status, final String location) throws Exception {
        assertEquals(Status.FOUND, status);
        dataList.add(location);
      }
    }.redirect("/red");
    assertEquals("/red", dataList.getFirst());
  }

  @Test
  public void statusCode() throws Exception {
    LinkedList<Object> dataList = new LinkedList<>();
    new ResponseMock() {
      @Override
      public Response status(final Status status) {
        dataList.add(status);
        return this;
      }
    }.status(200);
    assertEquals(Status.OK, dataList.getFirst());
  }

  @Test
  public void downloadFileWithName() throws Throwable {
    LinkedList<Object> dataList = new LinkedList<>();
    File resource = file("src/test/resources/org/jooby/ResponseTest.js");
    new ResponseMock() {
      @Override
      public void download(final String filename, final InputStream stream) throws Exception {
        assertNotNull(stream);
        stream.close();
        dataList.add(filename);
      }

      @Override
      public Response length(final long length) {
        dataList.add(length);
        return this;
      }
    }.download("alias.js", resource);
    assertEquals("[20, alias.js]", dataList.toString());
  }

  @Test
  public void cookieWithNameAndValue() throws Exception {
    LinkedList<Cookie.Definition> dataList = new LinkedList<>();
    new ResponseMock() {
      @Override
      public Response cookie(final Cookie.Definition cookie) {
        dataList.add(cookie);
        return this;
      }
    }.cookie("name", "value");

    assertEquals("name", dataList.getFirst().name().get());
    assertEquals("value", dataList.getFirst().value().get());
  }

  /**
   * Attempt to load a file from multiple location. required by unit and integration tests.
   *
   * @param location
   * @return
   */
  private File file(final String location) {
    for (String candidate : new String[]{location, "jooby/" + location, "../jooby/" + location }) {
      File file = new File(candidate);
      if (file.exists()) {
        return file;
      }
    }
    return file(location);
  }

}