def test_one(self): """ Test one source. """ source = TopoNode() source.save(self.manager) target = TopoNode() target.save(self.manager) edge = TopoEdge(sources=source.id, targets=target.id) edge.save(self.manager) check = _all(event={}, ctx={}, vertice=target, manager=self.manager) self.assertTrue(check) source.state = Check.MINOR source.save(self.manager) check = _all(event={}, ctx={}, vertice=target, manager=self.manager) self.assertFalse(check)
def test_many(self): target = TopoNode() target.save(self.manager) count = 5 sources = [TopoNode() for i in range(count)] for source in sources: source.save(self.manager) edge = TopoEdge( sources=[source.id for source in sources], targets=target.id ) edge.save(self.manager) check = _all(event={}, ctx={}, vertice=target, manager=self.manager) self.assertTrue(check) sources[0].state = Check.MINOR sources[0].save(self.manager) check = _all(event={}, ctx={}, vertice=target, manager=self.manager) self.assertFalse(check)