simstudioai--sim/apps/sim/blocks/visibility/client.test.ts

/**
 * @vitest-environment node
 */
import { beforeEach, describe, expect, it, vi } from 'vitest'

const { mockNotify } = vi.hoisted(() => ({ mockNotify: vi.fn() }))

vi.mock('@/blocks/custom/client-overlay', () => ({
  notifyBlockOverlayChanged: mockNotify,
}))

import type { BlockVisibilityState } from '@/lib/core/config/block-visibility'
// client-boundary-allow: vitest ignores the 'use client' directive; this node-env test exercises the module directly
import { hydrateBlockVisibility, resetBlockVisibilityForSwitch } from '@/blocks/visibility/client'
import { overlayVisibility, registerBlockCacheInvalidator } from '@/blocks/visibility/context'

function state(revealed: string[], disabled: string[] = []): BlockVisibilityState {
  return {
    revealed: new Set(revealed),
    disabled: new Set(disabled),
    previewTagged: new Set(revealed),
  }
}

describe('hydrateBlockVisibility', () => {
  beforeEach(() => vi.clearAllMocks())

  // Must run FIRST: module state starts null and persists across tests.
  it('treats an empty state while none is set as a no-op (null ≡ empty)', () => {
    const invalidator = vi.fn()
    const unregister = registerBlockCacheInvalidator(invalidator)

    hydrateBlockVisibility(state([]))
    expect(overlayVisibility()).toBeNull()
    expect(invalidator).not.toHaveBeenCalled()
    expect(mockNotify).not.toHaveBeenCalled()

    unregister()
  })

  it('applies state, fires invalidators, and bumps the overlay version', () => {
    const invalidator = vi.fn()
    const unregister = registerBlockCacheInvalidator(invalidator)

    hydrateBlockVisibility(state(['gmail_v2']))
    expect(overlayVisibility()?.revealed.has('gmail_v2')).toBe(true)
    expect(invalidator).toHaveBeenCalledTimes(1)
    expect(mockNotify).toHaveBeenCalledTimes(1)

    unregister()
  })

  it('drops reveals but carries kill switches over on workspace switch', () => {
    const invalidator = vi.fn()
    const unregister = registerBlockCacheInvalidator(invalidator)

    // Reveal + kill-switch in workspace A, then switch (loader resets while the
    // new projection loads). (notion_v3: distinct from prior tests' state —
    // module state persists.)
    hydrateBlockVisibility(state(['notion_v3'], ['slack']))
    resetBlockVisibilityForSwitch()
    expect(overlayVisibility()?.revealed.size).toBe(0)
    expect(overlayVisibility()?.previewTagged.size).toBe(0)
    expect(overlayVisibility()?.disabled).toEqual(new Set(['slack']))
    expect(invalidator).toHaveBeenCalledTimes(2)

    // Repeated resets are no-ops (deep-equal).
    resetBlockVisibilityForSwitch()
    expect(invalidator).toHaveBeenCalledTimes(2)

    unregister()
  })

  it('no-ops on a deep-equal state (fresh objects, same content)', () => {
    const invalidator = vi.fn()
    const unregister = registerBlockCacheInvalidator(invalidator)

    hydrateBlockVisibility(state(['gmail_v2'], ['slack']))
    hydrateBlockVisibility(state(['gmail_v2'], ['slack']))
    expect(invalidator).toHaveBeenCalledTimes(1)
    expect(mockNotify).toHaveBeenCalledTimes(1)

    hydrateBlockVisibility(state(['gmail_v2', 'notion_v3'], ['slack']))
    expect(invalidator).toHaveBeenCalledTimes(2)
    expect(mockNotify).toHaveBeenCalledTimes(2)

    unregister()
  })
})