| 707 | class FailingQueueTest(BlockingTestMixin): |
| 708 | |
| 709 | def setUp(self): |
| 710 | |
| 711 | Queue = self.queue.Queue |
| 712 | |
| 713 | class FailingQueue(Queue): |
| 714 | def __init__(self, *args): |
| 715 | self.fail_next_put = False |
| 716 | self.fail_next_get = False |
| 717 | Queue.__init__(self, *args) |
| 718 | def _put(self, item): |
| 719 | if self.fail_next_put: |
| 720 | self.fail_next_put = False |
| 721 | raise FailingQueueException("You Lose") |
| 722 | return Queue._put(self, item) |
| 723 | def _get(self): |
| 724 | if self.fail_next_get: |
| 725 | self.fail_next_get = False |
| 726 | raise FailingQueueException("You Lose") |
| 727 | return Queue._get(self) |
| 728 | |
| 729 | self.FailingQueue = FailingQueue |
| 730 | |
| 731 | super().setUp() |
| 732 | |
| 733 | def failing_queue_test(self, q): |
| 734 | if q.qsize(): |