class TestDispatcherRound(TestCase): def setUp(self): self.dispatcher = RoundRobinDispatcher(["w1", "w2"]) def test_round_robin(self): self.assertEqual(self.dispatcher.get_next(), "w1") self.assertEqual(self.dispatcher.get_next(), "w2") def test_round_robin_return_to_zero(self): self.assertEqual(self.dispatcher.get_next(), "w1") self.assertEqual(self.dispatcher.get_next(), "w2") self.assertEqual(self.dispatcher.get_next(), "w1") self.assertEqual(self.dispatcher.get_next(), "w2")
def setUp(self): self.dispatcher = RoundRobinDispatcher(["w1", "w2"])