// src/plugin/__tests__/workspace-iterate-root.test.ts
import { describe, it, expect } from 'vitest'
import { Workspace, WorkspaceLeaf } from '../workspace'

describe('Workspace.iterateRootLeaves', () => {
  it('iterates only main-area leaves, not sidebars', () => {
    const container = document.createElement('div')
    const workspace = new Workspace(container)

    const main1 = workspace.getLeaf(true)
    const main2 = workspace.getLeaf(true)
    workspace.getLeftLeaf(true)
    workspace.getRightLeaf(true)

    const visited: WorkspaceLeaf[] = []
    workspace.iterateRootLeaves((leaf: WorkspaceLeaf) => visited.push(leaf))

    expect(visited).toHaveLength(2)
    expect(visited).toContain(main1)
    expect(visited).toContain(main2)
  })

  it('handles empty workspace', () => {
    const container = document.createElement('div')
    const workspace = new Workspace(container)

    const visited: WorkspaceLeaf[] = []
    workspace.iterateRootLeaves((leaf: WorkspaceLeaf) => visited.push(leaf))

    expect(visited).toHaveLength(0)
  })
})