def test_multiple_jobs(self): self.prov = providers.stub.StubJobProvider() establish_chronology(self.progressive_chronology()) ret = dsub.wait_for_any_job(self.prov, ['job-1', 'job-2'], 1) self.assertEqual(ret, set(['job-2']))
def test_already_succeeded(self): prov = providers.stub.StubJobProvider() prov.set_operations([{'job-id': 'myjob', 'status': 'SUCCESS'}]) establish_chronology(nothing_happens()) ret = dsub.wait_for_any_job(prov, ['myjob'], 1) self.assertEqual(ret, set([]))