public void testNodeManipulations() {
NodeDAO nodeDao = (NodeDAO) context.getBean("nodeDAO");
NodeImage image = getTestImage();
Node life = (Node) nodeDao.findNodesExactlyNamed("Life on Earth").get(0);
Node viruses = (Node) nodeDao.findNodesExactlyNamed("Viruses").get(0);
Node eukaryotes = (Node) nodeDao.findNodesExactlyNamed("Eukaryotes").get(0);
image.addToNodesSet(life);
dao.saveImage(image);
NodeImage foundImage = dao.getImageWithId(image.getId());
assertTrue(foundImage.getNodesSet().size() == 1);
Node firstNode = (Node) new Vector(foundImage.getNodesSet()).get(0);
assertEquals(firstNode.getId(), life.getId());
image.addToNodesSet(viruses);
dao.saveImage(image);
foundImage = dao.getImageWithId(image.getId());
assertTrue(foundImage.getNodesSet().size() == 2);
Vector nodes = new Vector(foundImage.getNodesSet());
firstNode = (Node) nodes.get(0);
Node secondNode = (Node) nodes.get(1);
assertTrue(firstNode.getId() == life.getId() || firstNode.getId() == viruses.getId());
assertTrue(secondNode.getId() == life.getId() || secondNode.getId() == viruses.getId());
image.removeFromNodesSet(life);
image.removeFromNodesSet(viruses);
image.addToNodesSet(eukaryotes);
dao.saveImage(image);
foundImage = dao.getImageWithId(image.getId());
assertTrue(foundImage.getNodesSet().size() == 1);
nodes = new Vector(foundImage.getNodesSet());
firstNode = (Node) nodes.get(0);
assertTrue(firstNode.getId() == eukaryotes.getId());
dao.deleteImage(image);
}
public void testDeleteImage() {
NodeImage img = new NodeImage();
img.setLocation("shouldbedeleted.jpg");
img.setContributor(david);
dao.addImage(img, david, false);
int id = img.getId();
dao.deleteImage(img);
NodeImage shouldBeNullImg = dao.getImageWithId(id);
assertNull(shouldBeNullImg);
}
