Run Predefined Task after Another

[nicholaswkc34]

Dear all forumers, I had coded load test using locust. I have two api endpoint to test which are register and login.

In the execution, during register, email will be generate random using pattern and this will be read from csv file in login. My questions is How to run register @task all first and then run login task? Is there any method to do it? I know how to define higher priority but this does not guarantee all users has register first.

#@task
    def register(self):
        email: str
        telcoCode: int
        phone: int

        self.client.request_name = "API Register User"
        with open('email.csv', 'w', newline='\n') as csvfile:
            for x in range(5):
                email = ''.join(random.choices(string.ascii_letters + string.digits, k=10)) + str(x + 1) + "@gmail.com"
                telcocode = random.randint(10, 19)
                phone = random.randint(10000000, 99999999)
                phonestr = "+60" + str(telcocode) + str(phone)

                csvwriter = csv.writer(csvfile, delimiter=',')
                csvwriter.writerow(email)
                self.client.headers = {
                    "x-api-key": "KZThJSCH1vWj5snaeQQ+TQ5rTbucayxXtDVHFmfRSJag5gLkUKOMgrpitL1jHIVr6bYAN5N9NrKWg6RhdgNThQ=="}
                response = self.client.post("https://staging-sso-api-pub.revmedia.my/register",
                                            {"email": email, "phone": phonestr,
                                             "password": "abcd1234", "password_confirmation": "abcd1234"}
                                            )
                logging.info(response.text)


    @task
    def login(self):
        self.client.request_name = "API Login"
        self.client.headers = {
            "x-api-key": "KZThJSCH1vWj5snaeQQ+TQ5rTbucayxXtDVHFmfRSJag5gLkUKOMgrpitL1jHIVr6bYAN5N9NrKWg6RhdgNThQ=="}
        #email = next(email_reader)
        response = self.client.post("https://staging-sso-api-pub.revmedia.my/auth",
                                    {"email": "hazreenaaida@gmail.com",
                                     "password": "abcd1234"})

Please help. Thanks in advance.

[cyberw]

You can use SequentialTaskSet, but usually it is easier and more flexible to just have a single task with all the steps https://docs.locust.io/en/stable/tasksets.html#sequentialtaskset-class

[nicholaswkc34]

First of all, Thanks for your reply. I'm appreciated it.