src/containers/Storage/__test__/useStorageQueryParams.test.ts - main - githubmirror/ydb-embedded-ui

src/containers/Storage/__test__/useStorageQueryParams.test.ts
133 строки5 KB
Anton Standrik
feat(storage): add blob storage capacity metrics (#4167)
05 авг 2026, 17:41
05 авг 2026, 17:412881245
import {renderHook} from '@testing-library/react';

import {SETTING_KEYS} from '../../../store/reducers/settings/constants';
import {getStorageGroupByCleanupPatch, useIsStorageExpertMode} from '../useStorageQueryParams';

jest.mock('use-query-params', () => ({
    BooleanParam: {},
    StringParam: {},
    useQueryParam: jest.fn(),
    useQueryParams: jest.fn(),
}));

jest.mock('../../../store/reducers/capabilities/hooks', () => ({
    useBlobStorageCapacityMetricsAvailable: jest.fn(),
    useBlobStorageCapacityMetricsEnabled: jest.fn(),
}));

jest.mock('../../../utils/hooks', () => ({
    useSetting: jest.fn(),
}));

jest.mock('../../../utils/hooks/useIsUserAllowedToMakeChanges', () => ({
    useIsUserAllowedToMakeChanges: jest.fn(),
}));

const {useQueryParam} = jest.requireMock('use-query-params');
const {useBlobStorageCapacityMetricsAvailable} = jest.requireMock(
    '../../../store/reducers/capabilities/hooks',
);
const {useSetting} = jest.requireMock('../../../utils/hooks');
const {useIsUserAllowedToMakeChanges} = jest.requireMock(
    '../../../utils/hooks/useIsUserAllowedToMakeChanges',
);

describe('getStorageGroupByCleanupPatch', () => {
    test('clears legacy group-by values when blob metrics are enabled', () => {
        expect(
            getStorageGroupByCleanupPatch({
                blobMetricsEnabled: true,
                storageGroupsGroupBy: 'Usage',
                storageNodesGroupBy: 'DiskSpaceUsage',
            }),
        ).toStrictEqual({
            storageGroupsGroupBy: undefined,
            storageNodesGroupBy: undefined,
        });
    });

    test('keeps capacity alert group-by values when blob metrics are enabled', () => {
        expect(
            getStorageGroupByCleanupPatch({
                blobMetricsEnabled: true,
                storageGroupsGroupBy: 'CapacityAlert',
                storageNodesGroupBy: 'CapacityAlert',
            }),
        ).toStrictEqual({});
    });

    test('clears legacy disk usage group-by for groups when blob metrics are enabled', () => {
        expect(
            getStorageGroupByCleanupPatch({
                blobMetricsEnabled: true,
                storageGroupsGroupBy: 'DiskSpaceUsage',
            }),
        ).toStrictEqual({storageGroupsGroupBy: undefined});
    });

    test('clears capacity alert group-by values when blob metrics are disabled', () => {
        expect(
            getStorageGroupByCleanupPatch({
                blobMetricsEnabled: false,
                storageGroupsGroupBy: 'CapacityAlert',
                storageNodesGroupBy: 'CapacityAlert',
            }),
        ).toStrictEqual({
            storageGroupsGroupBy: undefined,
            storageNodesGroupBy: undefined,
        });
    });

    test('keeps legacy node disk usage group-by when blob metrics are disabled', () => {
        expect(
            getStorageGroupByCleanupPatch({
                blobMetricsEnabled: false,
                storageNodesGroupBy: 'DiskSpaceUsage',
            }),
        ).toStrictEqual({});
    });
});

describe('useIsStorageExpertMode', () => {
    beforeEach(() => {
        jest.clearAllMocks();

        useQueryParam.mockReturnValue([true]);
        useSetting.mockImplementation((key: string) => [
            key === SETTING_KEYS.ENABLE_STORAGE_EXPERT_MODE ||
                key === SETTING_KEYS.STORAGE_EXPERT_MODE,
        ]);
        useIsUserAllowedToMakeChanges.mockReturnValue(true);
    });

    test.each([
        {available: false, expected: false},
        {available: true, expected: true},
    ])(
        'returns $expected when requested mode availability is $available',
        ({available, expected}) => {
            useBlobStorageCapacityMetricsAvailable.mockReturnValue(available);

            const {result} = renderHook(() => useIsStorageExpertMode());

            expect(result.current).toBe(expected);
        },
    );

    test.each([
        {source: 'query parameter', queryValue: true, savedValue: false},
        {source: 'saved setting', queryValue: undefined, savedValue: true},
    ])('returns false for unauthorized users with $source state', ({queryValue, savedValue}) => {
        useBlobStorageCapacityMetricsAvailable.mockReturnValue(true);
        useQueryParam.mockReturnValue([queryValue]);
        useSetting.mockImplementation((key: string) => [
            key === SETTING_KEYS.ENABLE_STORAGE_EXPERT_MODE ||
                (key === SETTING_KEYS.STORAGE_EXPERT_MODE && savedValue),
        ]);
        useIsUserAllowedToMakeChanges.mockReturnValue(false);

        const {result} = renderHook(() => useIsStorageExpertMode());

        expect(result.current).toBe(false);
    });
});