Example usage for org.springframework.mock.web MockHttpServletResponse getHeader

List of usage examples for org.springframework.mock.web MockHttpServletResponse getHeader

Introduction

In this page you can find the example usage for org.springframework.mock.web MockHttpServletResponse getHeader.

Prototype

@Override
@Nullable
public String getHeader(String name) 

Source Link

Document

Return the primary value for the given header as a String, if any.

Usage

From source file:org.opennms.web.rest.AbstractSpringJerseyRestTestCase.java

protected void putXmlObject(final JAXBContext context, final String url, final int expectedStatus,
        final Object object, final String expectedUrlSuffix) throws Exception {
    final ByteArrayOutputStream out = new ByteArrayOutputStream();
    final Marshaller marshaller = context.createMarshaller();
    marshaller.marshal(object, out);//  w w  w .j av a 2s.co m
    final byte[] content = out.toByteArray();

    final MockHttpServletRequest request = createRequest(PUT, url);
    request.setContentType(MediaType.APPLICATION_XML);
    request.setContent(content);
    final MockHttpServletResponse response = createResponse();
    dispatch(request, response);
    assertEquals(expectedStatus, response.getStatus());

    final String location = response.getHeader("Location").toString();
    assertTrue("location '" + location + "' should end with '" + expectedUrlSuffix + "'",
            location.endsWith(expectedUrlSuffix));
}

From source file:org.osaf.cosmo.cmp.CmpGetTest.java

/**
 *///from  ww w .j a  v  a 2  s .c  om
public void testGetAccount() throws Exception {
    User u1 = testHelper.makeDummyUser();
    u1 = userService.createUser(u1);

    logInUser(u1);

    MockHttpServletRequest request = createMockRequest("GET", "/account");
    MockHttpServletResponse response = new MockHttpServletResponse();
    servlet.service(request, response);

    assertEquals(response.getStatus(), MockHttpServletResponse.SC_OK);
    assertNotNull(response.getHeader("ETag"));

    CmpUser user = createUserFromXml(readXmlResponse(response));
    assertNotNull("user null", user);
    assertNotNull("user has no username", user.getUsername());
    assertEquals("usernames don't match", user.getUsername(), u1.getUsername());
    assertNotNull("user has no first name", user.getFirstName());
    assertEquals("first names don't match", user.getFirstName(), u1.getFirstName());
    assertNotNull("user has no last name", user.getLastName());
    assertEquals("last names don't match", user.getLastName(), u1.getLastName());
    assertNotNull("user has no email", user.getEmail());
    assertEquals("emails don't match", user.getEmail(), u1.getEmail());
    assertNotNull("user has no url", user.getUrl());
    assertNotNull("user has no homedir url", user.getHomedirUrl());
}

From source file:org.osaf.cosmo.cmp.CmpGetTest.java

/**
 *//*  w w w .j a  v a  2s .co  m*/
public void testGetUser() throws Exception {
    User u1 = testHelper.makeDummyUser();
    u1 = userService.createUser(u1);

    MockHttpServletRequest request = createMockRequest("GET", "/user/" + u1.getUsername());
    MockHttpServletResponse response = new MockHttpServletResponse();
    servlet.service(request, response);

    assertEquals(response.getStatus(), MockHttpServletResponse.SC_OK);
    assertNotNull(response.getHeader("ETag"));

    CmpUser user = createUserFromXml(readXmlResponse(response));
    assertNotNull("user null", user);
    assertNotNull("user has no username", user.getUsername());
    assertEquals("usernames don't match", user.getUsername(), u1.getUsername());
    assertNotNull("user has no first name", user.getFirstName());
    assertEquals("first names don't match", user.getFirstName(), u1.getFirstName());
    assertNotNull("user has no last name", user.getLastName());
    assertEquals("last names don't match", user.getLastName(), u1.getLastName());
    assertNotNull("user has no email", user.getEmail());
    assertEquals("emails don't match", user.getEmail(), u1.getEmail());
    assertNotNull("user has no creation date", user.getCreationDate());
    assertEquals("creation dates don't match", DateUtil.formatRfc3339Date(user.getCreationDate()),
            DateUtil.formatRfc3339Date(u1.getCreationDate()));
    assertNotNull("user has no modification date", user.getModifiedDate());
    assertEquals("modification dates don't match", DateUtil.formatRfc3339Date(user.getModifiedDate()),
            DateUtil.formatRfc3339Date(u1.getModifiedDate()));
    assertNotNull("user has no administrator boolean", user.getAdmin());
    assertEquals("administrator booleans don't match", user.getAdmin(), u1.getAdmin());
    assertNotNull("user has no url", user.getUrl());
    assertNotNull("user has no homedir url", user.getHomedirUrl());
}

From source file:org.osaf.cosmo.cmp.CmpGetTest.java

/**
 *///  w w w . ja v  a2s. c o  m
