How to use test_custom_exit_code method in locust

Best Python code snippet using locust

test_main.py

Source:test_main.py

...122        self.assertIn("-f LOCUSTFILE, --locustfile LOCUSTFILE", output)123        self.assertIn("Logging options:", output)124        self.assertIn("--skip-log-setup      Disable Locust's logging setup.", output)125    126    def test_custom_exit_code(self):127        with temporary_file(content=textwrap.dedent("""128            from locust import User, task, constant, events129            @events.quitting.add_listener130            def _(environment, **kw):131                environment.process_exit_code = 42132            class TestUser(User):133                wait_time = constant(3)134                @task135                def my_task():136                    print("running my_task()")137        """)) as file_path:138            proc = subprocess.Popen(["locust", "-f", file_path], stdout=PIPE, stderr=PIPE)139            gevent.sleep(1)140            proc.send_signal(signal.SIGTERM)...

test_main_decorator.py

Source:test_main_decorator.py

...5        pass6    retval = _my_main()7    assert isinstance(retval, int)8    assert retval == 09def test_custom_exit_code():10    expected_error_code = 2411    @main()12    def _my_main():13        return expected_error_code14    retval = _my_main()15    assert isinstance(retval, int)...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.