ComfyUI-Lora-Manager/tests/frontend/managers/RematchModalManager.test.js

import { describe, it, beforeEach, afterEach, expect, vi } from 'vitest';

const showToastMock = vi.fn();
const translateMock = vi.fn((key, params, fallback) => (typeof fallback === 'string' ? fallback : key));
const modalManagerMock = {
  showModal: vi.fn(),
  closeModal: vi.fn(),
};

vi.mock('../../../static/js/managers/ModalManager.js', () => ({
  modalManager: modalManagerMock,
}));

vi.mock('../../../static/js/utils/i18nHelpers.js', () => ({
  translate: translateMock,
}));

vi.mock('../../../static/js/utils/uiHelpers.js', () => ({
  showToast: showToastMock,
}));

async function getManager() {
  const { rematchModalManager } = await import(
    '../../../static/js/managers/RematchModalManager.js'
  );
  return rematchModalManager;
}

describe('RematchModalManager options dialog', () => {
  beforeEach(() => {
    vi.clearAllMocks();
    document.body.innerHTML = `
      <p id="rematchOptionsMessage"></p>
      <input type="checkbox" id="rematchOptionsRelaxed">
    `;
  });

  afterEach(() => {
    document.body.innerHTML = '';
  });

  it('does not invoke the callback until confirmOptions is called', async () => {
    const manager = await getManager();
    const onConfirm = vi.fn();

    manager.showOptionsModal({ recipeCount: 3, onConfirm });

    expect(modalManagerMock.showModal).toHaveBeenCalledWith('rematchOptionsModal');
    expect(onConfirm).not.toHaveBeenCalled();
    // Bulk message mentions the selection size.
    expect(document.getElementById('rematchOptionsMessage').textContent).toContain('3');
    // The checkbox always starts unchecked.
    expect(document.getElementById('rematchOptionsRelaxed').checked).toBe(false);

    manager.confirmOptions();
    expect(onConfirm).toHaveBeenCalledWith({ relaxed: false });
    expect(modalManagerMock.closeModal).toHaveBeenCalledWith('rematchOptionsModal');
  });

  it('uses the generic message when no recipe count is given', async () => {
    const manager = await getManager();

    manager.showOptionsModal({ onConfirm: vi.fn() });

    expect(translateMock).toHaveBeenCalledWith(
      'modals.rematchOptions.messageGlobal',
      {},
      'All recipes will be scanned against your local model library.'
    );
  });

  it('uses the single-recipe message for scope: single', async () => {
    const manager = await getManager();

    manager.showOptionsModal({ scope: 'single', onConfirm: vi.fn() });

    expect(translateMock).toHaveBeenCalledWith(
      'modals.rematchOptions.messageSingle',
      {},
      'This recipe will be scanned against your local model library.'
    );
  });

  it('passes relaxed: true when the checkbox is checked', async () => {
    const manager = await getManager();
    const onConfirm = vi.fn();

    manager.showOptionsModal({ onConfirm });
    document.getElementById('rematchOptionsRelaxed').checked = true;
    manager.confirmOptions();

    expect(onConfirm).toHaveBeenCalledWith({ relaxed: true });
  });

  it('resets the checkbox to unchecked each time the dialog opens', async () => {
    const manager = await getManager();
    const checkbox = document.getElementById('rematchOptionsRelaxed');
    checkbox.checked = true;

    manager.showOptionsModal({ onConfirm: vi.fn() });

    expect(checkbox.checked).toBe(false);
  });

  it('cancelOptions runs nothing and clears the callback', async () => {
    const manager = await getManager();
    const onConfirm = vi.fn();

    manager.showOptionsModal({ onConfirm });
    manager.cancelOptions();

    expect(modalManagerMock.closeModal).toHaveBeenCalledWith('rematchOptionsModal');
    // A later confirm must not fire the cancelled callback.
    manager.confirmOptions();
    expect(onConfirm).not.toHaveBeenCalled();
  });
});

