From 41d6484106e21408c233525f980936df403db26c Mon Sep 17 00:00:00 2001
From: Lars Holmberg <lars.holmberg@redshirt.se>
Date: Wed, 11 Dec 2024 10:31:19 +0100
Subject: [PATCH 1/2] Treat exceptions in init event handler as fatal and shut
 down locust right away.

---
 locust/main.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/locust/main.py b/locust/main.py
index 67b278f8a1..91b449e4f9 100644
--- a/locust/main.py
+++ b/locust/main.py
@@ -544,6 +544,9 @@ def assign_equal_weights(environment, **kwargs):
     # Fire locust init event which can be used by end-users' code to run setup code that
     # need access to the Environment, Runner or WebUI.
     environment.events.init.fire(environment=environment, runner=runner, web_ui=web_ui)
+    if log.unhandled_greenlet_exception:
+        # treat exceptions in init handlers as fatal. They are already logged so no need to log anything more.
+        sys.exit(1)
 
     if web_ui:
         web_ui.start()

From c0fe9c04df46fb02287cce8e7847380ba3259ae0 Mon Sep 17 00:00:00 2001
From: Lars Holmberg <lars.holmberg@redshirt.se>
Date: Wed, 11 Dec 2024 10:44:34 +0100
Subject: [PATCH 2/2] Add test case for exceptions in init handler

---
 locust/test/test_main.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/locust/test/test_main.py b/locust/test/test_main.py
index 8f6d4b8501..4ed88b8e2b 100644
--- a/locust/test/test_main.py
+++ b/locust/test/test_main.py
@@ -1442,6 +1442,39 @@ def my_task(self):
             # ensure stats printer printed at least one report before shutting down and that there was a final report printed as well
             self.assertRegex(stderr, r".*Aggregated[\S\s]*Shutting down[\S\s]*Aggregated.*")
 
+    def test_exception_in_init_event(self):
+        with mock_locustfile(
+            content=textwrap.dedent(
+                """
+            from locust import User, task, constant, events
+            @events.init.add_listener
+            def _(*args, **kw):
+                raise Exception("something went wrong")
+
+            class TestUser(User):
+                wait_time = constant(10)
+                @task
+                def my_task(self):
+                    pass
+                    """
+            )
+        ) as mocked:
+            proc = subprocess.Popen(
+                [
+                    "locust",
+                    "-f",
+                    mocked.file_path,
+                    "--host",
+                    "https://test.com/",
+                ],
+                stdout=PIPE,
+                stderr=PIPE,
+                text=True,
+            )
+            stdout, stderr = proc.communicate(timeout=3)
+            self.assertIn("something went wrong", stderr)
+            self.assertEqual(1, proc.returncode)
+
 
 class DistributedIntegrationTests(ProcessIntegrationTest):
     failed_port_check = False