Merge pull request #3094 from andrewbaldwin44/feature/api-base-url

Web UI: Add optional base url for locust requests to an external API

Author: cyberw

Diff for: locust/webui/src/components/StateButtons/ResetButton.tsx

@@ -2,7 +2,7 @@ import { Button } from '@mui/material';
 
 export default function ResetButton() {
   const onResetStatsClick = () => {
-    fetch('stats/reset');
+    fetch((window.baseUrl ? `${window.baseUrl}/` : '') + 'stats/reset');
   };
 
   return (

Diff for: locust/webui/src/components/StateButtons/StopButton.tsx

@@ -9,7 +9,7 @@ export default function StopButton() {
   }, []);
 
   const onStopButtonClick = () => {
-    fetch('stop');
+    fetch((window.baseUrl ? `${window.baseUrl}/` : '') + 'stop');
     setIsLoading(true);
   };
 

Diff for: locust/webui/src/components/SwarmForm/SwarmForm.tsx

@@ -203,7 +203,9 @@ function SwarmForm({
               </Accordion>
             </>
           )}
-          {!isEmpty(extraOptions) && <CustomParameters extraOptions={extraOptions} />}
+          {!!extraOptions && !isEmpty(extraOptions) && (
+            <CustomParameters extraOptions={extraOptions} />
+          )}
           {alert && !errorMessage && (
             <Alert severity={alert.level || 'info'}>{alert.message}</Alert>
           )}

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

@@ -20,7 +20,7 @@ export const swarmTemplateArgs = window.templateArgs
 export const htmlReportProps: IReport | false = !!(swarmTemplateArgs as IReportTemplateArgs)
   .isReport && {
   ...(swarmTemplateArgs as IReportTemplateArgs),
-  charts: swarmTemplateArgs.history?.reduce(
+  charts: (swarmTemplateArgs.history || []).reduce(
     (charts, { currentResponseTimePercentiles, ...history }) =>
       updateArraysAtProps(charts, { ...currentResponseTimePercentiles, ...history }),
     {} as ICharts,

Diff for: locust/webui/src/redux/api/swarm.ts

@@ -10,8 +10,18 @@ import {
 import { createFormData } from 'utils/object';
 import { camelCaseKeys, snakeCaseKeys } from 'utils/string';
 
+const baseQuery = (args: any, api: any, extraOptions: any) =>
+  fetchBaseQuery(
+    window.baseUrl
+      ? {
+          baseUrl: window.baseUrl,
+          credentials: 'include',
+        }
+      : undefined,
+  )(args, api, extraOptions);
+
 export const api = createApi({
-  baseQuery: fetchBaseQuery(),
+  baseQuery: baseQuery,
   endpoints: builder => ({
     getStats: builder.query<IStatsResponse, void>({
       query: () => 'stats/requests',

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

@@ -7,4 +7,5 @@ export interface IWindow {
   templateArgs: IReportTemplateArgs | ISwarmState;
   authArgs: IAuthArgs;
   theme?: PaletteMode;
+  baseUrl?: string;
 }