ComfyUI_frontend/browser_tests/tests/releaseNotifications.spec.ts

import { expect } from '@playwright/test'

import type { components } from '@comfyorg/registry-types'

type ReleaseNote = components['schemas']['ReleaseNote']
import { comfyPageFixture as test } from '@e2e/fixtures/ComfyPage'
import { TestIds } from '@e2e/fixtures/selectors'

function createMockRelease(overrides?: Partial<ReleaseNote>): ReleaseNote {
  return {
    id: 1,
    project: 'comfyui',
    version: 'v0.3.44',
    attention: 'medium',
    content: '## New Features\n\n- Added awesome feature',
    published_at: new Date().toISOString(),
    ...overrides
  }
}

test.describe('Release Notifications', () => {
  test.beforeEach(async ({ comfyPage }) => {
    await comfyPage.settings.setSetting('Comfy.UseNewMenu', 'Top')
  })

  test('should show help center with release information', async ({
    comfyPage
  }) => {
    // Mock release API with test data instead of empty array
    await comfyPage.page.route('**/releases**', async (route) => {
      const url = route.request().url()
      if (
        url.includes('api.comfy.org') ||
        url.includes('stagingapi.comfy.org')
      ) {
        await route.fulfill({
          status: 200,
          contentType: 'application/json',
          body: JSON.stringify([
            createMockRelease({
              content:
                '## New Features\n\n- Added awesome feature\n- Fixed important bug'
            })
          ])
        })
      } else {
        await route.continue()
      }
    })

    // Setup with release mocking disabled for this test
    await comfyPage.setup({ mockReleases: false })

    // Open help center
    const helpCenterButton = comfyPage.page.locator('.comfy-help-center-btn')
    await helpCenterButton.waitFor({ state: 'visible' })
    await helpCenterButton.click()

    // Verify help center menu appears
    const helpMenu = comfyPage.page.locator('.help-center-menu')
    await expect(helpMenu).toBeVisible()

    // Verify "What's New?" section shows the release
    const whatsNewSection = comfyPage.page.getByTestId(
      TestIds.dialogs.whatsNewSection
    )
    await expect(whatsNewSection).toBeVisible()

    // Should show the release version
    await expect(
      whatsNewSection.locator('text=Comfy v0.3.44 Release')
    ).toBeVisible()

    // Close help center by dismissable mask
    await comfyPage.page.locator('.help-center-backdrop').click()
    await expect(helpMenu).toBeHidden()
  })

  test('should not show release notifications when mocked (default behavior)', async ({
    comfyPage
  }) => {
    // Use default setup (mockReleases: true)
    await comfyPage.setup()

    // Open help center
    const helpCenterButton = comfyPage.page.locator('.comfy-help-center-btn')
    await helpCenterButton.waitFor({ state: 'visible' })
    await helpCenterButton.click()

    // Verify help center menu appears
    const helpMenu = comfyPage.page.locator('.help-center-menu')
    await expect(helpMenu).toBeVisible()

    // Verify "What's New?" section shows no releases
    const whatsNewSection = comfyPage.page.getByTestId(
      TestIds.dialogs.whatsNewSection
    )
    await expect(whatsNewSection).toBeVisible()

    // Should show "No recent releases" message
    await expect(
      whatsNewSection.locator('text=No recent releases')
    ).toBeVisible()

    // Should not show any popups or toasts
    await expect(comfyPage.page.locator('.whats-new-popup')).toBeHidden()
    await expect(
      comfyPage.page.locator('.release-notification-toast')
    ).toBeHidden()
  })

  test('should handle release API errors gracefully', async ({ comfyPage }) => {
    // Mock API to return an error
    await comfyPage.page.route('**/releases**', async (route) => {
      const url = route.request().url()
      if (
        url.includes('api.comfy.org') ||
        url.includes('stagingapi.comfy.org')
      ) {
        await route.fulfill({
          status: 500,
          contentType: 'application/json',
          body: JSON.stringify({ error: 'Server error' })
        })
      } else {
        await route.continue()
      }
    })

    // Setup with release mocking disabled
    await comfyPage.setup({ mockReleases: false })

    // Open help center
    const helpCenterButton = comfyPage.page.locator('.comfy-help-center-btn')
    await helpCenterButton.waitFor({ state: 'visible' })
    await helpCenterButton.click()

    // Verify help center still works despite API error
    const helpMenu = comfyPage.page.locator('.help-center-menu')
    await expect(helpMenu).toBeVisible()

    // Should show no releases due to error
    const whatsNewSection = comfyPage.page.getByTestId(
      TestIds.dialogs.whatsNewSection
    )
    await expect(
      whatsNewSection.locator('text=No recent releases')
    ).toBeVisible()
  })

  test('should hide "What\'s New" section when notifications are disabled', async ({
    comfyPage
  }) => {
    // Disable version update notifications
    await comfyPage.settings.setSetting(
      'Comfy.Notification.ShowVersionUpdates',
      false
    )

    // Mock release API with test data
    await comfyPage.page.route('**/releases**', async (route) => {
      const url = route.request().url()
      if (
        url.includes('api.comfy.org') ||
        url.includes('stagingapi.comfy.org')
      ) {
        await route.fulfill({
          status: 200,
          contentType: 'application/json',
          body: JSON.stringify([createMockRelease({ attention: 'high' })])
        })
      } else {
        await route.continue()
      }
    })

    await comfyPage.setup({ mockReleases: false })

    // Open help center
    const helpCenterButton = comfyPage.page.locator('.comfy-help-center-btn')
    await helpCenterButton.waitFor({ state: 'visible' })
    await helpCenterButton.click()

    // Verify help center menu appears
    const helpMenu = comfyPage.page.locator('.help-center-menu')
    await expect(helpMenu).toBeVisible()

    // Verify "What's New?" section is hidden
    const whatsNewSection = comfyPage.page.getByTestId(
      TestIds.dialogs.whatsNewSection
    )
    await expect(whatsNewSection).toBeHidden()

    // Should not show any popups or toasts
    await expect(comfyPage.page.locator('.whats-new-popup')).toBeHidden()
    await expect(
      comfyPage.page.locator('.release-notification-toast')
    ).toBeHidden()
  })

  test('should not make API calls when notifications are disabled', async ({
    comfyPage
  }) => {
    // Disable version update notifications
    await comfyPage.settings.setSetting(
      'Comfy.Notification.ShowVersionUpdates',
      false
    )

    // Track API calls
    let apiCallCount = 0
    await comfyPage.page.route('**/releases**', async (route) => {
      const url = route.request().url()
      if (
        url.includes('api.comfy.org') ||
        url.includes('stagingapi.comfy.org')
      ) {
        apiCallCount++
        await route.fulfill({
          status: 200,
          contentType: 'application/json',
          body: JSON.stringify([])
        })
      } else {
        await route.continue()
      }
    })

    await comfyPage.setup({ mockReleases: false })

    // Verify no API calls were made
    expect(apiCallCount).toBe(0)
  })

  test('should show "What\'s New" section when notifications are enabled', async ({
    comfyPage
  }) => {
    // Enable version update notifications (default behavior)
    await comfyPage.settings.setSetting(
      'Comfy.Notification.ShowVersionUpdates',
      true
    )

    // Mock release API with test data
    await comfyPage.page.route('**/releases**', async (route) => {
      const url = route.request().url()
      if (
        url.includes('api.comfy.org') ||
        url.includes('stagingapi.comfy.org')
      ) {
        await route.fulfill({
          status: 200,
          contentType: 'application/json',
          body: JSON.stringify([createMockRelease()])
        })
      } else {
        await route.continue()
      }
    })

    await comfyPage.setup({ mockReleases: false })

    // Open help center
    const helpCenterButton = comfyPage.page.locator('.comfy-help-center-btn')
    await helpCenterButton.waitFor({ state: 'visible' })
    await helpCenterButton.click()

    // Verify help center menu appears
    const helpMenu = comfyPage.page.locator('.help-center-menu')
    await expect(helpMenu).toBeVisible()

    // Verify "What's New?" section is visible
    const whatsNewSection = comfyPage.page.getByTestId(
      TestIds.dialogs.whatsNewSection
    )
    await expect(whatsNewSection).toBeVisible()

    // Should show the release
    await expect(
      whatsNewSection.locator('text=Comfy v0.3.44 Release')
    ).toBeVisible()
  })

  test('should toggle "What\'s New" section when setting changes', async ({
    comfyPage
  }) => {
    // Mock release API with test data
    await comfyPage.page.route('**/releases**', async (route) => {
      const url = route.request().url()
      if (
        url.includes('api.comfy.org') ||
        url.includes('stagingapi.comfy.org')
      ) {
        await route.fulfill({
          status: 200,
          contentType: 'application/json',
          body: JSON.stringify([
            createMockRelease({
              attention: 'low',
              content: '## Bug Fixes\n\n- Fixed minor issue'
            })
          ])
        })
      } else {
        await route.continue()
      }
    })

    // Start with notifications enabled
    await comfyPage.settings.setSetting(
      'Comfy.Notification.ShowVersionUpdates',
      true
    )
    await comfyPage.setup({ mockReleases: false })

    // Open help center
    const helpCenterButton = comfyPage.page.locator('.comfy-help-center-btn')
    await helpCenterButton.waitFor({ state: 'visible' })
    await helpCenterButton.click()

    // Verify "What's New?" section is visible
    const whatsNewSection = comfyPage.page.getByTestId(
      TestIds.dialogs.whatsNewSection
    )
    await expect(whatsNewSection).toBeVisible()

    // Close help center
    await comfyPage.page.locator('.help-center-backdrop').click()

    // Disable notifications
    await comfyPage.settings.setSetting(
      'Comfy.Notification.ShowVersionUpdates',
      false
    )

    // Reopen help center
    await helpCenterButton.click()

    // Verify "What's New?" section is now hidden
    await expect(whatsNewSection).toBeHidden()
  })

  test('should handle edge case with empty releases and disabled notifications', async ({
    comfyPage
  }) => {
    // Disable notifications
    await comfyPage.settings.setSetting(
      'Comfy.Notification.ShowVersionUpdates',
      false
    )

    // Mock empty releases
    await comfyPage.page.route('**/releases**', async (route) => {
      const url = route.request().url()
      if (
        url.includes('api.comfy.org') ||
        url.includes('stagingapi.comfy.org')
      ) {
        await route.fulfill({
          status: 200,
          contentType: 'application/json',
          body: JSON.stringify([])
        })
      } else {
        await route.continue()
      }
    })

    await comfyPage.setup({ mockReleases: false })

    // Open help center
    const helpCenterButton = comfyPage.page.locator('.comfy-help-center-btn')
    await helpCenterButton.waitFor({ state: 'visible' })
    await helpCenterButton.click()

    // Verify help center still works
    const helpMenu = comfyPage.page.locator('.help-center-menu')
    await expect(helpMenu).toBeVisible()

    // Section should be hidden regardless of empty releases
    const whatsNewSection = comfyPage.page.getByTestId(
      TestIds.dialogs.whatsNewSection
    )
    await expect(whatsNewSection).toBeHidden()
  })
})