def test_decrement(self, fake_put): # test tracker = Tracker(None) tracker.decrement() # validation fake_put.assert_called_once_with(0)
def test_construction(self): # test canceled = Mock() tracker = Tracker(canceled) # validation self.assertEqual(tracker.canceled, canceled) self.assertTrue(isinstance(tracker.queue, Queue))
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))
def test_wait_value_error(self): # test tracker = Tracker(None) self.assertRaises(ValueError, tracker.wait, -1)
def test_init(self): # test tracker = Tracker() # validation self.assertTrue(isinstance(tracker.queue, Queue))