def test_random_lb_with_mixed_members(self):
cluster = _MockClusterService(3, 3)
lb = RandomLB()
lb.init(cluster)
for _ in range(20):
self.assertTrue(0 <= lb.next().id < 6)
self.assertTrue(0 <= lb.next_data_member().id < 3)
def test_random_lb_with_data_members(self):
cluster = _MockClusterService(3, 0)
lb = RandomLB()
lb.init(cluster)
self.assertTrue(lb.can_get_next_data_member())
for _ in range(10):
self.assertTrue(0 <= lb.next().id <= 2)
self.assertTrue(0 <= lb.next_data_member().id <= 2)
def test_random_lb_with_lite_members(self):
cluster = _MockClusterService(0, 3)
lb = RandomLB()
lb.init(cluster)
for _ in range(10):
self.assertTrue(0 <= lb.next().id <= 2)
self.assertIsNone(lb.next_data_member())
def test_random_lb_with_no_members(self):
cluster = _MockClusterService(0, 0)
lb = RandomLB()
lb.init(cluster)
self.assertIsNone(lb.next())
self.assertIsNone(lb.next_data_member())