wehub-resource-sync
129 lines
5.1 KiB
Swift
129 lines
5.1 KiB
Swift
import CodexBarCore
|
|
import Foundation
|
|
import Testing
|
|
@testable import CodexBar
|
|
|
|
@MainActor
|
|
struct MenuDescriptorSakanaTests {
|
|
@Test
|
|
func `sakana pay as you go rows render when optional usage is enabled`() throws {
|
|
let lines = try Self.menuLines(showOptionalUsage: true)
|
|
|
|
#expect(lines.contains("Balance: $12.34"))
|
|
#expect(lines.contains("Usage: $5.67"))
|
|
}
|
|
|
|
@Test
|
|
func `sakana pay as you go rows are hidden when optional usage is disabled`() throws {
|
|
// Regression for the render-path staleness gap: toggling "Show optional credits and extra
|
|
// usage" off only rebuilds the menu, it does not immediately refetch, so a
|
|
// previously-populated sakanaPayAsYouGo lingers in the cached snapshot. The rows must be
|
|
// gated on the setting, not only on the presence of the (possibly stale) snapshot field.
|
|
let lines = try Self.menuLines(showOptionalUsage: false)
|
|
|
|
#expect(!lines.contains(where: { $0.hasPrefix("Balance:") }))
|
|
#expect(!lines.contains(where: { $0.hasPrefix("Usage:") }))
|
|
// The required quota windows must still render regardless of the optional-usage setting.
|
|
#expect(lines.contains(where: { $0.hasPrefix("5-hour") }))
|
|
#expect(lines.contains(where: { $0.hasPrefix("Weekly") }))
|
|
}
|
|
|
|
private static func menuLines(showOptionalUsage: Bool) throws -> [String] {
|
|
let suite = "MenuDescriptorSakanaTests-\(showOptionalUsage)"
|
|
let defaults = try #require(UserDefaults(suiteName: suite))
|
|
defaults.removePersistentDomain(forName: suite)
|
|
|
|
let settings = SettingsStore(
|
|
userDefaults: defaults,
|
|
configStore: testConfigStore(suiteName: suite),
|
|
zaiTokenStore: NoopZaiTokenStore(),
|
|
syntheticTokenStore: NoopSyntheticTokenStore())
|
|
settings.statusChecksEnabled = false
|
|
settings.showOptionalCreditsAndExtraUsage = showOptionalUsage
|
|
|
|
let store = UsageStore(
|
|
fetcher: UsageFetcher(environment: [:]),
|
|
browserDetection: BrowserDetection(cacheTTL: 0),
|
|
settings: settings)
|
|
let snapshot = SakanaUsageSnapshot(
|
|
planName: "Standard",
|
|
priceLabel: "$20/mo",
|
|
fiveHour: .init(usedPercent: 10, resetsAt: nil),
|
|
weekly: .init(usedPercent: 20, resetsAt: nil),
|
|
payAsYouGo: SakanaPayAsYouGoSnapshot(
|
|
creditBalance: 12.34,
|
|
periodUsageTotal: 5.67,
|
|
periodLabel: "Jun 02, 2026 - Jul 01, 2026"))
|
|
store._setSnapshotForTesting(snapshot.toUsageSnapshot(), provider: .sakana)
|
|
|
|
let descriptor = MenuDescriptor.build(
|
|
provider: .sakana,
|
|
store: store,
|
|
settings: settings,
|
|
account: AccountInfo(email: nil, plan: nil),
|
|
updateReady: false,
|
|
includeContextualActions: false)
|
|
return descriptor.sections
|
|
.flatMap(\.entries)
|
|
.compactMap { entry -> String? in
|
|
guard case let .text(text, _) = entry else { return nil }
|
|
return text
|
|
}
|
|
}
|
|
}
|
|
|
|
struct SakanaMenuCardModelTests {
|
|
@Test
|
|
func `pay as you go renders in the live menu card`() throws {
|
|
let model = try Self.model(showOptionalUsage: true)
|
|
|
|
#expect(model.providerCost?.title == "Extra usage")
|
|
#expect(model.providerCost?.spendLine == "Balance: $12.34")
|
|
#expect(model.providerCost?.percentLine == "Usage: $5.67")
|
|
#expect(model.providerCost?.percentUsed == nil)
|
|
}
|
|
|
|
@Test
|
|
func `pay as you go hides immediately when optional usage is disabled`() throws {
|
|
let model = try Self.model(showOptionalUsage: false)
|
|
|
|
#expect(model.providerCost == nil)
|
|
#expect(model.metrics.map(\.title) == ["5-hour", "Weekly"])
|
|
}
|
|
|
|
private static func model(showOptionalUsage: Bool) throws -> UsageMenuCardView.Model {
|
|
let now = Date(timeIntervalSince1970: 0)
|
|
let snapshot = SakanaUsageSnapshot(
|
|
planName: "Standard",
|
|
priceLabel: "$20/mo",
|
|
fiveHour: .init(usedPercent: 10, resetsAt: nil),
|
|
weekly: .init(usedPercent: 20, resetsAt: nil),
|
|
payAsYouGo: SakanaPayAsYouGoSnapshot(
|
|
creditBalance: 12.34,
|
|
periodUsageTotal: 5.67,
|
|
periodLabel: "Jun 02, 2026 - Jul 01, 2026"),
|
|
updatedAt: now)
|
|
let metadata = try #require(ProviderDefaults.metadata[.sakana])
|
|
|
|
return UsageMenuCardView.Model.make(.init(
|
|
provider: .sakana,
|
|
metadata: metadata,
|
|
snapshot: snapshot.toUsageSnapshot(),
|
|
credits: nil,
|
|
creditsError: nil,
|
|
dashboard: nil,
|
|
dashboardError: nil,
|
|
tokenSnapshot: nil,
|
|
tokenError: nil,
|
|
account: AccountInfo(email: nil, plan: nil),
|
|
isRefreshing: false,
|
|
lastError: nil,
|
|
usageBarsShowUsed: true,
|
|
resetTimeDisplayStyle: .countdown,
|
|
tokenCostUsageEnabled: false,
|
|
showOptionalCreditsAndExtraUsage: showOptionalUsage,
|
|
hidePersonalInfo: false,
|
|
now: now))
|
|
}
|
|
}
|