import { describe, expect, test } from 'vitest';

import {
  resolveMiniGameGenerationProgressTickState,
} from './PlatformEntryFlowShellImpl';
import { createMiniGameDraftGenerationState } from '../../services/miniGameDraftGenerationProgress';

describe('resolveMiniGameGenerationProgressTickState', () => {
  test('returns jump hop and wooden fish generation states for progress ticking', () => {
    const jumpHopState = createMiniGameDraftGenerationState('jump-hop');
    const woodenFishState = createMiniGameDraftGenerationState('wooden-fish');

    expect(
      resolveMiniGameGenerationProgressTickState('jump-hop-generating', {
        'jump-hop': jumpHopState,
      }),
    ).toBe(jumpHopState);
    expect(
      resolveMiniGameGenerationProgressTickState('wooden-fish-generating', {
        'wooden-fish': woodenFishState,
      }),
    ).toBe(woodenFishState);
  });

  test('returns null when the stage does not need generation ticking', () => {
    expect(
      resolveMiniGameGenerationProgressTickState('platform', {
        'jump-hop': createMiniGameDraftGenerationState('jump-hop'),
      }),
    ).toBeNull();
  });
});