unleash.unleash/frontend/src/component/common/FeatureArchiveDialog/FeatureArchiveDialog.test.tsx

import { vi } from 'vitest';
import React from 'react';
import { screen, waitFor } from '@testing-library/react';
import { render } from 'utils/testRenderer';
import { testServerRoute, testServerSetup } from 'utils/testServer';
import { FeatureArchiveDialog } from './FeatureArchiveDialog';

const server = testServerSetup();
const setupHappyPathForChangeRequest = () => {
    testServerRoute(
        server,
        '/api/admin/projects/projectId/environments/development/change-requests',
        {},
        'post',
    );
    testServerRoute(
        server,
        '/api/admin/projects/projectId/change-requests/config',
        [
            {
                environment: 'development',
                type: 'development',
                requiredApprovals: 1,
                changeRequestEnabled: true,
            },
        ],
    );
};
const setupArchiveValidation = (orphanParents: string[]) => {
    testServerRoute(server, '/api/admin/ui-config', {
        versionInfo: {
            current: { oss: 'version', enterprise: 'version' },
        },
    });
    testServerRoute(
        server,
        '/api/admin/projects/projectId/archive/validate',
        {
            hasDeletedDependencies: true,
            parentsWithChildFeatures: orphanParents,
        },
        'post',
    );
};

test('Add single archive feature change to change request', async () => {
    const onClose = vi.fn();
    const onConfirm = vi.fn();
    setupHappyPathForChangeRequest();
    setupArchiveValidation([]);
    render(
        <FeatureArchiveDialog
            featureIds={['featureA']}
            projectId={'projectId'}
            isOpen={true}
            onClose={onClose}
            onConfirm={onConfirm}
            featuresWithUsage={[]}
        />,
    );

    expect(screen.getByText('Archive feature toggle')).toBeInTheDocument();
    await screen.findByText(
        'Archiving features with dependencies will also remove those dependencies.',
    );
    const button = await screen.findByText('Add change to draft');

    button.click();

    await waitFor(() => {
        expect(onConfirm).toBeCalledTimes(1);
    });
    expect(onClose).toBeCalledTimes(1);
});

test('Add multiple archive feature changes to change request', async () => {
    const onClose = vi.fn();
    const onConfirm = vi.fn();
    setupHappyPathForChangeRequest();
    setupArchiveValidation([]);
    render(
        <FeatureArchiveDialog
            featureIds={['featureA', 'featureB']}
            projectId={'projectId'}
            isOpen={true}
            onClose={onClose}
            onConfirm={onConfirm}
            featuresWithUsage={[]}
        />,
    );

    await screen.findByText('Archive feature toggles');
    await screen.findByText(
        'Archiving features with dependencies will also remove those dependencies.',
    );
    const button = await screen.findByText('Add to change request');

    button.click();

    await waitFor(() => {
        expect(onConfirm).toBeCalledTimes(1);
    });
    expect(onClose).toBeCalledTimes(1);
});

test('Skip change request', async () => {
    const onClose = vi.fn();
    const onConfirm = vi.fn();
    setupHappyPathForChangeRequest();
    setupArchiveValidation([]);
    render(
        <FeatureArchiveDialog
            featureIds={['featureA', 'featureB']}
            projectId={'projectId'}
            isOpen={true}
            onClose={onClose}
            onConfirm={onConfirm}
            featuresWithUsage={[]}
        />,
        { permissions: [{ permission: 'SKIP_CHANGE_REQUEST' }] },
    );

    await screen.findByText('Archive feature toggles');
    const button = await screen.findByText('Archive toggles');

    await waitFor(() => expect(button).toBeEnabled());
    button.click();

    await waitFor(() => {
        expect(onClose).toBeCalledTimes(1);
    });
    expect(onConfirm).toBeCalledTimes(0); // we didn't setup non Change Request flow so failure
});

test('Show error message when multiple parents of orphaned children are archived', async () => {
    const onClose = vi.fn();
    const onConfirm = vi.fn();
    setupArchiveValidation(['parentA', 'parentB']);
    render(
        <FeatureArchiveDialog
            featureIds={['parentA', 'parentB']}
            projectId={'projectId'}
            isOpen={true}
            onClose={onClose}
            onConfirm={onConfirm}
            featuresWithUsage={[]}
        />,
    );

    await screen.findByText('2 feature toggles');
    await screen.findByText(
        'have child features that depend on them and are not part of the archive operation. These parent features can not be archived:',
    );
    expect(
        screen.queryByText(
            'Archiving features with dependencies will also remove those dependencies.',
        ),
    ).not.toBeInTheDocument();
});

test('Show error message when 1 parent of orphaned children is archived', async () => {
    const onClose = vi.fn();
    const onConfirm = vi.fn();
    setupArchiveValidation(['parent']);
    render(
        <FeatureArchiveDialog
            featureIds={['parent', 'someOtherFeature']}
            projectId={'projectId'}
            isOpen={true}
            onClose={onClose}
            onConfirm={onConfirm}
            featuresWithUsage={[]}
        />,
    );

    await screen.findByText('parent');
    await screen.findByText(
        'has child features that depend on it and are not part of the archive operation.',
    );
    expect(
        screen.queryByText(
            'Archiving features with dependencies will also remove those dependencies.',
        ),
    ).not.toBeInTheDocument();
});