ComfyUI_frontend/scripts/qa-video-review.test.ts

import { describe, expect, it } from 'vitest'

import {
  extractPlatformFromArtifactDirName,
  pickLatestVideosByPlatform,
  selectVideoCandidateByFile
} from './qa-video-review'

describe('extractPlatformFromArtifactDirName', () => {
  it('extracts and normalizes known qa artifact directory names', () => {
    expect(
      extractPlatformFromArtifactDirName('qa-report-Windows-22818315023')
    ).toBe('windows')
    expect(
      extractPlatformFromArtifactDirName('qa-report-macOS-22818315023')
    ).toBe('macos')
    expect(
      extractPlatformFromArtifactDirName('qa-report-Linux-22818315023')
    ).toBe('linux')
  })

  it('falls back to slugifying unknown directory names', () => {
    expect(extractPlatformFromArtifactDirName('custom platform run')).toBe(
      'custom-platform-run'
    )
  })
})

describe('pickLatestVideosByPlatform', () => {
  it('keeps only the latest candidate per platform', () => {
    const selected = pickLatestVideosByPlatform([
      {
        platformName: 'windows',
        videoPath: '/tmp/windows-old.mp4',
        mtimeMs: 100
      },
      {
        platformName: 'windows',
        videoPath: '/tmp/windows-new.mp4',
        mtimeMs: 200
      },
      {
        platformName: 'linux',
        videoPath: '/tmp/linux.mp4',
        mtimeMs: 150
      }
    ])

    expect(selected).toEqual([
      {
        platformName: 'linux',
        videoPath: '/tmp/linux.mp4',
        mtimeMs: 150
      },
      {
        platformName: 'windows',
        videoPath: '/tmp/windows-new.mp4',
        mtimeMs: 200
      }
    ])
  })
})

describe('selectVideoCandidateByFile', () => {
  it('selects a single candidate by artifacts-relative path', () => {
    const selected = selectVideoCandidateByFile(
      [
        {
          platformName: 'windows',
          videoPath: '/tmp/qa-artifacts/qa-report-Windows-1/qa-session.mp4',
          mtimeMs: 100
        },
        {
          platformName: 'linux',
          videoPath: '/tmp/qa-artifacts/qa-report-Linux-1/qa-session.mp4',
          mtimeMs: 200
        }
      ],
      {
        artifactsDir: '/tmp/qa-artifacts',
        videoFile: 'qa-report-Linux-1/qa-session.mp4'
      }
    )

    expect(selected).toEqual({
      platformName: 'linux',
      videoPath: '/tmp/qa-artifacts/qa-report-Linux-1/qa-session.mp4',
      mtimeMs: 200
    })
  })

  it('throws when basename matches multiple videos', () => {
    expect(() =>
      selectVideoCandidateByFile(
        [
          {
            platformName: 'windows',
            videoPath: '/tmp/qa-artifacts/qa-report-Windows-1/qa-session.mp4',
            mtimeMs: 100
          },
          {
            platformName: 'linux',
            videoPath: '/tmp/qa-artifacts/qa-report-Linux-1/qa-session.mp4',
            mtimeMs: 200
          }
        ],
        {
          artifactsDir: '/tmp/qa-artifacts',
          videoFile: 'qa-session.mp4'
        }
      )
    ).toThrow('matched 2 videos')
  })

  it('throws when there is no matching video', () => {
    expect(() =>
      selectVideoCandidateByFile(
        [
          {
            platformName: 'windows',
            videoPath: '/tmp/qa-artifacts/qa-report-Windows-1/qa-session.mp4',
            mtimeMs: 100
          }
        ],
        {
          artifactsDir: '/tmp/qa-artifacts',
          videoFile: 'qa-report-macOS-1/qa-session.mp4'
        }
      )
    ).toThrow('No video matched')
  })

  it('throws when video file is missing', () => {
    expect(() =>
      selectVideoCandidateByFile(
        [
          {
            platformName: 'windows',
            videoPath: '/tmp/qa-artifacts/qa-report-Windows-1/qa-session.mp4',
            mtimeMs: 100
          }
        ],
        {
          artifactsDir: '/tmp/qa-artifacts',
          videoFile: '   '
        }
      )
    ).toThrow('--video-file is required')
  })
})