Files
2026-07-13 12:22:33 +08:00

83 lines
4.1 KiB
Swift

import Foundation
import Testing
@testable import CodexBarCore
struct ClaudePlanResolverTests {
@Test
func `oauth rate limit tier maps to branded plan`() {
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "claude_pro") == "Claude Pro")
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "claude_team") == "Claude Team")
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "claude_enterprise") == "Claude Enterprise")
}
@Test
func `oauth rate limit tier preserves the Max usage multiplier`() {
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "default_claude_max_5x") == "Claude Max 5x")
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "default_claude_max_20x") == "Claude Max 20x")
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "v2_default_claude_max_20x") == "Claude Max 20x")
// A bare Max tier without a multiplier keeps the plain label.
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "claude_max") == "Claude Max")
#expect(ClaudePlan.oauthLoginMethod(rateLimitTier: "default_claude_team_5x") == "Claude Team")
// A resolved non-Max plan never inherits a Max multiplier from a disagreeing tier.
#expect(
ClaudePlan.oauthLoginMethod(subscriptionType: "team", rateLimitTier: "default_claude_max_5x")
== "Claude Team")
#expect(
ClaudePlan.webLoginMethod(rateLimitTier: "default_claude_max_20x", billingType: nil)
== "Claude Max 20x")
}
@Test
func `oauth subscription type overrides generic rate limit tier`() {
#expect(
ClaudePlan.oauthLoginMethod(subscriptionType: "pro", rateLimitTier: "default_claude_ai")
== "Claude Pro")
#expect(
ClaudePlan.oauthLoginMethod(subscriptionType: "team", rateLimitTier: "default_claude_max_5x")
== "Claude Team")
#expect(ClaudePlan.oauthLoginMethod(subscriptionType: nil, rateLimitTier: "default_claude_ai") == nil)
}
@Test
func `web fallback preserves stripe Claude compatibility`() {
#expect(
ClaudePlan.webLoginMethod(
rateLimitTier: "default_claude",
billingType: "stripe_subscription")
== "Claude Pro")
}
@Test
func `compatibility parser understands current labels`() {
#expect(ClaudePlan.fromCompatibilityLoginMethod("Claude Max") == .max)
#expect(ClaudePlan.fromCompatibilityLoginMethod("Max") == .max)
#expect(ClaudePlan.fromCompatibilityLoginMethod("Claude Pro") == .pro)
#expect(ClaudePlan.fromCompatibilityLoginMethod("Ultra") == .ultra)
#expect(ClaudePlan.fromCompatibilityLoginMethod("Claude Team") == .team)
#expect(ClaudePlan.fromCompatibilityLoginMethod("Claude Enterprise") == .enterprise)
}
@Test
func `CLI projection keeps compact compatibility and unknown fallback`() {
#expect(ClaudePlan.cliCompatibilityLoginMethod("Claude Max Account") == "Max")
#expect(ClaudePlan.cliCompatibilityLoginMethod("Team") == "Team")
#expect(ClaudePlan.cliCompatibilityLoginMethod("Claude Enterprise Account") == "Enterprise")
#expect(ClaudePlan.cliCompatibilityLoginMethod("Claude Ultra Account") == "Ultra")
#expect(ClaudePlan.cliCompatibilityLoginMethod("Experimental") == "Experimental")
#expect(ClaudePlan.cliCompatibilityLoginMethod("Profile") == "Profile")
#expect(ClaudePlan.cliCompatibilityLoginMethod("Browser profile") == "Browser profile")
}
@Test
func `subscription compatibility preserves ultra and excludes enterprise`() {
#expect(ClaudePlan.isSubscriptionLoginMethod("Claude Max"))
#expect(ClaudePlan.isSubscriptionLoginMethod("Pro"))
#expect(ClaudePlan.isSubscriptionLoginMethod("Ultra"))
#expect(ClaudePlan.isSubscriptionLoginMethod("Team"))
#expect(!ClaudePlan.isSubscriptionLoginMethod("Claude Enterprise"))
#expect(!ClaudePlan.isSubscriptionLoginMethod("Profile"))
#expect(!ClaudePlan.isSubscriptionLoginMethod("Browser profile"))
#expect(!ClaudePlan.isSubscriptionLoginMethod("API"))
}
}