def test_jobs(self): jt = JobsTable() jobs_send = [ JobRef( 'id_%d' % i , time.time(), Status(Status.BUILDING), 'owner') for i in range(0, 5) ] for j in jobs_send: jt.put(j) jobs_recieved = jt.jobs() self.assertEqual( jobs_send, jobs_recieved)
def test_put_get_pop(self): jt = JobsTable() j1 = JobRef( 'id_1', time.time(), Status(Status.BUILDING), 'owner') jt.put(j1) self.assertEqual(len(jt.jobs()), 1) j2 = jt.get(j1.id) self.assertEqual(j1, j2) self.assertEqual(len(jt.jobs()), 1) j3 = jt.pop(j1.id) self.assertEqual(j1, j3) self.assertEqual(len(jt.jobs()), 0)