83 lines
4.1 KiB
Swift
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"))
|
|
}
|
|
}
|