import SwiftUI
import Testing
@testable import Oppi

@Suite("Workspace catalog availability")
struct WorkspaceCatalogAvailabilityTests {
    @Test func workspaceDisclosureStartsExpandedAndUsesStableStorageKey() {
        #expect(WorkspaceSidebarDisclosurePolicy.defaultExpanded)
        #expect(AppIdentifiers.workspaceSidebarExpandedKey == "workspace.sidebar.workspaces.expanded")
    }

    @Test func authoritativeEmptyCatalogIsDistinctFromUnloadedAndFailedCatalogs() {
        #expect(WorkspaceCatalogAvailability(
            hasWorkspaces: false,
            isLoaded: true,
            isSyncing: false,
            lastSyncFailed: false
        ) == .empty)
        #expect(WorkspaceCatalogAvailability(
            hasWorkspaces: false,
            isLoaded: false,
            isSyncing: false,
            lastSyncFailed: false
        ) == .loading)
        #expect(WorkspaceCatalogAvailability(
            hasWorkspaces: false,
            isLoaded: false,
            isSyncing: false,
            lastSyncFailed: true
        ) == .unavailable)
    }

    @Test func cachedCatalogRemainsAvailableAfterRefreshFailure() {
        #expect(WorkspaceCatalogAvailability(
            hasWorkspaces: true,
            isLoaded: true,
            isSyncing: false,
            lastSyncFailed: true
        ) == .available)
    }
}

@Suite("Workspace sidebar session status")
struct WorkspaceSidebarSessionStatusTests {
    @Test func countsAttentionWorkingAndDoneUsingSessionRowSemantics() {
        let status = WorkspaceSidebarSessionStatus(sessions: [
            makeTestSession(id: "busy", status: .busy, messageCount: 1, firstMessage: "Working"),
            makeTestSession(id: "starting", status: .starting, messageCount: 1, firstMessage: "Starting"),
            makeTestSession(id: "ready", status: .ready, messageCount: 1, firstMessage: "Finished"),
            makeTestSession(id: "stopped", status: .stopped, messageCount: 1, firstMessage: "Stopped"),
            makeTestSession(id: "error", status: .error, messageCount: 1, firstMessage: "Failed"),
        ])

        #expect(status.questionCount == 0)
        #expect(status.errorCount == 1)
        #expect(status.attentionCount == 1)
        #expect(status.workingCount == 2)
        #expect(status.doneCount == 1)
        #expect(status.showsDone == false)
        #expect(status.accessibilityValue == "1 session has an error, 2 working sessions, 1 done session")
    }

    @Test func excludesDraftSessionsAndPrioritizesQuestionsOverDone() {
        let status = WorkspaceSidebarSessionStatus(
            sessions: [
                makeTestSession(id: "ready-draft", status: .ready),
                makeTestSession(id: "starting-draft", status: .starting),
                makeTestSession(
                    id: "question",
                    status: .ready,
                    messageCount: 1,
                    firstMessage: "Need input"
                ),
            ],
            pendingAskCountForSession: { $0 == "question" ? 1 : 0 }
        )

        #expect(status.questionCount == 1)
        #expect(status.errorCount == 0)
        #expect(status.attentionCount == 1)
        #expect(status.doneCount == 0)
        #expect(status.showsDone == false)
        #expect(status.isVisible)
        #expect(status.accessibilityValue == "1 session needs attention")
    }

    @Test func showsDoneWhenNoSessionNeedsAttention() {
        let status = WorkspaceSidebarSessionStatus(sessions: [
            makeTestSession(id: "working", status: .busy, messageCount: 1, firstMessage: "Working"),
            makeTestSession(id: "done", status: .ready, messageCount: 1, firstMessage: "Finished"),
        ])

        #expect(status.showsDone)
    }

    @MainActor
    @Test func rendersPriorityMetricsWithinTheNarrowTrailingBudget() throws {
        let status = WorkspaceSidebarSessionStatus(sessions: [
            makeTestSession(id: "error", status: .error, messageCount: 1, firstMessage: "Failed"),
            makeTestSession(id: "working", status: .busy, messageCount: 1, firstMessage: "Working"),
            makeTestSession(id: "done", status: .ready, messageCount: 1, firstMessage: "Finished"),
        ])
        let renderer = ImageRenderer(
            content: WorkspaceSidebarSessionStatusIndicator(status: status)
        )
        renderer.scale = 3

        let image = try #require(renderer.uiImage)
        #expect(image.size.width <= 64)
        #expect(image.size.height <= 44)
    }
}