Bug in StagesShapeWithCustomUsers

[Elvintest]

locust/examples/custom_shape/staging_user_classes.py

Line 49 in e20bef5

for stage in self.stages:

Hello, I tried your code in distributing stages, in this version it has one disadvantage - we can not switch to another stage cause run_time in the rest of cases will be more than stage_duration:

            if run_time < stage["duration"]
                try:
                    tick_data = (stage["users"], stage["spawn_rate"], stage["user_classes"])
                except:
                    tick_data = (stage["users"], stage["spawn_rate"])
                return tick_data

So to switch our test to another stage but at the same time to keep the feature of stopping our test i would offer you to try this solution:

    def tick(self):
        run_time = self.get_run_time()
        duration = 0
        for stage in self.stages:
            duration +=stage["duration"]
            if run_time < duration:
                try:
                    tick_data = (stage["users"], stage["spawn_rate"], stage["user_classes"])
                except:
                    tick_data = (stage["users"], stage["spawn_rate"])
                return tick_data
        return None

In my version we check that common run_time doesn't exceed duration of previous stages and incoming new one