Add create store export

locust/webui/src/lib.tsx

@@ -33,7 +33,7 @@ export { default as Logo } from 'assets/Logo';
 export { default as Modal } from 'components/Modal/Modal';
 export { default as useCreateTheme } from 'hooks/useCreateTheme';
 export { tabConfig } from 'components/Tabs/Tabs.constants';
-export { store as locustStore } from 'redux/store';
+export { store as locustStore, createStore as createLocustStore } from 'redux/store';
 export { pushQuery } from 'utils/url';
 
 export type { IRootState } from 'redux/store';

locust/webui/src/redux/store.ts

@@ -1,12 +1,19 @@
 import { configureStore } from '@reduxjs/toolkit';
 
 import { api } from 'redux/api/swarm';
-import rootReducer from 'redux/slice/root.slice';
+import rootReducer, { IRootState } from 'redux/slice/root.slice';
 
 export const store = configureStore({
   reducer: rootReducer,
   middleware: getDefaultMiddleware => getDefaultMiddleware().concat(api.middleware),
 });
 
+export const createStore = (preloadedState: Partial<IRootState>) =>
+  configureStore({
+    reducer: rootReducer,
+    middleware: getDefaultMiddleware => getDefaultMiddleware().concat(api.middleware),
+    preloadedState,
+  });
+
 export type { IRootState, Action } from 'redux/slice/root.slice';
 export default configureStore;