/**
* Tests adding a project through the API
*
* @throws InvalidXmlException
* @throws BizInternalException
* @throws BizPolicyException
* @throws IOException
*/
@Test
public void testAddProject()
throws InvalidXmlException, BizInternalException, BizPolicyException, IOException {
Project newProject = new Project();
newProject.setName("Test Project To Add");
newProject.setDescription("adding this project");
List<String> numbers = new ArrayList<String>();
numbers.add("1");
numbers.add("2");
newProject.setNumbers(numbers);
newProject.setFundingEntity("The Fed");
newProject.setStartDate(new DateTime(2012, 5, 4, 0, 0));
newProject.setEndDate(new DateTime(2013, 12, 23, 0, 0));
newProject.addPi(admin.getId());
final String mimeType = "application/xml";
final MockHttpServletRequest mockReq = newMockRequest("POST", "/project", "test.org", 80);
MockHttpServletResponse resp = new MockHttpServletResponse();
ByteArrayOutputStream sink = new ByteArrayOutputStream();
businessObjectBuilder.buildProject(newProject, sink);
controller.setAuthenticatedUser(admin);
controller.handleProjectPostRequest(mimeType, sink.toByteArray(), mockReq, resp);
assertNotNull(resp);
assertEquals(201, resp.getStatus());
Bop bop =
businessObjectBuilder.buildBusinessObjectPackage(
new ByteArrayInputStream(resp.getContentAsByteArray()));
assertNotNull(bop);
Set<Project> projects = bop.getProjects();
assertNotNull(projects);
assertEquals(1, projects.size());
Project returnedProject = projects.iterator().next();
// Have to set the original project id to the id set by the biz service.
newProject.setId(returnedProject.getId());
assertEquals(newProject, returnedProject);
}
/**
* Tests that admin can be added and removed from a project
*
* @throws IOException
* @throws BizPolicyException
* @throws ProjectServiceException
* @throws InvalidXmlException
* @throws BizInternalException
*/
@Test
public void testUpdateProjectAdmin()
throws InvalidXmlException, ProjectServiceException, BizPolicyException, IOException,
BizInternalException {
// Test adding an admin to the project.
Project newProject = new Project(projectOne);
newProject.addPi(user.getId());
final String mimeType = "application/xml";
final MockHttpServletRequest mockReq =
newMockRequest("PUT", projectOne.getId(), "test.org", 80);
MockHttpServletResponse resp = new MockHttpServletResponse();
ByteArrayOutputStream sink = new ByteArrayOutputStream();
businessObjectBuilder.buildProject(newProject, sink);
controller.setAuthenticatedUser(admin);
controller.handleUpdateProjectRequest(
newProject.getId(), mimeType, sink.toByteArray(), mockReq, resp);
assertNotNull(resp);
assertEquals(resp.getErrorMessage(), 200, resp.getStatus());
Bop bop =
businessObjectBuilder.buildBusinessObjectPackage(
new ByteArrayInputStream(resp.getContentAsByteArray()));
assertNotNull(bop);
Set<Project> projects = bop.getProjects();
assertNotNull(projects);
assertEquals(1, projects.size());
Project returnedProject = projects.iterator().next();
assertNotNull(returnedProject);
assertEquals(newProject.getPis().size(), returnedProject.getPis().size());
// Test removing an admin from the project.
newProject.removePi(user.getId());
resp = new MockHttpServletResponse();
sink = new ByteArrayOutputStream();
businessObjectBuilder.buildProject(newProject, sink);
controller.setAuthenticatedUser(admin);
controller.handleUpdateProjectRequest(
newProject.getId(), mimeType, sink.toByteArray(), mockReq, resp);
assertNotNull(resp);
bop =
businessObjectBuilder.buildBusinessObjectPackage(
new ByteArrayInputStream(resp.getContentAsByteArray()));
assertNotNull(bop);
projects = bop.getProjects();
assertNotNull(projects);
assertEquals(1, projects.size());
returnedProject = projects.iterator().next();
assertNotNull(returnedProject);
assertEquals(newProject.getPis().size(), returnedProject.getPis().size());
}