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); }