@Test
 public void setStatusCode() throws Exception {
   response.setStatusCode(HttpStatus.NOT_FOUND);
   assertEquals("Invalid status code", 404, mockResponse.getStatus());
 }