生成された課金コードを Sandbox の前に検証する — SKTestSession で返金・失効・購入失敗を自動テストする

func testPurchaseGrantsEntitlement() async throws {
    // 購入前は権利なし
    let before = await EntitlementStore.shared.isSubscribed()
    XCTAssertFalse(before, "購入前に権利が付いています")
 
    // セッション側から購入を発生させる（UI を経由しない）
    try await session.buyProduct(identifier: "com.example.pro.monthly")
 
    // アプリ側の判定ロジックを呼ぶ
    let after = await EntitlementStore.shared.isSubscribed()
    XCTAssertTrue(after, "購入後も権利が付いていません")
}

func testExpirationRevokesEntitlement() async throws {
    session.timeRate = .oneSecondIsOneDay
 
    try await session.buyProduct(identifier: "com.example.pro.monthly")
    let granted = await EntitlementStore.shared.isSubscribed()
    XCTAssertTrue(granted)
 
    // 更新を止めて、明示的に失効させる
    try session.expireSubscription(productIdentifier: "com.example.pro.monthly")
 
    let revoked = await EntitlementStore.shared.isSubscribed()
    XCTAssertFalse(revoked, "失効後も権利が残っています")
}

func testRefundRevokesEntitlement() async throws {
    try await session.buyProduct(identifier: "com.example.pro.lifetime")
    XCTAssertTrue(await EntitlementStore.shared.isSubscribed())
 
    // 直近のトランザクションを取り出して返金する
    let transactions = session.allTransactions()
    guard let target = transactions.first else {
        return XCTFail("トランザクションが見つかりません")
    }
    try session.refundTransaction(identifier: target.identifier)
 
    // Transaction.updates の反映を待つ
    try await Task.sleep(nanoseconds: 300_000_000)
 
    let stillHasAccess = await EntitlementStore.shared.isSubscribed()
    XCTAssertFalse(stillHasAccess, "返金後も権利が残っています")
}

// アプリ起動時に1度だけ。ボタンを押した時ではありません
@main
struct MyApp: App {
    @State private var updatesTask: Task<Void, Never>?
 
    var body: some Scene {
        WindowGroup {
            ContentView()
                .task {
                    updatesTask = Task.detached {
                        for await result in Transaction.updates {
                            guard case .verified(let transaction) = result else { continue }
                            await EntitlementStore.shared.refresh()
                            await transaction.finish()
                        }
                    }
                }
        }
    }
}

func testPurchaseFailureSurfacesToUser() async throws {
    // 購入 API だけを失敗させる
    session.setSimulatedError(.generic(.unknown), forAPI: .purchase)
 
    let viewModel = PaywallViewModel()
    await viewModel.purchase(productID: "com.example.pro.monthly")
 
    // 失敗がユーザーに見えているか
    XCTAssertNotNil(viewModel.errorMessage, "購入失敗がユーザーに伝わっていません")
    XCTAssertFalse(viewModel.isPurchasing, "ローディングが解除されていません")
 
    // 失敗したのに権利が付いていないか
    let granted = await EntitlementStore.shared.isSubscribed()
    XCTAssertFalse(granted, "失敗した購入で権利が付いています")
 
    // 後片付け
    session.setSimulatedError(nil, forAPI: .purchase)
}

func testAskToBuyPendingState() async throws {
    session.askToBuyEnabled = true
 
    try await session.buyProduct(identifier: "com.example.pro.monthly")
 
    // 承認前に権利が付いていないこと
    let beforeApproval = await EntitlementStore.shared.isSubscribed()
    XCTAssertFalse(beforeApproval, "承認前に権利が付いています")
 
    // 保護者が承認したことにする
    let pending = session.allTransactions().filter { $0.state == .deferred }
    guard let target = pending.first else {
        return XCTFail("保留トランザクションがありません")
    }
    try session.approveAskToBuyTransaction(identifier: target.identifier)
    try await Task.sleep(nanoseconds: 300_000_000)
 
    let afterApproval = await EntitlementStore.shared.isSubscribed()
    XCTAssertTrue(afterApproval, "承認後に権利が付いていません")
}