import AppKitimport CodexBarCoreimport Foundationimport Testing@testable import CodexBar struct CodexBarTests {    @Test    func `icon renderer produces template image`() {        let image = IconRenderer.makeIcon(            primaryRemaining: 50,            weeklyRemaining: 75,            creditsRemaining: 500,            stale: false,            style: .codex)        #expect(image.isTemplate)        #expect(image.size.width > 0)    }     @Test    func `icon renderer renders at pixel aligned size`() {        let image = IconRenderer.makeIcon(            primaryRemaining: 50,            weeklyRemaining: 75,            creditsRemaining: 500,            stale: false,            style: .claude)        let bitmapReps = image.representations.compactMap { $0 as? NSBitmapImageRep }        #expect(bitmapReps.contains { rep in            rep.pixelsWide == 36 && rep.pixelsHigh == 36        })    }     @Test    func `icon renderer caches static icons`() {        let first = IconRenderer.makeIcon(            primaryRemaining: 42,            weeklyRemaining: 17,            creditsRemaining: 250,            stale: false,            style: .codex)        let second = IconRenderer.makeIcon(            primaryRemaining: 42,            weeklyRemaining: 17,            creditsRemaining: 250,            stale: false,            style: .codex)        #expect(first === second)    }     @Test    func `antigravity icon ignores legacy model quota lanes`() {        let snapshot = UsageSnapshot(            primary: RateWindow(usedPercent: 30, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            secondary: RateWindow(usedPercent: 60, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            tertiary: RateWindow(usedPercent: 80, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            extraRateWindows: [                NamedRateWindow(                    id: "antigravity-compact-fallback-model",                    title: "New Model",                    window: RateWindow(                        usedPercent: 64,                        windowMinutes: nil,                        resetsAt: nil,                        resetDescription: nil)),            ],            updatedAt: Date())         let remaining = IconRemainingResolver.resolvedRemaining(snapshot: snapshot, style: .antigravity)         #expect(remaining.primary == nil)        #expect(remaining.secondary == nil)    }     @Test    func `antigravity quota summary icon shows session on top and weekly on bottom`() {        let snapshot = UsageSnapshot(            primary: RateWindow(usedPercent: 84, windowMinutes: 10080, resetsAt: nil, resetDescription: nil),            secondary: RateWindow(usedPercent: 99, windowMinutes: 10080, resetsAt: nil, resetDescription: nil),            tertiary: nil,            extraRateWindows: [                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-weekly",                    title: "Gemini Weekly",                    window: RateWindow(usedPercent: 84, windowMinutes: 10080, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-5h",                    title: "Gemini Session",                    window: RateWindow(usedPercent: 97, windowMinutes: 300, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-3p-weekly",                    title: "Claude + GPT Weekly",                    window: RateWindow(usedPercent: 99, windowMinutes: 10080, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-3p-5h",                    title: "Claude + GPT Session",                    window: RateWindow(usedPercent: 98, windowMinutes: 300, resetsAt: nil, resetDescription: nil)),            ],            updatedAt: Date())         let windows = IconRemainingResolver.resolvedWindows(snapshot: snapshot, style: .antigravity)         #expect(windows.primary?.windowMinutes == 300)        #expect(windows.primary?.remainingPercent == 3)        #expect(windows.secondary?.windowMinutes == 10080)        #expect(windows.secondary?.remainingPercent == 16)    }     @Test    func `antigravity renderer draws primary above secondary`() throws {        let image = IconRenderer.makeIcon(            primaryRemaining: 100,            weeklyRemaining: 10,            creditsRemaining: nil,            stale: false,            style: .antigravity)        let bitmapReps = image.representations.compactMap { $0 as? NSBitmapImageRep }        let matchingRep = bitmapReps.first { rep in            rep.pixelsWide == 36 && rep.pixelsHigh == 36        }        let rep = try #require(matchingRep)         func averageAlpha(xRange: ClosedRange<Int>, yRange: ClosedRange<Int>) -> CGFloat {            var total: CGFloat = 0            var count: CGFloat = 0            for y in yRange {                for x in xRange {                    total += (rep.colorAt(x: x, y: y) ?? .clear).alphaComponent                    count += 1                }            }            return total / count        }         let visualTopRightAlpha = averageAlpha(xRange: 24...30, yRange: 7...10)        let visualBottomRightAlpha = averageAlpha(xRange: 24...30, yRange: 22...28)         #expect(visualTopRightAlpha > visualBottomRightAlpha + 0.2)    }     @Test    func `antigravity quota summary icon prefers gemini ids over display titles`() {        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            tertiary: nil,            extraRateWindows: [                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-weekly",                    title: "Renamed Weekly",                    window: RateWindow(usedPercent: 30, windowMinutes: 10080, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-5h",                    title: "Renamed Session",                    window: RateWindow(usedPercent: 40, windowMinutes: 300, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-3p-weekly",                    title: "Gemini Weekly",                    window: RateWindow(usedPercent: 99, windowMinutes: 10080, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-3p-5h",                    title: "Gemini Session",                    window: RateWindow(usedPercent: 98, windowMinutes: 300, resetsAt: nil, resetDescription: nil)),            ],            updatedAt: Date())         let windows = IconRemainingResolver.resolvedWindows(snapshot: snapshot, style: .antigravity)         #expect(windows.primary?.remainingPercent == 60)        #expect(windows.secondary?.remainingPercent == 70)    }     @Test    func `antigravity quota summary icon does not borrow missing gemini weekly from claude gpt`() {        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            tertiary: nil,            extraRateWindows: [                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-5h",                    title: "Gemini Session",                    window: RateWindow(usedPercent: 40, windowMinutes: 300, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-3p-weekly",                    title: "Claude + GPT Weekly",                    window: RateWindow(usedPercent: 99, windowMinutes: 10080, resetsAt: nil, resetDescription: nil)),            ],            updatedAt: Date())         let windows = IconRemainingResolver.resolvedWindows(snapshot: snapshot, style: .antigravity)         #expect(windows.primary?.remainingPercent == 60)        #expect(windows.secondary == nil)    }     @Test    func `antigravity quota summary icon treats unknown gemini rows as present`() {        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            tertiary: nil,            extraRateWindows: [                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-weekly",                    title: "Gemini Weekly",                    window: RateWindow(usedPercent: 100, windowMinutes: 10080, resetsAt: nil, resetDescription: nil),                    usageKnown: false),                NamedRateWindow(                    id: "antigravity-quota-summary-3p-weekly",                    title: "Claude + GPT Weekly",                    window: RateWindow(usedPercent: 99, windowMinutes: 10080, resetsAt: nil, resetDescription: nil)),            ],            updatedAt: Date())         let windows = IconRemainingResolver.resolvedWindows(snapshot: snapshot, style: .antigravity)         #expect(windows.primary == nil)        #expect(windows.secondary == nil)    }     @Test    func `antigravity quota summary icon falls back when gemini rows are absent`() {        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            tertiary: nil,            extraRateWindows: [                NamedRateWindow(                    id: "antigravity-quota-summary-3p-5h",                    title: "Claude + GPT Session",                    window: RateWindow(usedPercent: 75, windowMinutes: 300, resetsAt: nil, resetDescription: nil)),                NamedRateWindow(                    id: "antigravity-quota-summary-3p-weekly",                    title: "Claude + GPT Weekly",                    window: RateWindow(usedPercent: 88, windowMinutes: 10080, resetsAt: nil, resetDescription: nil)),            ],            updatedAt: Date())         let windows = IconRemainingResolver.resolvedWindows(snapshot: snapshot, style: .antigravity)         #expect(windows.primary?.remainingPercent == 25)        #expect(windows.secondary?.remainingPercent == 12)    }     @Test    func `antigravity quota summary icon tie break is stable`() {        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            tertiary: nil,            extraRateWindows: [                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-z-5h",                    title: "Gemini Session",                    window: RateWindow(                        usedPercent: 50,                        windowMinutes: 300,                        resetsAt: nil,                        resetDescription: "second-by-id")),                NamedRateWindow(                    id: "antigravity-quota-summary-gemini-a-5h",                    title: "Gemini Session",                    window: RateWindow(                        usedPercent: 50,                        windowMinutes: 300,                        resetsAt: nil,                        resetDescription: "first-by-id")),            ],            updatedAt: Date())         let windows = IconRemainingResolver.resolvedWindows(snapshot: snapshot, style: .antigravity)         #expect(windows.primary?.resetDescription == "first-by-id")        #expect(windows.secondary == nil)    }     @Test    func `perplexity icon falls back to purchased lane when bonus is exhausted`() {        let snapshot = UsageSnapshot(            primary: nil,            secondary: RateWindow(usedPercent: 100, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            tertiary: RateWindow(usedPercent: 20, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            updatedAt: Date())         let remaining = IconRemainingResolver.resolvedRemaining(snapshot: snapshot, style: .perplexity)        #expect(remaining.primary == 80)        #expect(remaining.secondary == 0)    }     @Test    func `perplexity icon skips exhausted recurring lane when purchased credits remain`() {        let snapshot = UsageSnapshot(            primary: RateWindow(usedPercent: 100, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            secondary: RateWindow(usedPercent: 100, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            tertiary: RateWindow(usedPercent: 20, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            updatedAt: Date())         let remaining = IconRemainingResolver.resolvedRemaining(snapshot: snapshot, style: .perplexity)        #expect(remaining.primary == 80)        #expect(remaining.secondary == 0)    }     @Test    func `perplexity icon prefers purchased lane before bonus`() {        let snapshot = UsageSnapshot(            primary: nil,            secondary: RateWindow(usedPercent: 20, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            tertiary: RateWindow(usedPercent: 45, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            updatedAt: Date())         let remaining = IconRemainingResolver.resolvedRemaining(snapshot: snapshot, style: .perplexity)        #expect(remaining.primary == 55)        #expect(remaining.secondary == 80)    }     @Test    func `kimi icon renders primary bar when secondary is nil`() throws {        // Regression: Kimi account connected with usage, but no progress bar shown (issue #1043).        // When secondary (rate limit) is absent, the icon renderer must still show        // the primary (weekly quota) bar.        let snapshot = UsageSnapshot(            primary: RateWindow(usedPercent: 18.3, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            secondary: nil,            updatedAt: Date())         let remaining = IconRemainingResolver.resolvedRemaining(snapshot: snapshot, style: .kimi)         guard let primaryRemaining = remaining.primary else {            Issue.record("remaining.primary was nil after IconRemainingResolver check")            return        }        #expect(primaryRemaining > 0) // 81.7% remaining        #expect(remaining.secondary == nil)         let image = IconRenderer.makeIcon(            primaryRemaining: remaining.primary,            weeklyRemaining: remaining.secondary,            creditsRemaining: nil,            stale: false,            style: .kimi)        #expect(image.size.width > 0)        #expect(image.isTemplate)         // Prove the primary bar is actually rendered using pixel inspection.        // Top bar rect: x ∈ [3, 33], y ∈ [19, 31] in the 36×36 canvas (barXPx=3, barWidthPx=30, y=19, h=12).        let bitmapReps = image.representations.compactMap { $0 as? NSBitmapImageRep }        let rep = try #require(bitmapReps.first { $0.pixelsWide == 36 && $0.pixelsHigh == 36 })         func alphaAt(px x: Int, _ y: Int) -> CGFloat {            (rep.colorAt(x: x, y: y) ?? .clear).alphaComponent        }         func regionHasFill(xRange: ClosedRange<Int>, yRange: ClosedRange<Int>) -> Bool {            for y in yRange {                for x in xRange where alphaAt(px: x, y) > 0.05 {                    return true                }            }            return false        }         // Primary bar (top track) must have fill to prove the progress bar rendered.        #expect(regionHasFill(xRange: 3...33, yRange: 19...31))    }     @Test    func `copilot icon can use selected budget as secondary lane`() {        let snapshot = UsageSnapshot(            primary: RateWindow(usedPercent: 20, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            secondary: RateWindow(usedPercent: 30, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            extraRateWindows: [                NamedRateWindow(                    id: "copilot-budget-agent",                    title: "Budget - Copilot Agent Premium Requests",                    window: RateWindow(usedPercent: 65, windowMinutes: nil, resetsAt: nil, resetDescription: nil)),            ],            updatedAt: Date())         let remaining = IconRemainingResolver.resolvedRemaining(            snapshot: snapshot,            style: .copilot,            secondaryOverrideWindowID: "copilot-budget-agent")         #expect(remaining.primary == 80)        #expect(remaining.secondary == 35)    }     @Test    func `copying extra rate windows preserves subscription dates`() {        let expiresAt = Date(timeIntervalSince1970: 1_810_656_000)        let renewsAt = Date(timeIntervalSince1970: 1_810_569_600)        let ampUsage = AmpUsageDetails(            individualCredits: 12.5,            workspaceBalances: [AmpWorkspaceBalance(name: "Team", remaining: 7.25)])        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            ampUsage: ampUsage,            subscriptionExpiresAt: expiresAt,            subscriptionRenewsAt: renewsAt,            updatedAt: Date(timeIntervalSince1970: 1_800_000_000))         let copied = snapshot.with(extraRateWindows: [])         #expect(copied.subscriptionExpiresAt == expiresAt)        #expect(copied.subscriptionRenewsAt == renewsAt)        #expect(copied.ampUsage == ampUsage)    }     @Test    func `copying rate windows preserves provider payloads`() {        let updatedAt = Date(timeIntervalSince1970: 1_800_000_000)        let mimoUsage = MiMoUsageSnapshot(            balance: 12.5,            currency: "USD",            tokenUsed: 25,            tokenLimit: 100,            tokenPercent: 0.25,            updatedAt: updatedAt)        let identity = ProviderIdentitySnapshot(            providerID: .codex,            accountEmail: "test@example.com",            accountOrganization: "Example",            loginMethod: "OAuth")        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            tertiary: RateWindow(usedPercent: 30, windowMinutes: 60, resetsAt: nil, resetDescription: nil),            mimoUsage: mimoUsage,            cursorRequests: CursorRequestUsage(used: 10, limit: 50),            subscriptionExpiresAt: updatedAt.addingTimeInterval(100),            subscriptionRenewsAt: updatedAt.addingTimeInterval(200),            updatedAt: updatedAt,            identity: identity)         let copied = snapshot.with(            primary: RateWindow(usedPercent: 40, windowMinutes: 300, resetsAt: nil, resetDescription: nil),            secondary: RateWindow(usedPercent: 50, windowMinutes: 10080, resetsAt: nil, resetDescription: nil))         #expect(copied.primary?.usedPercent == 40)        #expect(copied.secondary?.usedPercent == 50)        #expect(copied.tertiary?.usedPercent == 30)        #expect(copied.mimoUsage?.balance == 12.5)        #expect(copied.cursorRequests?.used == 10)        #expect(copied.subscriptionExpiresAt == updatedAt.addingTimeInterval(100))        #expect(copied.subscriptionRenewsAt == updatedAt.addingTimeInterval(200))        #expect(copied.identity?.accountOrganization == "Example")    }     @Test    func `copying identity preserves provider payloads`() {        let updatedAt = Date(timeIntervalSince1970: 1_800_000_000)        let ampUsage = AmpUsageDetails(            individualCredits: 12.5,            workspaceBalances: [AmpWorkspaceBalance(name: "Team", remaining: 7.25)])        let snapshot = UsageSnapshot(            primary: nil,            secondary: nil,            ampUsage: ampUsage,            cursorRequests: CursorRequestUsage(used: 10, limit: 50),            subscriptionExpiresAt: updatedAt.addingTimeInterval(100),            subscriptionRenewsAt: updatedAt.addingTimeInterval(200),            updatedAt: updatedAt)        let identity = ProviderIdentitySnapshot(            providerID: .kilo,            accountEmail: "test@example.com",            accountOrganization: "Example",            loginMethod: "API")         let copied = snapshot.withIdentity(identity)         #expect(copied.ampUsage == ampUsage)        #expect(copied.cursorRequests?.used == 10)        #expect(copied.subscriptionExpiresAt == updatedAt.addingTimeInterval(100))        #expect(copied.subscriptionRenewsAt == updatedAt.addingTimeInterval(200))        #expect(copied.identity?.accountOrganization == "Example")    }     @Test    func `copilot icon falls back to chat lane when selected budget is unavailable`() {        let snapshot = UsageSnapshot(            primary: RateWindow(usedPercent: 20, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            secondary: RateWindow(usedPercent: 30, windowMinutes: nil, resetsAt: nil, resetDescription: nil),            extraRateWindows: nil,            updatedAt: Date())         let remaining = IconRemainingResolver.resolvedRemaining(            snapshot: snapshot,            style: .copilot,            secondaryOverrideWindowID: "copilot-budget-agent")         #expect(remaining.primary == 80)        #expect(remaining.secondary == 70)    }     @Test    func `copilot icon uses selected budget in show used mode`() {        let snapshot = UsageSnapshot(            primary: RateWindow(usedPercent: 20, windowMinutes: nil, resetsAt: nil, resetDescription: nil),

Lines 1–500 of 828. Load more