public void testStateCookies() throws Exception {
final String resourceName = "something/interesting";
ServletRunner sr = new ServletRunner();
sr.registerServlet(resourceName, StatefulServlet.class.getName());
WebRequest request = new PostMethodWebRequest("http://localhost/" + resourceName);
request.setParameter("color", "red");
WebResponse response = sr.getResponse(request);
assertNotNull("No response received", response);
assertEquals("Returned cookie count", 1, response.getNewCookieNames().length);
}
public void testNoInitialState() throws Exception {
final String resourceName = "something/interesting";
ServletRunner sr = new ServletRunner();
sr.registerServlet(resourceName, StatefulServlet.class.getName());
WebRequest request = new GetMethodWebRequest("http://localhost/" + resourceName);
WebResponse response = sr.getResponse(request);
assertNotNull("No response received", response);
assertEquals("content type", "text/plain", response.getContentType());
assertEquals("requested resource", "No session found", response.getText());
assertEquals("Returned cookie count", 0, response.getNewCookieNames().length);
}
public void testSessionPreloading() throws Exception {
final String resourceName1 = "something/interesting/start";
final String resourceName2 = "something/continue";
ServletRunner sr = new ServletRunner();
sr.registerServlet(resourceName1, StatefulServlet.class.getName());
sr.registerServlet(resourceName2, StatefulServlet.class.getName());
ServletUnitClient wc = sr.newClient();
wc.getSession(true).setAttribute("color", "green");
WebRequest request = new GetMethodWebRequest("http://localhost/" + resourceName2);
WebResponse response = wc.getResponse(request);
assertNotNull("No response received", response);
assertEquals("content type", "text/plain", response.getContentType());
assertEquals("requested resource", "You posted green", response.getText());
assertEquals("Returned cookie count", 0, response.getNewCookieNames().length);
}
public void testInvocationCompletion() throws Exception {
final String resourceName = "something/interesting";
ServletRunner sr = new ServletRunner();
sr.registerServlet(resourceName, StatefulServlet.class.getName());
ServletUnitClient suc = sr.newClient();
WebRequest request = new PostMethodWebRequest("http://localhost/" + resourceName);
request.setParameter("color", "red");
InvocationContext ic = suc.newInvocation(request);
StatefulServlet ss = (StatefulServlet) ic.getServlet();
ss.setColor(ic.getRequest(), "blue");
ss.writeSelectMessage("blue", ic.getResponse().getWriter());
WebResponse response = ic.getServletResponse();
assertEquals("requested resource", "You selected blue", response.getText());
assertEquals("Returned cookie count", 1, response.getNewCookieNames().length);
}
public void testInvocationContextUpdate() throws Exception {
final String resourceName = "something/interesting";
ServletRunner sr = new ServletRunner();
sr.registerServlet(resourceName, StatefulServlet.class.getName());
ServletUnitClient suc = sr.newClient();
WebRequest request = new PostMethodWebRequest("http://localhost/" + resourceName);
request.setParameter("color", "red");
InvocationContext ic = suc.newInvocation(request);
StatefulServlet ss = (StatefulServlet) ic.getServlet();
ss.setColor(ic.getRequest(), "blue");
suc.getResponse(ic);
WebResponse response = suc.getResponse("http://localhost/" + resourceName);
assertNotNull("No response received", response);
assertEquals("content type", "text/plain", response.getContentType());
assertEquals("requested resource", "You posted blue", response.getText());
assertEquals("Returned cookie count", 0, response.getNewCookieNames().length);
}