def test_weight_locusts(self):
maxDiff = 2048
class BaseUser(User):
pass
class L1(BaseUser):
weight = 101
class L2(BaseUser):
weight = 99
class L3(BaseUser):
weight = 100
runner = Environment(user_classes=[L1, L2, L3]).create_local_runner()
self.assert_locust_class_distribution({
L1: 10,
L2: 9,
L3: 10
}, runner.weight_users(29))
self.assert_locust_class_distribution({
L1: 10,
L2: 10,
L3: 10
}, runner.weight_users(30))
self.assert_locust_class_distribution({
L1: 11,
L2: 10,
L3: 10
}, runner.weight_users(31))
def test_weight_locusts_fewer_amount_than_user_classes(self):
class BaseUser(User):
pass
class L1(BaseUser):
weight = 101
class L2(BaseUser):
weight = 99
class L3(BaseUser):
weight = 100
runner = Environment(user_classes=[L1, L2, L3]).create_local_runner()
self.assertEqual(1, len(runner.weight_users(1)))
self.assert_locust_class_distribution({L1:1}, runner.weight_users(1))