Webui: adding profile argument and display in html report

Author: schwannden

Diff for: locust/argument_parser.py

@@ -838,6 +838,11 @@ def setup_parser_arguments(parser):
         dest="equal_weights",
         help="Use equally distributed task weights, overriding the weights specified in the locustfile.",
     )
+    other_group.add_argument(
+        "--profile",
+        type=str,
+        help="Set a profile to group the testruns together",
+    )
 
     user_classes_group = parser.add_argument_group("User classes")
     user_classes_group.add_argument(

Diff for: locust/env.py

@@ -39,6 +39,7 @@ def __init__(
         available_shape_classes: dict[str, LoadTestShape] | None = None,
         available_user_tasks: dict[str, list[TaskSet | Callable]] | None = None,
         dispatcher_class: type[UsersDispatcher] = UsersDispatcher,
+        profile: str | None = None,
     ):
         self.runner: Runner | None = None
         """Reference to the :class:`Runner <locust.runners.Runner>` instance"""
@@ -76,6 +77,8 @@ def __init__(
         """Base URL of the target system"""
         self.reset_stats = reset_stats
         """Determines if stats should be reset once all simulated users have been spawned"""
+        self.profile = profile
+        """Profile name for the test run"""
         if stop_timeout is not None:
             self.stop_timeout = stop_timeout
         elif parsed_options:

Diff for: locust/html.py

@@ -105,6 +105,7 @@ def get_html_report(
             "locustfile": escape(str(environment.locustfile)),
             "tasks": task_data,
             "percentiles_to_chart": stats_module.PERCENTILES_TO_CHART,
+            "profile": escape(str(environment.profile)),
         },
         theme=theme,
     )

Diff for: locust/locustfile.py

@@ -0,0 +1,21 @@
+from locust import HttpUser, between, task
+
+import time
+
+
+class QuickstartUser(HttpUser):
+    wait_time = between(1, 2)
+
+    @task
+    def hello_world(self):
+        self.client.get("/hello")
+        self.client.get("/world")
+
+    @task(3)
+    def view_item(self):
+        for item_id in range(10):
+            self.client.get(f"/item?id={item_id}", name="/item")
+            time.sleep(1)
+
+    def on_start(self):
+        self.client.post("/login", json={"username": "foo", "password": "bar"})

Diff for: locust/main.py

@@ -89,6 +89,7 @@ def create_environment(
         available_user_classes=available_user_classes,
         available_shape_classes=available_shape_classes,
         available_user_tasks=available_user_tasks,
+        profile=options.profile,
     )
 
 

Diff for: locust/test/test_load_locustfile.py

@@ -225,3 +225,9 @@ def test_locustfile_from_url(self):
                 f"{os.getcwd()}/examples/basic.py",
             )
         )
+
+    def test_profile_flag(self):
+        options = parse_options()
+        self.assertEqual(None, options.profile)
+        options = parse_options(args=["--profile", "test-profile"])
+        self.assertEqual("test-profile", options.profile)

Diff for: locust/webui/report.html

@@ -17,6 +17,9 @@
     <script>
       window.templateArgs = {{ template_args|tojson }}
       window.theme = "{{theme}}"
+      if (window.templateArgs.profile) {
+        document.title = `Locust - ${window.templateArgs.profile}`
+      }
     </script>
 
     <script type="module" src="./src/HtmlReport.tsx"></script>

Diff for: locust/webui/src/pages/HtmlReport.tsx

@@ -3,6 +3,7 @@ import CssBaseline from '@mui/material/CssBaseline';
 import { ThemeProvider } from '@mui/material/styles';
 import { combineReducers, configureStore } from '@reduxjs/toolkit';
 import { Provider } from 'react-redux';
+import { useEffect } from 'react';
 
 import { ExceptionsTable } from 'components/ExceptionsTable/ExceptionsTable';
 import { FailuresTable } from 'components/FailuresTable/FailuresTable';
@@ -51,6 +52,12 @@ export default function HtmlReport({
   responseTimeStatistics,
   tasks,
 }: IReport) {
+  useEffect(() => {
+    document.title = window.templateArgs.profile 
+      ? `Locust - ${window.templateArgs.profile}`
+      : 'Locust';
+  }, []);
+
   return (
     <Provider store={reportStore}>
       <ThemeProvider theme={muiTheme}>
@@ -73,6 +80,12 @@ export default function HtmlReport({
             )}
           </Box>
           <Box sx={{ my: 2 }}>
+            {window.templateArgs.profile && (
+              <Box sx={{ display: 'flex', columnGap: 0.5 }}>
+                <Typography fontWeight={600}>Profile:</Typography>
+                <Typography>{window.templateArgs.profile}</Typography>
+              </Box>
+            )}
             <Box sx={{ display: 'flex', columnGap: 0.5 }}>
               <Typography fontWeight={600}>During:</Typography>
               <Typography>

Diff for: locust/webui/src/pages/tests/HtmlReport.test.tsx

@@ -1,11 +1,24 @@
-import { describe, expect, test } from 'vitest';
+import { describe, expect, test, vi, beforeEach } from 'vitest';
 
 import HtmlReport from 'pages/HtmlReport';
 import { swarmReportMock } from 'test/mocks/swarmState.mock';
 import { renderWithProvider } from 'test/testUtils';
 import { formatLocaleString } from 'utils/date';
+import { IReportTemplateArgs } from 'types/swarm.types';
+
+declare global {
+  interface Window {
+    templateArgs: IReportTemplateArgs;
+  }
+}
 
 describe('HtmlReport', () => {
+  beforeEach(() => {
+    // Reset window.templateArgs before each test
+    window.templateArgs = {} as IReportTemplateArgs;
+    document.title = '';
+  });
+
   test('renders a report', () => {
     const { getByRole, getByText } = renderWithProvider(<HtmlReport {...swarmReportMock} />);
 
@@ -19,6 +32,21 @@ describe('HtmlReport', () => {
     expect(getByText(swarmReportMock.host)).toBeTruthy();
   });
 
+  test('profile is not rendered when it is not present', () => {
+    const { queryByText } = renderWithProvider(<HtmlReport {...swarmReportMock} />);
+
+    expect(queryByText('Profile:')).toBeNull();
+  });
+
+  test('profile is rendered when it is present', () => {
+    window.templateArgs.profile = 'test-profile';
+    const { getByText } = renderWithProvider(<HtmlReport {...swarmReportMock} />);
+
+    expect(getByText('Profile:')).toBeTruthy();
+    expect(getByText('test-profile')).toBeTruthy();
+    expect(document.title).toBe('Locust - test-profile');
+  });
+
   test('formats the start and end time as expected', () => {
     const { getByText } = renderWithProvider(<HtmlReport {...swarmReportMock} />);
 

Diff for: locust/webui/src/types/swarm.types.ts

@@ -64,6 +64,7 @@ export interface ISwarmState {
   users: { [key: string]: ISwarmUser };
   version: string;
   workerCount: number;
+  profile?: string;
 }
 
 export interface IReport {
@@ -86,6 +87,7 @@ export interface IReportTemplateArgs extends Omit<IReport, 'charts'> {
   isReport?: boolean;
   percentilesToChart: number[];
   percentilesToStatistics: number[];
+  profile?: string;
 }
 
 export interface ISwarmFormInput