def test_delete_orphans(self):
"""
Test if graph vertices exist after deleting a graph.
"""
graph = Graph()
vertice = Vertice()
graph.add_elts(vertice)
graph.save(manager=self.manager)
vertice = self.manager.get_elts(vertice.id)
self.assertIsNotNone(vertice)
graph.delete(manager=self.manager)
vertice = self.manager.get_elts(vertice.id)
self.assertIsNone(vertice)
def test_orphans(self):
"""
Test get orphans method.
"""
# check if no orphans exist
orphans = self.manager.get_orphans()
self.assertFalse(orphans)
# generate self.count vertices and edges
[Vertice().save(self.manager) for i in range(self.count)]
[Edge().save(self.manager) for i in range(self.count)]
# check if previous vertices and edges are orphans
orphans = self.manager.get_orphans()
self.assertEqual(len(orphans), 2 * self.count)
# create a graph and add orphans to the graph
graph = Graph()
graph.add_elts(orphans)
graph.save(manager=self.manager)
# check if only the graph is an orphan
orphans = self.manager.get_orphans()
self.assertEqual(len(orphans), 1)
# delete the graph and check if vertices and edges became orphans
graph.delete(manager=self.manager, del_orphans=False)
orphans = self.manager.get_orphans()
self.assertEqual(len(orphans), 2 * self.count)
