public

anonymous / AbstractControllerTest.java

Created 2012-08-13

package controllers;
 
import static org.fest.assertions.Assertions.assertThat;
import static play.test.Helpers.callAction;
import static play.test.Helpers.fakeRequest;
import static play.test.Helpers.status;
 
import java.util.HashMap;
import java.util.Map;
 
import models.User;
import play.mvc.HandlerRef;
import play.mvc.Http.Cookie;
import play.mvc.Http.Session;
import play.mvc.Result;
import play.test.FakeRequest;
import play.test.Helpers;
import test.FakeApplicationAwareTest;
 
public abstract class AbstractControllerTest extends FakeApplicationAwareTest {
 
  public static class Given {
 
    private Result result;
    private HandlerRef action;
    private User loggedUser;
    private Map<String, String> formData = new HashMap<String, String>();
 
    public Given(HandlerRef action) {
      this.action = action;
    }
 
    public static Given action(HandlerRef action) {
      return new Given(action);
    }
 
    public Given withUser(User user) {
      this.loggedUser = user;
      return this;
    }
 
    public Given withInput(String name, String value) {
      this.formData.put(name, value);
      return this;
    }
 
    public Given withInput(String name, Long value) {
      this.formData.put(name, String.valueOf(value));
      return this;
    }
 
    public Given withInput(String name, Integer value) {
      this.formData.put(name, String.valueOf(value));
      return this;
    }
 
    public Given expectStatus(int expectedStatus) {
      assertThat(status(result)).isEqualTo(expectedStatus);
      return this;
    }
 
    public Given expectContentType(String contentType) {
      assertThat(Helpers.contentType(result)).isEqualTo(contentType);
      return this;
    }
 
    public Given expectTextPresent(String text) {
      assertThat(Helpers.contentAsString(this.result)).contains(text);
      return this;
    }
 
    public Given expectTextNotPresent(String text) {
      assertThat(Helpers.contentAsString(this.result)).doesNotContain(text);
      return this;
    }
 
    public Given expectRedirectTo(String location) {
      assertThat(Helpers.header("Location", this.result)).isEqualTo(location);
      return this;
    }
 
    public Given expectedCookie(String expectedValue) {
      Cookie cookie = Helpers.cookie("PLAY_SESSION", this.result);
      assertThat(cookie.value()).isEqualTo(expectedValue);
      return this;
    }
 
    public Given expectedSessionValue(String key, String value) {
      Session session = Helpers.session(this.result);
      assertThat(session.get(key)).isEqualTo(value);
      return this;
    }
 
    public Given execute() {
      FakeRequest request = fakeRequest();
      request.withFormUrlEncodedBody(this.formData);
 
      if(loggedUser != null) {
        request.withSession(Secured.SESSION_KEY, String.valueOf(loggedUser.id));
      }
 
      this.result = callAction(this.action, request);
 
      return this;
    }
  }
 
}