Peter Steinbergerfix: harden Vertex AI quota usagef73e28d10 months agoHistory
69 lines · 2.1 KB · Swift
RawDownload
import Foundationimport Testing@testable import CodexBarCore @Suitestruct FactoryStatusSnapshotTests {    @Test    func mapsUsageSnapshotWindowsAndLoginMethod() {        let periodEnd = Date(timeIntervalSince1970: 1_738_368_000) // Feb 1, 2025        let snapshot = FactoryStatusSnapshot(            standardUserTokens: 50,            standardOrgTokens: 0,            standardAllowance: 100,            premiumUserTokens: 25,            premiumOrgTokens: 0,            premiumAllowance: 50,            periodStart: nil,            periodEnd: periodEnd,            planName: "Pro",            tier: "enterprise",            organizationName: "Acme",            accountEmail: "user@example.com",            userId: "user-1",            rawJSON: nil)         let usage = snapshot.toUsageSnapshot()         #expect(usage.primary?.usedPercent == 50)        #expect(usage.primary?.resetsAt == periodEnd)        #expect(usage.primary?.resetDescription?.hasPrefix("Resets ") == true)        #expect(usage.secondary?.usedPercent == 50)        #expect(usage.loginMethod(for: .factory) == "Factory Enterprise - Pro")    }     @Test    func treatsLargeAllowancesAsUnlimited() {        let snapshot = FactoryStatusSnapshot(            standardUserTokens: 50_000_000,            standardOrgTokens: 0,            standardAllowance: 2_000_000_000_000,            premiumUserTokens: 0,            premiumOrgTokens: 0,            premiumAllowance: 0,            periodStart: nil,            periodEnd: nil,            planName: nil,            tier: nil,            organizationName: nil,            accountEmail: nil,            userId: nil,            rawJSON: nil)         let usage = snapshot.toUsageSnapshot()         #expect(usage.primary?.usedPercent == 50)    }} @Suitestruct FactoryStatusProbeWorkOSTests {    @Test    func detectsMissingRefreshTokenPayload() {        let payload = Data("""        {"error":"invalid_request","error_description":"Missing refresh token."}        """.utf8)         #expect(FactoryStatusProbe.isMissingWorkOSRefreshToken(payload))    }}