describe('RematchModalManager results modal', () => {
  beforeEach(() => {
    vi.clearAllMocks();
    document.body.innerHTML = '<ul id="rematchResultsList"></ul>';
    global.fetch = vi.fn();
  });

  afterEach(() => {
    document.body.innerHTML = '';
    delete global.fetch;
  });

  it('does nothing for an empty match list', async () => {
    const manager = await getManager();

    manager.showResultsModal([]);

    expect(modalManagerMock.showModal).not.toHaveBeenCalled();
  });

  it('renders one row per L4 match with entry, file name and recipe', async () => {
    const manager = await getManager();

    manager.showResultsModal([
      { recipe_id: 'r1', type: 'lora', entry: 'old.safetensors', file_name: 'new.safetensors', lora_index: 2 },
      { recipe_id: 'r2', type: 'checkpoint', entry: 'cp-old', file_name: 'cp-new.safetensors' },
    ]);

    const rows = document.querySelectorAll('#rematchResultsList .rematch-results-row');
    expect(rows).toHaveLength(2);
    expect(rows[0].textContent).toContain('old.safetensors');
    expect(rows[0].textContent).toContain('new.safetensors');
    expect(rows[0].textContent).toContain('r1');
    expect(rows[1].textContent).toContain('cp-new.safetensors');
    expect(modalManagerMock.showModal).toHaveBeenCalledWith('rematchResultsModal');
  });

  it('undo posts to the lora restore endpoint and disables the row', async () => {
    const manager = await getManager();
    global.fetch.mockResolvedValue({
      ok: true,
      json: async () => ({ success: true }),
    });

    manager.showResultsModal([
      { recipe_id: 'r1', type: 'lora', entry: 'old.safetensors', file_name: 'new.safetensors', lora_index: 2 },
    ]);

    const row = document.querySelector('.rematch-results-row');
    const button = row.querySelector('.rematch-results-undo');
    button.click();
    await vi.waitFor(() => expect(button.disabled).toBe(true));

    expect(global.fetch).toHaveBeenCalledWith('/api/lm/recipe/lora/restore', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ recipe_id: 'r1', lora_index: 2 }),
    });
    expect(row.classList.contains('undone')).toBe(true);
  });

  it('undo posts to the checkpoint restore endpoint with recipe_id only', async () => {
    const manager = await getManager();
    global.fetch.mockResolvedValue({
      ok: true,
      json: async () => ({ success: true }),
    });

    manager.showResultsModal([
      { recipe_id: 'r2', type: 'checkpoint', entry: 'cp-old', file_name: 'cp-new.safetensors' },
    ]);

    const button = document.querySelector('.rematch-results-undo');
    button.click();
    await vi.waitFor(() => expect(button.disabled).toBe(true));

    expect(global.fetch).toHaveBeenCalledWith('/api/lm/recipe/checkpoint/restore', {
      method: 'POST',
      headers: { 'Content-Type': 'application/json' },
      body: JSON.stringify({ recipe_id: 'r2' }),
    });
  });

  it('keeps the row actionable and toasts when undo fails', async () => {
    const manager = await getManager();
    global.fetch.mockResolvedValue({
      ok: true,
      json: async () => ({ success: false, error: 'no snapshot' }),
    });

    manager.showResultsModal([
      { recipe_id: 'r1', type: 'lora', entry: 'old.safetensors', file_name: 'new.safetensors', lora_index: 0 },
    ]);

    const row = document.querySelector('.rematch-results-row');
    const button = row.querySelector('.rematch-results-undo');
    button.click();
    await vi.waitFor(() => expect(showToastMock).toHaveBeenCalled());

    expect(button.disabled).toBe(false);
    expect(row.classList.contains('undone')).toBe(false);
    expect(showToastMock).toHaveBeenCalledWith(
      'modals.rematchResults.undoFailed',
      { message: 'no snapshot' },
      'error'
    );
  });
});