class TriggerProcessorTestCase(unittest.TestCase): def setUp(self): self.entries = range(10) self.trigger_processor = TriggerProcessor([]) def test_single_trigger(self): random.seed(0) self.trigger_processor.triggers = [Random()] entries = self.trigger_processor.process(self.entries) results = [2, 3, 5, 7, 8] self.assertEqual(list(entries), results) def test_two_triggers(self): random.seed(0) self.trigger_processor.triggers = [Random(), Random()] entries = self.trigger_processor.process(self.entries) results = [1, 2, 3, 4, 5, 7, 9] self.assertEqual(list(entries), results) def test_no_trigger(self): self.trigger_processor.triggers = [] entries = self.trigger_processor.process(self.entries) self.assertEqual(list(entries), self.entries)
def setUp(self): self.entries = range(10) self.trigger_processor = TriggerProcessor([])