def test__db_cnode_status_and_wrong_job_id(self): job = engine.prepare_job() expected = {} for node, status in [("O1", "up"), ("O2", "down"), ("O3", "down")]: ns = models.CNodeStats(oq_job=job, node=node, current_status=status) ns.save(using="job_superv") self.assertEqual(expected, monitor._db_cnode_status(-1))
def test__db_cnode_status(self): job = engine.prepare_job() expected = {} for node, status in [("N1", "up"), ("N2", "down"), ("N3", "down")]: ns = models.CNodeStats(oq_job=job, node=node, current_status=status) ns.save(using="job_superv") expected[node] = ns self.assertEqual(expected, monitor._db_cnode_status(job.id))
def test__db_cnode_status_and_two_jobs(self): job1 = engine.prepare_job() for node, status in [("P1", "up"), ("P2", "down"), ("P3", "down")]: ns = models.CNodeStats(oq_job=job1, node=node, current_status=status) ns.save(using="job_superv") job2 = engine.prepare_job() expected = {} for node, status in [("Q2", "down"), ("Q3", "down")]: ns = models.CNodeStats(oq_job=job2, node=node, current_status=status) ns.save(using="job_superv") expected[node] = ns self.assertEqual(expected, monitor._db_cnode_status(job2.id))