def test_wait_raising_empty(self, fake_get): tokens = [Empty(), 0, 0, 0] fake_get.side_effect = SideEffect(tokens) # test tracker = Tracker() tracker.wait(len(tokens) - 1) # validation self.assertEqual(fake_get.call_count, len(tokens))
def test_wait(self, fake_get): events = [0, 0, 0] fake_get.side_effect = events # test n = len(events) tracker = Tracker() tracker.wait(n) # validation self.assertEqual(fake_get.call_count, n)
def test_wait(self, fake_get): events = [0, 0, 0] canceled = Mock() canceled.is_set.return_value = False fake_get.side_effect = events # test n = len(events) tracker = Tracker(canceled) tracker.wait(n) # validation self.assertEqual(canceled.is_set.call_count, n) self.assertEqual(fake_get.call_count, n)
def test_wait_raising_empty(self, fake_get): canceled = Mock() canceled.is_set.return_value = False tokens = [Empty(), 0, 0, 0] fake_get.side_effect = SideEffect(tokens) # test tracker = Tracker(canceled) tracker.wait(len(tokens) - 1) # validation self.assertEqual(canceled.is_set.call_count, len(tokens)) self.assertEqual(fake_get.call_count, len(tokens))