feat: refresh tests tree

Author: rocketraccoon

lib/gui/server.ts

@@ -180,6 +180,16 @@ export const start = async (args: ServerArgs): Promise<ServerReadyData> => {
         }
     });
 
+    server.post('/refresh', async (_, res) => {
+        try {
+            await app.initialize({refresh: true});
+
+            res.sendStatus(OK);
+        } catch (e) {
+            res.status(INTERNAL_SERVER_ERROR).send(`Error while trying to refresh tests: ${(e as Error).message}`);
+        }
+    });
+
     server.post('/run-custom-gui-action', async ({body: payload}, res) => {
         try {
             if (toolAdapter.toolName === ToolName.Testplane) {

lib/gui/tool-runner/index.ts

@@ -61,6 +61,10 @@ export interface RunParams {
     retry?: boolean;
 }
 
+export interface InitializeParams {
+    refresh: boolean;
+}
+
 export class ToolRunner {
     private _testFiles: string[];
     private _toolAdapter: ToolAdapter;
@@ -119,7 +123,7 @@ export class ToolRunner {
         });
     }
 
-    async initialize(): Promise<void> {
+    async initialize(params?: InitializeParams): Promise<void> {
         await mergeDatabasesForReuse(this._reportPath);
         await prepareLocalDatabase(this._reportPath);
 
@@ -144,9 +148,13 @@ export class ToolRunner {
         this._collection = await this._readTests();
 
         this._toolAdapter.htmlReporter.emit(PluginEvents.DATABASE_CREATED, dbClient.getRawConnection());
-        await this._reportBuilder.saveStaticFiles();
 
-        this._reportBuilder.setApiValues(this._toolAdapter.htmlReporter.values);
+        if (!params?.refresh) {
+            await this._reportBuilder.saveStaticFiles();
+
+            this._reportBuilder.setApiValues(this._toolAdapter.htmlReporter.values);
+        }
+
         await this._handleRunnableCollection();
     }
 

lib/static/modules/action-names.ts

@@ -82,5 +82,6 @@ export default {
     SNAPSHOTS_PLAYER_GO_TO_TIME: 'SNAPSHOTS_PLAYER_GO_TO_TIME',
     SNAPSHOTS_PLAYER_TOGGLE_VISIBILITY: 'SNAPSHOTS_PLAYER_TOGGLE_VISIBILITY',
     SET_BROWSER_FEATURES: 'SET_BROWSER_FEATURES',
-    SET_SEARCH_LOADING: 'SET_SEARCH_LOADING'
+    SET_SEARCH_LOADING: 'SET_SEARCH_LOADING',
+    SET_REFRESH_LOADING: 'SET_REFRESH_LOADING'
 } as const;

lib/static/modules/actions/filters.ts

@@ -36,6 +36,11 @@ export const setSearchLoading = (payload: boolean): SetSearchLoading => {
     return {type: actionNames.SET_SEARCH_LOADING, payload};
 };
 
+export type SetRefreshLoading = Action<typeof actionNames.SET_REFRESH_LOADING, boolean>;
+export const setRefreshLoading = (payload: boolean): SetRefreshLoading => {
+    return {type: actionNames.SET_REFRESH_LOADING, payload};
+};
+
 export type ChangeViewModeAction = Action<typeof actionNames.CHANGE_VIEW_MODE, FilterPayload<ViewMode>>;
 export const changeViewMode = (payload: ChangeViewModeAction['payload']): ChangeViewModeAction => ({type: actionNames.CHANGE_VIEW_MODE, payload});
 
@@ -44,5 +49,6 @@ export type FiltersAction =
     | SetMatchCaseFilterAction
     | SetUseRegexFilterAction
     | SelectBrowsersAction
+    | SetRefreshLoading
     | ChangeViewModeAction
     | SetSearchLoading;

lib/static/modules/reducers/filters.ts

@@ -42,6 +42,7 @@ export default (state: State, action: FiltersAction | InitGuiReportAction | Init
                 state,
                 {
                     app: {
+                        isRefreshLoading: false,
                         [Page.suitesPage]: {
                             viewMode: suitesPageViewMode
                         },
@@ -114,6 +115,9 @@ export default (state: State, action: FiltersAction | InitGuiReportAction | Init
         case actionNames.SET_SEARCH_LOADING:
             return applyStateUpdate(state, {app: {isSearchLoading: action.payload}});
 
+        case actionNames.SET_REFRESH_LOADING:
+            return applyStateUpdate(state, {app: {isRefreshLoading: action.payload}});
+
         default:
             return state;
     }

lib/static/new-ui/components/TreeActionsToolbar/index.module.css

@@ -13,6 +13,15 @@
     z-index: 1;
 }
 
+@keyframes spin {
+    from { transform: rotate(0deg); }
+    to { transform: rotate(360deg); }
+}
+
+.is-refresh-loading {
+    animation: spin 0.8s linear infinite;
+}
+
 .buttons-container {
     margin-left: auto;
     display: flex;

lib/static/new-ui/components/TreeActionsToolbar/index.tsx

@@ -1,4 +1,4 @@
-import {Hotkey, Icon, Popover, Spin} from '@gravity-ui/uikit';
+import {Hotkey, Icon, Popover} from '@gravity-ui/uikit';
 import classNames from 'classnames';
 import {
     ArrowUturnCcwLeft,
@@ -13,7 +13,8 @@ import {
     SquareMinus,
     ListUl,
     Hierarchy,
-    GearPlay
+    GearPlay,
+    ArrowsRotateRight
 } from '@gravity-ui/icons';
 import React, {ReactNode, useCallback, useMemo} from 'react';
 import {useDispatch, useSelector} from 'react-redux';
@@ -22,9 +23,9 @@ import styles from './index.module.css';
 import {
     deselectAll,
     selectAll,
-    setAllTreeNodesState, setTreeViewMode,
+    setAllTreeNodesState, setRefreshLoading, setTreeViewMode,
     staticAccepterStageScreenshot,
-    staticAccepterUnstageScreenshot, thunkRunTests
+    staticAccepterUnstageScreenshot, thunkInitGuiReport, thunkRunTests
 } from '@/static/modules/actions';
 import {ImageEntity, TreeViewMode} from '@/static/new-ui/types/store';
 import {CHECKED, INDETERMINATE} from '@/constants/checked-statuses';
@@ -74,6 +75,7 @@ export function TreeActionsToolbar({onHighlightCurrentTest, className}: TreeActi
     const selectedTests = useSelector(getCheckedTests);
     const visibleBrowserIds: string[] = useSelector(getVisibleBrowserIds);
     const isInitialized = useSelector(getIsInitialized);
+    const isRefreshLoading = useSelector((state) => state.app.isRefreshLoading);
 
     const isRunTestsAvailable = useSelector(state => state.app.availableFeatures)
         .find(feature => feature.name === RunTestsFeature.name);
@@ -139,6 +141,12 @@ export function TreeActionsToolbar({onHighlightCurrentTest, className}: TreeActi
     const selectedOrVisible = isSelectedAtLeastOne ? 'selected' : 'visible';
     const areActionsDisabled = isRunning || !isInitialized;
 
+    const handleRefresh = useCallback(async () => {
+        dispatch(setRefreshLoading(true));
+        await fetch('/refresh', {method: 'POST'});
+        dispatch(thunkInitGuiReport({isNewUi: true}));
+    }, []);
+
     const handleRun = useCallback((): void => {
         analytics?.trackFeatureUsage({featureName: `${ANALYTICS_PREFIX} run tests`});
         if (isSelectedAtLeastOne) {
@@ -195,19 +203,24 @@ export function TreeActionsToolbar({onHighlightCurrentTest, className}: TreeActi
     const getViewButtons = (): ReactNode => (
         <>
             {isRunTestsAvailable && (
-                isRunning
-                    ? (
-                        <Spin size={'xs'} style={{marginRight: '6px'}}/>
-                    ) : (
-                        <IconButton
-                            className={styles.iconButton}
-                            icon={<Icon data={Play} height={14}/>}
-                            tooltip={<>Run {selectedOrVisible} ⋅ <Hotkey value="shift+r" view="light" /></>}
-                            view={'flat'}
-                            onClick={handleRun}
-                            disabled={isRunning || !isInitialized}
-                        />
-                    )
+                <IconButton
+                    className={styles.iconButton}
+                    icon={<Icon className={classNames({[styles.isRefreshLoading]: isRefreshLoading})} data={ArrowsRotateRight} height={14}/>}
+                    tooltip="Refresh tests tree"
+                    view="flat"
+                    onClick={handleRefresh}
+                    disabled={isRunning || !isInitialized || isRefreshLoading}
+                />
+            )}
+            {isRunTestsAvailable && (
+                <IconButton
+                    className={styles.iconButton}
+                    icon={<Icon className={classNames({[styles.isRunning]: isRunning})} data={Play} height={14}/>}
+                    tooltip={<>Run {selectedOrVisible} ⋅ <Hotkey value="shift+r" view="light" /></>}
+                    view={'flat'}
+                    onClick={handleRun}
+                    disabled={isRunning || !isInitialized}
+                />
             )}
             {isRunTestsAvailable && hasRunTestOptions && <Popover
                 content={<div className={styles.runOptionsContainer}><ExtensionPoint name={ExtensionPointName.RunTestOptions}></ExtensionPoint></div>}

lib/static/new-ui/features/suites/selectors.ts

@@ -14,7 +14,7 @@ export const getCurrentBrowser = (state: State): BrowserEntity | null => {
 export const getCurrentResultId = (state: State): string | null => {
     const browserId = state.app.suitesPage.currentBrowserId;
     const treeNodeId = state.app.suitesPage.currentTreeNodeId;
-    if (!browserId || !treeNodeId) {
+    if (!browserId || !treeNodeId || !state.tree.browsers.byId[browserId]) {
         return null;
     }
 
@@ -89,7 +89,7 @@ export const isTimeTravelPlayerAvailable = (state: State): boolean => {
 export const getAttempt = (state: State): number | null => {
     const browserId = state.app.suitesPage.currentBrowserId;
 
-    if (browserId) {
+    if (browserId && state.tree.browsers.stateById[browserId]) {
         return state.tree.browsers.stateById[browserId].retryIndex;
     }
 

lib/static/new-ui/types/store.ts

@@ -260,6 +260,7 @@ export interface State {
         isInitialized: boolean;
         availableFeatures: Feature[],
         isSearchLoading?: boolean;
+        isRefreshLoading?: boolean;
         [Page.suitesPage]: {
             currentTreeNodeId: string | null;
             currentBrowserId: string | null;