Update logging tests

Author: ilyabe

Diff for: locust/test/test_log.py

@@ -67,6 +67,8 @@ def my_task(self):
                     "-t",
                     "1",
                     "--headless",
+                    "--host",
+                    "https://test.com",
                 ],
                 stderr=subprocess.STDOUT,
                 timeout=10,
@@ -126,6 +128,8 @@ def my_task(self):
                     "-t",
                     "1",
                     "--headless",
+                    "--host",
+                    "https://test.com",
                     "--skip-log-setup",
                 ],
                 stderr=subprocess.STDOUT,
@@ -167,6 +171,8 @@ def my_task(self):
                             "-t",
                             "1",
                             "--headless",
+                            "--host",
+                            "https://test.com",
                             "--logfile",
                             log_file_path,
                         ],
@@ -213,25 +219,28 @@ def test_user_broken_on_start(self):
             from locust import HttpUser, task
 
             class TestUser(HttpUser):
-                host = "invalidhost"
 
                 def on_start(self):
                     self.client.get("/")
                 """
             )
         ) as file_path:
-            output = subprocess.check_output(
-                [
-                    "locust",
-                    "-f",
-                    file_path,
-                    "-t",
-                    "1",
-                    "--headless",
-                ],
-                stderr=subprocess.STDOUT,
-                timeout=10,
-                text=True,
-            )
-
-        self.assertIn("ERROR/locust.user.users: Invalid URL", output)
+            output = ""
+            try:
+                output = subprocess.check_output(
+                    [
+                        "locust",
+                        "-f",
+                        file_path,
+                        "-t",
+                        "1",
+                        "--headless",
+                        "--host",
+                        "invalidhost",
+                    ],
+                    stderr=subprocess.STDOUT,
+                    timeout=10,
+                    text=True,
+                )
+            except subprocess.CalledProcessError as e:
+                self.assertIn("Invalid --host option: invalidhost", e.output)