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)