feat: add refresh tests

rocketraccoon · rocketraccoon · commit 57fffb6c9dfc · 2026-07-03T01:38:16.000+07:00
diff --git a/lib/gui/app.ts b/lib/gui/app.ts
@@ -53,6 +53,11 @@ export class App {
         return this._toolRunner.findEqualDiffs(data);
     }
 
+    async refreshTests(): Promise<ToolRunnerTree | null> {
+        await this._toolRunner.refreshTests();
+        return this._toolRunner.tree;
+    }
+
     addClient(connection: Response): void {
         this._toolRunner.addClient(connection);
     }
diff --git a/lib/gui/server.ts b/lib/gui/server.ts
@@ -274,6 +274,15 @@ export const start = async (args: ServerArgs): Promise<ServerReadyData> => {
         logger.log('server shutting down');
     });
 
+    server.get('/refresh-tests', async (_req, res) => {
+        try {
+            const tree = await app.refreshTests();
+            res.json(tree satisfies GetInitResponse);
+        } catch (e: unknown) {
+            res.status(INTERNAL_SERVER_ERROR).send(`Error while refreshing tests: ${(e as Error).message}`);
+        }
+    });
+
     server.post('/stop', (_req, res) => {
         try {
             // pass 0 to prevent terminating testplane process
diff --git a/lib/gui/tool-runner/index.ts b/lib/gui/tool-runner/index.ts
@@ -154,6 +154,13 @@ export class ToolRunner {
         return this._toolAdapter.readTests(this._testFiles, this._globalOpts);
     }
 
+    async refreshTests(): Promise<void> {
+        this._collection = await this._readTests();
+        this._testAdapters = {};
+        this._ensureReportBuilder().resetTree();
+        await this._handleRunnableCollection();
+    }
+
     protected _ensureReportBuilder(): GuiReportBuilder {
         if (!this._reportBuilder) {
             throw new Error('ToolRunner has to be initialized before usage');
@@ -424,6 +431,8 @@ export class ToolRunner {
     protected async _loadDataFromDatabase(): Promise<Tree | null> {
         const dbPath = path.resolve(this._reportPath, LOCAL_DATABASE_NAME);
 
+        this._ensureReportBuilder().flushDb();
+
         if (await fs.pathExists(dbPath)) {
             return getTestsTreeFromDatabase(dbPath, this._reporterConfig.baseHost);
         }
diff --git a/lib/report-builder/gui.ts b/lib/report-builder/gui.ts
@@ -1,5 +1,6 @@
 import _ from 'lodash';
 import {StaticReportBuilder, StaticReportBuilderOptions} from './static';
+import {TestAttemptManager} from '../test-attempt-manager';
 import {GuiTestsTreeBuilder, TestBranch, TestEqualDiffsData, TestRefUpdateData} from '../tests-tree-builder/gui';
 import {UPDATED, DB_COLUMNS, TestStatus, DEFAULT_TITLE_DELIMITER, SKIPPED, SUCCESS} from '../constants';
 import {ConfigForStaticFile, getConfigForStaticFile} from '../server-utils';
@@ -76,6 +77,12 @@ export class GuiReportBuilder extends StaticReportBuilder {
         };
     }
 
+    resetTree(): void {
+        this._testsTree = GuiTestsTreeBuilder.create({baseHost: this._reporterConfig.baseHost});
+        this._testAttemptManager = new TestAttemptManager();
+        this._skips = [];
+    }
+
     getTestBranch(id: string): TestBranch {
         return this._testsTree.getTestBranch(id);
     }
diff --git a/lib/report-builder/static.ts b/lib/report-builder/static.ts
@@ -76,6 +76,10 @@ export class StaticReportBuilder {
         this._workers = workers;
     }
 
+    flushDb(): void {
+        this._dbClient.flush();
+    }
+
     registerAttempt(testInfo: {fullName: string, browserId: string}, status: TestStatus): number {
         return this._testAttemptManager.registerAttempt(testInfo, status);
     }
diff --git a/lib/sqlite-client.ts b/lib/sqlite-client.ts
@@ -120,6 +120,10 @@ export class SqliteClient {
         return this._db;
     }
 
+    flush(): void {
+        fs.writeFileSync(this._dbPath, this._db.export());
+    }
+
     close(): void {
         fs.writeFileSync(this._dbPath, this._db.run('VACUUM').export());
 
diff --git a/lib/static/modules/action-names.ts b/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_REFRESH_LOADING: 'SET_REFRESH_LOADING',
     SET_SEARCH_LOADING: 'SET_SEARCH_LOADING'
 } as const;
diff --git a/lib/static/modules/actions/filters.ts b/lib/static/modules/actions/filters.ts
@@ -38,10 +38,16 @@ export const setSearchLoading = (payload: boolean): SetSearchLoading => {
 export type ChangeViewModeAction = Action<typeof actionNames.CHANGE_VIEW_MODE, FilterPayload<ViewMode>>;
 export const changeViewMode = (payload: ChangeViewModeAction['payload']): ChangeViewModeAction => ({type: actionNames.CHANGE_VIEW_MODE, 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 FiltersAction =
     | UpdateNameFilterAction
     | SetMatchCaseFilterAction
     | SetUseRegexFilterAction
     | SelectBrowsersAction
     | ChangeViewModeAction
+    | SetRefreshLoading
     | SetSearchLoading;
diff --git a/lib/static/modules/actions/lifecycle.ts b/lib/static/modules/actions/lifecycle.ts
@@ -18,6 +18,7 @@ import {GetInitResponse} from '@/gui/server';
 import {Tree} from '@/tests-tree-builder/base';
 import {BrowserItem} from '@/types';
 import {createNotificationError} from '@/static/modules/actions/notifications';
+import {setRefreshLoading} from '@/static/modules/actions/filters';
 import {LocalStorageKey} from '@/constants/local-storage';
 import * as localStorageWrapper from '@/static/modules/local-storage-wrapper';
 import {updateTimeTravelSettings} from '../../new-ui/utils/api';
@@ -139,6 +140,25 @@ export const thunkInitStaticReport = ({isNewUi}: InitStaticReportData = {}): App
     };
 };
 
+export const thunkRefreshGuiReport = (): AppThunk => {
+    return async (dispatch, getState) => {
+        dispatch(setRefreshLoading(true));
+        try {
+            const response = await axios.get<GetInitResponse>('/refresh-tests');
+
+            if (!response.data) {
+                throw new Error('Could not refresh tests data.');
+            }
+
+            const {db} = getState();
+            dispatch(initGuiReport({...response.data, db: db!}));
+        } catch (e: unknown) {
+            dispatch(createNotificationError('thunkRefreshGuiReport', e as Error));
+            dispatch(setRefreshLoading(false));
+        }
+    };
+};
+
 export type FinGuiReportAction = Action<typeof actionNames.FIN_GUI_REPORT>;
 export const finGuiReport = (): FinGuiReportAction => ({type: actionNames.FIN_GUI_REPORT});
 
diff --git a/lib/static/modules/reducers/filters.ts b/lib/static/modules/reducers/filters.ts
@@ -41,6 +41,7 @@ export default (state: State, action: FiltersAction | InitGuiReportAction | Init
                 state,
                 {
                     app: {
+                        isRefreshLoading: false,
                         viewMode,
                         [Page.visualChecksPage]: {
                             diffMode: visualChecksPageDiffMode
@@ -99,6 +100,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;
     }
diff --git a/lib/static/new-ui/components/AttemptPickerItem/index.tsx b/lib/static/new-ui/components/AttemptPickerItem/index.tsx
@@ -75,14 +75,14 @@ export const AttemptPickerItem = (props: AttemptPickerItemProps): ReactNode => {
 
     const isGroupingEnabled = useSelector(({view: {keyToGroupTestsBy}, app: {isNewUi, suitesPage: {currentGroupId}}}: State) => isNewUi ? Boolean(currentGroupId) : Boolean(keyToGroupTestsBy));
 
-    const status = isFailStatus(result.status) && hasUnrelatedToScreenshotsErrors((result as ResultEntityError).error) ? TestStatus.FAIL_ERROR : result.status;
-
-    const buttonStyle = getButtonStyleByStatus(status);
-
     if (!result) {
         return null;
     }
 
+    const status = isFailStatus(result.status) && hasUnrelatedToScreenshotsErrors((result as ResultEntityError).error) ? TestStatus.FAIL_ERROR : result.status;
+
+    const buttonStyle = getButtonStyleByStatus(status);
+
     const className = classNames(
         styles.attemptPickerItem,
         {[styles['attempt-picker-item--active']]: isActive},
diff --git a/lib/static/new-ui/components/TreeActionsToolbar/index.module.css b/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;
diff --git a/lib/static/new-ui/components/TreeActionsToolbar/index.tsx b/lib/static/new-ui/components/TreeActionsToolbar/index.tsx
@@ -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';
@@ -24,7 +25,9 @@ import {
     selectAll,
     setAllTreeNodesState, setTreeViewMode,
     staticAccepterStageScreenshot,
-    staticAccepterUnstageScreenshot, thunkRunTests
+    staticAccepterUnstageScreenshot,
+    thunkRunTests,
+    thunkRefreshGuiReport
 } from '@/static/modules/actions';
 import {ImageEntity, TreeViewMode} from '@/static/new-ui/types/store';
 import {CHECKED, INDETERMINATE} from '@/constants/checked-statuses';
@@ -74,6 +77,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);
@@ -192,8 +196,22 @@ export function TreeActionsToolbar({onHighlightCurrentTest, className}: TreeActi
     const pluginComponents = getExtensionPointComponents(loadedPluginConfigs, ExtensionPointName.RunTestOptions);
     const hasRunTestOptions = pluginComponents.length > 0;
 
+    const handleRefresh = useCallback(() => {
+        dispatch(thunkRefreshGuiReport());
+    }, [dispatch]);
+
     const getViewButtons = (): ReactNode => (
         <>
+            {isRunTestsAvailable && (
+                <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 && (
                 isRunning
                     ? (
diff --git a/lib/static/new-ui/features/suites/components/SuitesPage/index.tsx b/lib/static/new-ui/features/suites/components/SuitesPage/index.tsx
@@ -218,6 +218,12 @@ export function SuitesPage(): ReactNode {
         }
     }, [visibleTreeNodeIds, tree, dispatch]);
 
+    useEffect(() => {
+        if (currentBrowser && !currentBrowserEntity && isInitialized && visibleTreeNodeIds.length > 0) {
+            onSelectFirstResult();
+        }
+    }, [currentBrowser, currentBrowserEntity, isInitialized, visibleTreeNodeIds.length]);
+
     useHotkey('ArrowDown', goToNextSuite);
     useHotkey('ArrowUp', goToPrevSuite);
     useHotkey('ArrowRight', goToNextAttempt);
diff --git a/lib/static/new-ui/features/suites/selectors.ts b/lib/static/new-ui/features/suites/selectors.ts
@@ -18,7 +18,11 @@ export const getCurrentResultId = (state: State): string | null => {
         return null;
     }
 
-    const resultIds = state.tree.browsers.byId[browserId].resultIds;
+    const browser = state.tree.browsers.byId[browserId];
+    if (!browser) {
+        return null;
+    }
+    const resultIds = browser.resultIds;
 
     const groupId = state.app.suitesPage.currentGroupId;
 
@@ -90,7 +94,7 @@ export const getAttempt = (state: State): number | null => {
     const browserId = state.app.suitesPage.currentBrowserId;
 
     if (browserId) {
-        return state.tree.browsers.stateById[browserId].retryIndex;
+        return state.tree.browsers.stateById[browserId]?.retryIndex;
     }
 
     return null;
diff --git a/lib/static/new-ui/features/visual-checks/selectors.ts b/lib/static/new-ui/features/visual-checks/selectors.ts
@@ -172,7 +172,7 @@ export const getAttempt = (state: State): number | null => {
     const namedImage = getCurrentNamedImage(state);
 
     if (namedImage) {
-        return state.tree.browsers.stateById[namedImage?.browserId].retryIndex;
+        return state.tree.browsers.stateById[namedImage?.browserId]?.retryIndex;
     }
 
     return null;
diff --git a/lib/static/new-ui/types/store.ts b/lib/static/new-ui/types/store.ts
@@ -260,6 +260,7 @@ export interface State {
         isInitialized: boolean;
         availableFeatures: Feature[],
         isSearchLoading?: boolean;
+        isRefreshLoading?: boolean;
 
         // Filters in top of sidebar
         nameFilter: string;

Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,11 @@ export class App {`
`53`	`53`	`return this._toolRunner.findEqualDiffs(data);`
`54`	`54`	`}`
`55`	`55`
	`56`	`+ async refreshTests(): Promise<ToolRunnerTree \| null> {`
	`57`	`+ await this._toolRunner.refreshTests();`
	`58`	`+ return this._toolRunner.tree;`
	`59`	`+ }`
	`60`	`+`
`56`	`61`	`addClient(connection: Response): void {`
`57`	`62`	`this._toolRunner.addClient(connection);`
`58`	`63`	`}`
Original file line number	Diff line number	Diff line change
`@@ -76,6 +76,10 @@ export class StaticReportBuilder {`
`76`	`76`	`this._workers = workers;`
`77`	`77`	`}`
`78`	`78`
	`79`	`+ flushDb(): void {`
	`80`	`+ this._dbClient.flush();`
	`81`	`+ }`
	`82`	`+`
`79`	`83`	`registerAttempt(testInfo: {fullName: string, browserId: string}, status: TestStatus): number {`
`80`	`84`	`return this._testAttemptManager.registerAttempt(testInfo, status);`
`81`	`85`	`}`
Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,10 @@ export class SqliteClient {`
`120`	`120`	`return this._db;`
`121`	`121`	`}`
`122`	`122`
	`123`	`+ flush(): void {`
	`124`	`+ fs.writeFileSync(this._dbPath, this._db.export());`
	`125`	`+ }`
	`126`	`+`
`123`	`127`	`close(): void {`
`124`	`128`	`fs.writeFileSync(this._dbPath, this._db.run('VACUUM').export());`
`125`	`129`