public void testGetOverlord() throws Exception {
    MockHttpServletRequest request = createMockRequest("GET", "/user/" + User.USERNAME_OVERLORD);
    MockHttpServletResponse response = new MockHttpServletResponse();
    servlet.service(request, response);

    assertEquals(response.getStatus(), MockHttpServletResponse.SC_OK);
    assertNotNull(response.getHeader("ETag"));

    CmpUser overlord = createUserFromXml(readXmlResponse(response));
    assertNotNull("overlord null", overlord);
    assertNotNull("overlord has no username", overlord.getUsername());
    assertEquals("usernames don't match", overlord.getUsername(), User.USERNAME_OVERLORD);
    assertNotNull("overlord has no first name", overlord.getFirstName());
    assertNotNull("overlord has no last name", overlord.getLastName());
    assertNotNull("overlord has no email", overlord.getEmail());
    assertNotNull("overlord has no url", overlord.getUrl());
    assertNull("overlord has a homedir url", overlord.getHomedirUrl());
}

From source file:org.osaf.cosmo.cmp.CmpPutTest.java

/**
 */// w  w  w  .ja v a  2  s. c om
public void testSignup() throws Exception {
    User u1 = testHelper.makeDummyUser();
    u1.setAdmin(Boolean.TRUE);
    u1.setLocked(Boolean.TRUE);

    MockHttpServletRequest request = createMockRequest("PUT", "/signup");
    sendXmlRequest(request, new UserContent(u1));

    MockHttpServletResponse response = new MockHttpServletResponse();
    servlet.service(request, response);

    assertEquals("incorrect status", MockHttpServletResponse.SC_CREATED, response.getStatus());
    assertNotNull("null Content-Location", response.getHeader("Content-Location"));
    assertNotNull("null ETag", response.getHeader("ETag"));

    User u2 = userService.getUser(u1.getUsername());
    assertFalse("User signed up as admin", u2.getAdmin());
    assertFalse("User signed up as locked", u2.isLocked());
}

From source file:org.osaf.cosmo.cmp.CmpPutTest.java

/**
 *//*w  w  w.j a  v a  2  s .co m*/
public void testAccountUpdate() throws Exception {
    User u1 = testHelper.makeDummyUser();
    u1 = userService.createUser(u1);

    logInUser(u1);

    // make a new user with all new values for updating the old
    // user, except use the same username cos end users cannot
    // change their own usernames
    User cmpUser = testHelper.makeDummyUser();
    cmpUser.setUsername(u1.getUsername());

    MockHttpServletRequest request = createMockRequest("PUT", "/account");
    sendXmlRequest(request, new UserContent(cmpUser));

    MockHttpServletResponse response = new MockHttpServletResponse();
    servlet.service(request, response);

    assertEquals("incorrect status", MockHttpServletResponse.SC_NO_CONTENT, response.getStatus());
    assertNotNull("null ETag", response.getHeader("ETag"));

    User storedUser = userService.getUser(cmpUser.getUsername());
    assertNotNull("null updated user", storedUser);
    assertNotNull("null username", storedUser.getUsername());
    assertNotNull("null password", storedUser.getPassword());
    // don't bother to check the password of the stored user
    // against that of the cmp user since the stored user's
    // password is encrypted
    assertNotNull("null firstName", storedUser.getFirstName());
    assertEquals("updated firstName doesn't match", storedUser.getFirstName(), cmpUser.getFirstName());
    assertNotNull("null lastName", storedUser.getLastName());
    assertEquals("updated lastName doesn't match", storedUser.getLastName(), cmpUser.getLastName());
    assertNotNull("null email", storedUser.getEmail());
    assertEquals("updated email doesn't match", storedUser.getEmail(), cmpUser.getEmail());

    // Make sure users can't make themselves administrators

    cmpUser.setAdmin(true);

    request = createMockRequest("PUT", "/account");
    sendXmlRequest(request, new UserContent(cmpUser));

    response = new MockHttpServletResponse();
    servlet.service(request, response);

    this.assertEquals("user was allowed to make himself admin", MockHttpServletResponse.SC_FORBIDDEN,
            response.getStatus());

}

From source file:org.osaf.cosmo.cmp.CmpPutTest.java

/**
 *///w w w. j a v a 2 s  .  com
public void testUserCreate() throws Exception {
    User cmpUser = testHelper.makeDummyUser();
    // ensure that the user doesn't already exist
    User test = userService.getUser(cmpUser.getUsername());
    assertNull(test);

    MockHttpServletRequest request = createMockRequest("PUT", "/user/" + cmpUser.getUsername());
    sendXmlRequest(request, new UserContent(cmpUser));

    MockHttpServletResponse response = new MockHttpServletResponse();
    servlet.service(request, response);

    assertEquals("incorrect status", MockHttpServletResponse.SC_CREATED, response.getStatus());
    assertNotNull("null ETag", response.getHeader("ETag"));

    User storedUser = userService.getUser(cmpUser.getUsername());
    assertNotNull("null updated user", storedUser);
    assertNotNull("null username", storedUser.getUsername());
    assertNotNull("null password", storedUser.getPassword());
    // don't bother to check the password of the stored user
    // against that of the cmp user since the stored user's
    // password is encrypted
    assertNotNull("null firstName", storedUser.getFirstName());
    assertEquals("updated firstName doesn't match", storedUser.getFirstName(), cmpUser.getFirstName());
    assertNotNull("null lastName", storedUser.getLastName());
    assertEquals("updated lastName doesn't match", storedUser.getLastName(), cmpUser.getLastName());
    assertNotNull("null email", storedUser.getEmail());
    assertEquals("updated email doesn't match", storedUser.getEmail(), cmpUser.getEmail());
}

