298 lines
10 KiB
Swift
298 lines
10 KiB
Swift
//
|
|
// IAPManagerTests.swift
|
|
// Passepartout
|
|
//
|
|
// Created by Davide De Rosa on 9/12/24.
|
|
// Copyright (c) 2024 Davide De Rosa. All rights reserved.
|
|
//
|
|
// https://github.com/passepartoutvpn
|
|
//
|
|
// This file is part of Passepartout.
|
|
//
|
|
// Passepartout is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// Passepartout is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Passepartout. If not, see <http://www.gnu.org/licenses/>.
|
|
//
|
|
|
|
@testable import CommonLibrary
|
|
import CommonUtils
|
|
import Foundation
|
|
import XCTest
|
|
|
|
final class IAPManagerTests: XCTestCase {
|
|
// private let inApp = MockAppProductHelper()
|
|
|
|
private let olderBuildNumber = 500
|
|
|
|
private let defaultBuildNumber = 1000
|
|
|
|
private let newerBuildNumber = 1500
|
|
}
|
|
|
|
@MainActor
|
|
extension IAPManagerTests {
|
|
|
|
// MARK: Build products
|
|
|
|
func test_givenBuildProducts_whenOlder_thenFullVersion() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(withBuild: olderBuildNumber, identifiers: [])
|
|
let sut = IAPManager(receiptReader: reader) { build in
|
|
if build <= self.defaultBuildNumber {
|
|
return [.Full.allPlatforms]
|
|
}
|
|
return []
|
|
}
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: AppFeature.fullV2Features))
|
|
}
|
|
|
|
func test_givenBuildProducts_whenNewer_thenFreeVersion() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(withBuild: newerBuildNumber, products: [])
|
|
let sut = IAPManager(receiptReader: reader) { build in
|
|
if build <= self.defaultBuildNumber {
|
|
return [.Full.allPlatforms]
|
|
}
|
|
return []
|
|
}
|
|
await sut.reloadReceipt()
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.fullV2Features))
|
|
}
|
|
|
|
// MARK: Eligibility
|
|
|
|
func test_givenPurchasedFeature_whenReloadReceipt_thenIsEligible() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.fullV2Features))
|
|
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [.Full.allPlatforms])
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.fullV2Features))
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: AppFeature.fullV2Features))
|
|
}
|
|
|
|
func test_givenPurchasedFeatures_thenIsOnlyEligibleForFeatures() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [
|
|
.Features.networkSettings
|
|
])
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: .dns))
|
|
XCTAssertTrue(sut.isEligible(for: .httpProxy))
|
|
XCTAssertFalse(sut.isEligible(for: .onDemand))
|
|
XCTAssertTrue(sut.isEligible(for: .routing))
|
|
XCTAssertFalse(sut.isEligible(for: .sharing))
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.fullV2Features))
|
|
}
|
|
|
|
func test_givenPurchasedAndCancelledFeature_thenIsNotEligible() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(
|
|
withBuild: defaultBuildNumber,
|
|
products: [.Full.allPlatforms],
|
|
cancelledProducts: [.Full.allPlatforms]
|
|
)
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.fullV2Features))
|
|
}
|
|
|
|
func test_givenFreeVersion_thenIsNotEligibleForAnyFeature() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [])
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertFalse(sut.userLevel.isFullVersion)
|
|
AppFeature.fullV2Features.forEach {
|
|
XCTAssertFalse(sut.isEligible(for: $0))
|
|
}
|
|
}
|
|
|
|
func test_givenFreeVersion_thenIsNotEligibleForAppleTV() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [])
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertFalse(sut.isEligible(for: .appleTV))
|
|
}
|
|
|
|
func test_givenFullV2Version_thenIsEligibleForAnyFeatureExceptExcluded() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [.Full.allPlatforms])
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
let excluded: Set<AppFeature> = [
|
|
.appleTV,
|
|
.interactiveLogin
|
|
]
|
|
AppFeature.allCases.forEach {
|
|
if AppFeature.fullV2Features.contains($0) {
|
|
XCTAssertTrue(sut.isEligible(for: $0))
|
|
} else {
|
|
XCTAssertTrue(excluded.contains($0))
|
|
XCTAssertFalse(sut.isEligible(for: $0))
|
|
}
|
|
}
|
|
}
|
|
|
|
func test_givenAppleTV_thenIsEligibleForAppleTV() async {
|
|
let reader = MockAppReceiptReader()
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [.Features.appleTV])
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: .appleTV))
|
|
}
|
|
|
|
func test_givenPlatformVersion_thenIsFullVersionForPlatform() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
#if os(macOS)
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [.Full.macOS, .Features.networkSettings])
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: AppFeature.fullV2Features))
|
|
#else
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [.Full.iOS, .Features.networkSettings])
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: AppFeature.fullV2Features))
|
|
#endif
|
|
}
|
|
|
|
func test_givenPlatformVersion_thenIsNotFullVersionForOtherPlatform() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(receiptReader: reader)
|
|
|
|
#if os(macOS)
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [.Full.iOS, .Features.networkSettings])
|
|
await sut.reloadReceipt()
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.fullV2Features))
|
|
#else
|
|
await reader.setReceipt(withBuild: defaultBuildNumber, products: [.Full.macOS, .Features.networkSettings])
|
|
await sut.reloadReceipt()
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.fullV2Features))
|
|
#endif
|
|
}
|
|
|
|
// MARK: App level
|
|
|
|
func test_givenBetaApp_thenIsRestricted() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .beta, receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.userLevel.isRestricted)
|
|
}
|
|
|
|
func test_givenBetaApp_thenIsNotEligibleForAllFeatures() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .beta, receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertFalse(sut.isEligible(for: AppFeature.allCases))
|
|
}
|
|
|
|
func test_givenBetaApp_thenIsEligibleForUserLevelFeatures() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .beta, receiptReader: reader)
|
|
|
|
let eligible = AppUserLevel.beta.features
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: eligible))
|
|
}
|
|
|
|
func test_givenBetaApp_thenIsEligibleForUnrestrictedFeature() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .beta, receiptReader: reader, unrestrictedFeatures: [.onDemand])
|
|
|
|
var eligible = AppUserLevel.beta.features
|
|
eligible.append(.onDemand)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.isEligible(for: eligible))
|
|
}
|
|
|
|
func test_givenFullV2App_thenIsFullVersion() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .fullV2, receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.userLevel.isFullVersion)
|
|
}
|
|
|
|
func test_givenSubscriberApp_thenIsFullVersion() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .subscriber, receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
XCTAssertTrue(sut.userLevel.isFullVersion)
|
|
}
|
|
|
|
func test_givenFullV2App_thenIsEligibleForAnyFeatureExceptExcluded() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .fullV2, receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
let excluded: Set<AppFeature> = [
|
|
.appleTV,
|
|
.interactiveLogin
|
|
]
|
|
AppFeature.allCases.forEach {
|
|
if AppFeature.fullV2Features.contains($0) {
|
|
XCTAssertTrue(sut.isEligible(for: $0))
|
|
} else {
|
|
XCTAssertTrue(excluded.contains($0))
|
|
XCTAssertFalse(sut.isEligible(for: $0))
|
|
}
|
|
}
|
|
}
|
|
|
|
func test_givenSubscriberApp_thenIsEligibleForAnyFeature() async {
|
|
let reader = MockAppReceiptReader()
|
|
let sut = IAPManager(customUserLevel: .subscriber, receiptReader: reader)
|
|
|
|
await sut.reloadReceipt()
|
|
AppFeature.fullV2Features.forEach {
|
|
XCTAssertTrue(sut.isEligible(for: $0))
|
|
}
|
|
XCTAssertTrue(sut.isEligible(for: .appleTV))
|
|
}
|
|
}
|
|
|
|
private extension IAPManager {
|
|
convenience init(
|
|
customUserLevel: AppUserLevel? = nil,
|
|
receiptReader: AppReceiptReader,
|
|
unrestrictedFeatures: Set<AppFeature> = [],
|
|
productsAtBuild: BuildProducts<AppProduct>? = nil
|
|
) {
|
|
self.init(
|
|
customUserLevel: customUserLevel,
|
|
inAppHelper: MockAppProductHelper(),
|
|
receiptReader: receiptReader,
|
|
unrestrictedFeatures: unrestrictedFeatures,
|
|
productsAtBuild: productsAtBuild
|
|
)
|
|
}
|
|
}
|