Add run_time to template_args

locustio · Sep 25, 2023 · 7573de5 · 7573de5
1 parent 2956155
commit 7573de5
Show file tree

Hide file tree

Showing 6 changed files with 15 additions and 8 deletions.
diff --git a/locust/web.py b/locust/web.py
@@ -578,6 +578,7 @@ def update_template_args(self):
             "stats_history_enabled": options and options.stats_history_enabled,
             "tasks": dumps({}),
             "extra_options": extra_options,
+            "run_time": options and options.run_time,
             "show_userclass_picker": self.userclass_picker_is_active,
             "available_user_classes": available_user_classes,
             "available_shape_classes": available_shape_classes,

diff --git a/locust/webui/dist/assets/index-b0f93c80.js b/locust/webui/dist/assets/index-b0f93c80.js
diff --git a/locust/webui/dist/assets/index-c3bc88fd.js b/locust/webui/dist/assets/index-c3bc88fd.js
diff --git a/locust/webui/dist/index.html b/locust/webui/dist/index.html
@@ -13,7 +13,7 @@
     <meta name="theme-color" content="#000000" />
 
     <title>Locust</title>
-    <script type="module" crossorigin src="/assets/index-c3bc88fd.js"></script>
+    <script type="module" crossorigin src="/assets/index-b0f93c80.js"></script>
     <link rel="modulepreload" crossorigin href="/assets/vendor-f6987cc3.js">
   </head>
   <body>

diff --git a/locust/webui/src/components/SwarmForm/SwarmForm.tsx b/locust/webui/src/components/SwarmForm/SwarmForm.tsx
@@ -40,6 +40,7 @@ interface ISwarmForm
       | 'isShape'
       | 'host'
       | 'overrideHostWarning'
+      | 'runTime'
       | 'showUserclassPicker'
       | 'spawnRate'
       | 'userCount'
@@ -52,6 +53,7 @@ function SwarmForm({
   extraOptions,
   isShape,
   overrideHostWarning,
+  runTime,
   setSwarm,
   showUserclassPicker,
   spawnRate,
@@ -115,6 +117,7 @@ function SwarmForm({
             </AccordionSummary>
             <AccordionDetails>
               <TextField
+                defaultValue={runTime}
                 label='Run time (e.g. 20, 20s, 3m, 2h, 1h20m, 3h30m10s, etc.)'
                 name='runTime'
                 sx={{ width: '100%' }}
@@ -140,6 +143,7 @@ const storeConnector = ({
     host,
     numUsers,
     overrideHostWarning,
+    runTime,
     spawnRate,
     showUserclassPicker,
     userCount,
@@ -153,6 +157,7 @@ const storeConnector = ({
   overrideHostWarning,
   showUserclassPicker,
   numUsers,
+  runTime,
   spawnRate,
   userCount,
 });

diff --git a/locust/webui/src/redux/slice/swarm.slice.ts b/locust/webui/src/redux/slice/swarm.slice.ts
@@ -24,6 +24,7 @@ export interface ISwarmState {
   locustfile: string;
   numUsers: number | null;
   overrideHostWarning: boolean;
+  runTime: number;
   showUserclassPicker: boolean;
   spawnRate: number | null;
   state: string;