Show More
@@ -449,7 +449,32 b' class TestClient(ClusterTestCase):' | |||||
449 | self.client.purge_local_results(res[-1]) |
|
449 | self.client.purge_local_results(res[-1]) | |
450 | self.assertEqual(len(self.client.results),before-len(res[-1]), msg="Not removed from results") |
|
450 | self.assertEqual(len(self.client.results),before-len(res[-1]), msg="Not removed from results") | |
451 | self.assertEqual(len(self.client.metadata),before-len(res[-1]), msg="Not removed from metadata") |
|
451 | self.assertEqual(len(self.client.metadata),before-len(res[-1]), msg="Not removed from metadata") | |
452 |
|
|
452 | ||
|
453 | def test_purge_local_results_outstanding(self): | |||
|
454 | v = self.client[-1] | |||
|
455 | ar = v.apply_async(lambda : 1) | |||
|
456 | msg_id = ar.msg_ids[0] | |||
|
457 | ar.get() | |||
|
458 | self._wait_for_idle() | |||
|
459 | ar2 = v.apply_async(time.sleep, 1) | |||
|
460 | self.assertIn(msg_id, self.client.results) | |||
|
461 | self.assertIn(msg_id, self.client.metadata) | |||
|
462 | self.client.purge_local_results(ar) | |||
|
463 | self.assertNotIn(msg_id, self.client.results) | |||
|
464 | self.assertNotIn(msg_id, self.client.metadata) | |||
|
465 | with self.assertRaises(RuntimeError): | |||
|
466 | self.client.purge_local_results(ar2) | |||
|
467 | ar2.get() | |||
|
468 | self.client.purge_local_results(ar2) | |||
|
469 | ||||
|
470 | def test_purge_all_local_results_outstanding(self): | |||
|
471 | v = self.client[-1] | |||
|
472 | ar = v.apply_async(time.sleep, 1) | |||
|
473 | with self.assertRaises(RuntimeError): | |||
|
474 | self.client.purge_local_results('all') | |||
|
475 | ar.get() | |||
|
476 | self.client.purge_local_results('all') | |||
|
477 | ||||
453 | def test_purge_all_hub_results(self): |
|
478 | def test_purge_all_hub_results(self): | |
454 | self.client.purge_hub_results('all') |
|
479 | self.client.purge_hub_results('all') | |
455 | hist = self.client.hub_history() |
|
480 | hist = self.client.hub_history() |
General Comments 0
You need to be logged in to leave comments.
Login now