3x-ui/frontend/src/test/geodata-section.test.tsx

import { render, screen, waitFor } from '@testing-library/react';
import userEvent from '@testing-library/user-event';
import { afterEach, describe, expect, it, vi } from 'vitest';

import GeodataSection from '@/pages/index/GeodataSection';
import { HttpUtil, Msg } from '@/utils';

const STANDARD_SOURCES = [
  {
    url: 'https://github.com/Loyalsoldier/v2ray-rules-dat/releases/latest/download/geoip.dat',
    file: 'geoip.dat',
  },
  {
    url: 'https://github.com/Loyalsoldier/v2ray-rules-dat/releases/latest/download/geosite.dat',
    file: 'geosite.dat',
  },
  {
    url: 'https://github.com/chocolate4u/Iran-v2ray-rules/releases/latest/download/geoip.dat',
    file: 'geoip_IR.dat',
  },
  {
    url: 'https://github.com/chocolate4u/Iran-v2ray-rules/releases/latest/download/geosite.dat',
    file: 'geosite_IR.dat',
  },
  {
    url: 'https://github.com/runetfreedom/russia-v2ray-rules-dat/releases/latest/download/geoip.dat',
    file: 'geoip_RU.dat',
  },
  {
    url: 'https://github.com/runetfreedom/russia-v2ray-rules-dat/releases/latest/download/geosite.dat',
    file: 'geosite_RU.dat',
  },
];

const CUSTOM_SOURCE = {
  url: 'https://example.com/geosite_custom.dat',
  file: 'geosite_custom.dat',
};

afterEach(() => {
  vi.restoreAllMocks();
});

describe('GeodataSection', () => {
  it('adds the standard sources without removing custom sources', async () => {
    vi.spyOn(HttpUtil, 'post').mockResolvedValue(
      new Msg(
        true,
        '',
        JSON.stringify({
          xraySetting: { outbounds: [], geodata: { assets: [CUSTOM_SOURCE] } },
          geodataSources: STANDARD_SOURCES,
        }),
      ),
    );
    const user = userEvent.setup();

    render(<GeodataSection active onBusy={vi.fn()} onClose={vi.fn()} />);

    await screen.findByRole('button', {
      name: 'Use standard sources',
    });
    await waitFor(() =>
      expect(
        (screen.getByRole('button', { name: 'Use standard sources' }) as HTMLButtonElement)
          .disabled,
      ).toBe(false),
    );
    await user.click(screen.getByRole('button', { name: 'Use standard sources' }));

    await waitFor(() => {
      expect(screen.getByDisplayValue(CUSTOM_SOURCE.url)).toBeTruthy();
      for (const source of STANDARD_SOURCES) {
        expect(screen.getByDisplayValue(source.url)).toBeTruthy();
        expect(screen.getByDisplayValue(source.file)).toBeTruthy();
      }
    });
  });
});