From source file:org.osaf.cosmo.cmp.CmpPutTest.java

/**
 *//*from www  . ja v  a 2  s. c om*/
public void testUserUpdate() throws Exception {
    User u1 = testHelper.makeDummyUser();
    u1 = userService.createUser(u1);
    String originalUsername = u1.getUsername();

    // make a new user with all new values for updating the old
    // user, including username
    User cmpUser = testHelper.makeDummyUser();

    MockHttpServletRequest request = createMockRequest("PUT", "/user/" + u1.getUsername());
    sendXmlRequest(request, new UserContent(cmpUser));

    MockHttpServletResponse response = new MockHttpServletResponse();
    servlet.service(request, response);

    assertEquals("incorrect status", MockHttpServletResponse.SC_NO_CONTENT, response.getStatus());
    assertNotNull("null Content-Location", response.getHeader("Content-Location"));
    assertNotNull("null ETag", response.getHeader("ETag"));

    User storedUser = userService.getUser(cmpUser.getUsername());
    assertNotNull("null updated user", storedUser);
    assertNotNull("null username", storedUser.getUsername());
    assertEquals("updated username doesn't match", storedUser.getUsername(), cmpUser.getUsername());
    assertNotNull("null password", storedUser.getPassword());
    // don't bother to check the password of the stored user
    // against that of the cmp user since the stored user's
    // password is encrypted
    assertNotNull("null firstName", storedUser.getFirstName());
    assertEquals("updated firstName doesn't match", storedUser.getFirstName(), cmpUser.getFirstName());
    assertNotNull("null lastName", storedUser.getLastName());
    assertEquals("updated lastName doesn't match", storedUser.getLastName(), cmpUser.getLastName());
    assertNotNull("null email", storedUser.getEmail());
    assertEquals("updated email doesn't match", storedUser.getEmail(), cmpUser.getEmail());

    User test = userService.getUser(originalUsername);
    assertNull(test);
}

From source file:org.springframework.security.web.authentication.www.DigestAuthenticationEntryPointTests.java

@Test
public void testNormalOperation() throws Exception {
    DigestAuthenticationEntryPoint ep = new DigestAuthenticationEntryPoint();
    ep.setRealmName("hello");
    ep.setKey("key");

    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setRequestURI("/some_path");

    MockHttpServletResponse response = new MockHttpServletResponse();

    ep.afterPropertiesSet();//ww w .  j  a  va 2 s  .  c  o m

    ep.commence(request, response, new DisabledException("foobar"));

    // Check response is properly formed
    assertThat(response.getStatus()).isEqualTo(401);
    assertThat(response.getHeader("WWW-Authenticate").toString()).startsWith("Digest ");

    // Break up response header
    String header = response.getHeader("WWW-Authenticate").toString().substring(7);
    String[] headerEntries = StringUtils.commaDelimitedListToStringArray(header);
    Map<String, String> headerMap = DigestAuthUtils.splitEachArrayElementAndCreateMap(headerEntries, "=", "\"");

    assertThat(headerMap.get("realm")).isEqualTo("hello");
    assertThat(headerMap.get("qop")).isEqualTo("auth");
    assertThat(headerMap.get("stale")).isNull();

    checkNonceValid(headerMap.get("nonce"));
}

From source file:org.springframework.security.web.authentication.www.DigestAuthenticationEntryPointTests.java

@Test
public void testOperationIfDueToStaleNonce() throws Exception {
    DigestAuthenticationEntryPoint ep = new DigestAuthenticationEntryPoint();
    ep.setRealmName("hello");
    ep.setKey("key");

    MockHttpServletRequest request = new MockHttpServletRequest();
    request.setRequestURI("/some_path");

    MockHttpServletResponse response = new MockHttpServletResponse();

    ep.afterPropertiesSet();/*from w w  w .j a  v  a  2s .com*/

    ep.commence(request, response, new NonceExpiredException("expired nonce"));

    // Check response is properly formed
    assertThat(response.getStatus()).isEqualTo(401);
    assertThat(response.getHeader("WWW-Authenticate").toString()).startsWith("Digest ");

    // Break up response header
    String header = response.getHeader("WWW-Authenticate").toString().substring(7);
    String[] headerEntries = StringUtils.commaDelimitedListToStringArray(header);
    Map<String, String> headerMap = DigestAuthUtils.splitEachArrayElementAndCreateMap(headerEntries, "=", "\"");

    assertThat(headerMap.get("realm")).isEqualTo("hello");
    assertThat(headerMap.get("qop")).isEqualTo("auth");
    assertThat(headerMap.get("stale")).isEqualTo("true");

    checkNonceValid(headerMap.get("nonce"));
}