geo/apps/desktop-client/src/main/publish-scheduler.test.ts

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

import {
  enqueuePublishTask,
  getPublishSchedulerSnapshot,
  initPublishScheduler,
  notePublishTaskActivated,
  notePublishTaskCompleted,
  selectNextPublishTask,
} from './publish-scheduler'

describe('publish scheduler', () => {
  beforeEach(() => {
    vi.useFakeTimers()
    vi.setSystemTime(new Date('2026-04-24T00:00:00.000Z'))
    initPublishScheduler()
  })

  afterEach(() => {
    vi.useRealTimers()
    initPublishScheduler()
  })

  it('skips same-platform queued tasks while selecting an allowed platform', () => {
    enqueuePublishTask({
      taskId: 'toutiao-active',
      platform: 'toutiao',
      routing: 'rabbitmq-primary',
    })
    notePublishTaskActivated('toutiao-active', 'toutiao')
    enqueuePublishTask({ taskId: 'toutiao-next', platform: 'toutiao', routing: 'rabbitmq-primary' })
    enqueuePublishTask({ taskId: 'zhihu-next', platform: 'zhihu', routing: 'rabbitmq-primary' })

    const selected = selectNextPublishTask({
      maxConcurrency: 2,
      activeCount: 1,
    })

    expect(selected?.taskId).toBe('zhihu-next')
    expect(getPublishSchedulerSnapshot().leasingCount).toBe(1)
  })

  it('holds same-platform tasks during completion cooldown', () => {
    enqueuePublishTask({ taskId: 'first', platform: 'toutiao', routing: 'rabbitmq-primary' })
    expect(selectNextPublishTask({ maxConcurrency: 2, activeCount: 0 })?.taskId).toBe('first')
    notePublishTaskActivated('first', 'toutiao')
    notePublishTaskCompleted('first', 'toutiao')

    enqueuePublishTask({ taskId: 'second', platform: 'toutiao', routing: 'rabbitmq-primary' })

    expect(selectNextPublishTask({ maxConcurrency: 2, activeCount: 0 })).toBeNull()

    vi.advanceTimersByTime(6_001)

    expect(selectNextPublishTask({ maxConcurrency: 2, activeCount: 0 })?.taskId).toBe('second')
  })
})