Always reload CustomerInfo when Customer Center is loaded

RevenueCat · Dec 11, 2024 · fa27846 · fa27846
1 parent 24d6660
commit fa27846
Show file tree

Hide file tree

Showing 6 changed files with 28 additions and 6 deletions.
diff --git a/RevenueCatUI/CustomerCenter/Abstractions/CustomerCenterPurchasesType.swift b/RevenueCatUI/CustomerCenter/Abstractions/CustomerCenterPurchasesType.swift
@@ -23,7 +23,9 @@ protocol CustomerCenterPurchasesType: Sendable {
     var isSandbox: Bool { get }
 
     @Sendable
-    func customerInfo() async throws -> CustomerInfo
+    func customerInfo(
+        fetchPolicy: CacheFetchPolicy
+    ) async throws -> CustomerInfo
 
     @Sendable
     func products(_ productIdentifiers: [String]) async -> [StoreProduct]

diff --git a/RevenueCatUI/CustomerCenter/Data/CustomerCenterPurchases.swift b/RevenueCatUI/CustomerCenter/Data/CustomerCenterPurchases.swift
@@ -24,8 +24,10 @@ final class CustomerCenterPurchases: CustomerCenterPurchasesType {
         return Purchases.shared.isSandbox
     }
 
-    func customerInfo() async throws -> RevenueCat.CustomerInfo {
-        try await Purchases.shared.customerInfo()
+    func customerInfo(
+        fetchPolicy: CacheFetchPolicy
+    ) async throws -> RevenueCat.CustomerInfo {
+        try await Purchases.shared.customerInfo(fetchPolicy: fetchPolicy)
     }
 
     func products(_ productIdentifiers: [String]) async -> [StoreProduct] {

diff --git a/RevenueCatUI/CustomerCenter/Data/LoadPromotionalOfferUseCase.swift b/RevenueCatUI/CustomerCenter/Data/LoadPromotionalOfferUseCase.swift
@@ -41,7 +41,7 @@ class LoadPromotionalOfferUseCase: LoadPromotionalOfferUseCaseType {
         promoOfferDetails: CustomerCenterConfigData.HelpPath.PromotionalOffer
     ) async -> Result<PromotionalOfferData, Error> {
         do {
-            let customerInfo = try await self.purchasesProvider.customerInfo()
+            let customerInfo = try await self.purchasesProvider.customerInfo(fetchPolicy: .default)
 
             let subscribedProduct = try await getActiveSubscription(customerInfo)
             let discount = try findDiscount(for: subscribedProduct,

diff --git a/RevenueCatUI/CustomerCenter/ViewModels/CustomerCenterViewModel.swift b/RevenueCatUI/CustomerCenter/ViewModels/CustomerCenterViewModel.swift
@@ -99,7 +99,7 @@ import RevenueCat
 
     func loadPurchaseInformation() async {
         do {
-            let customerInfo = try await purchasesProvider.customerInfo()
+            let customerInfo = try await purchasesProvider.customerInfo(fetchPolicy: .fetchCurrent)
             let hasActiveProducts =
             !customerInfo.activeSubscriptions.isEmpty || !customerInfo.nonSubscriptions.isEmpty
 

diff --git a/Tests/RevenueCatUITests/CustomerCenter/CustomerCenterViewModelTests.swift b/Tests/RevenueCatUITests/CustomerCenter/CustomerCenterViewModelTests.swift
@@ -70,6 +70,21 @@ class CustomerCenterViewModelTests: TestCase {
         expect(viewModel.isLoaded) == true
     }
 
+    func testLoadPurchaseInformationAlwaysRefreshesCustomerInfo() async throws {
+        let mockPurchases = MockCustomerCenterPurchases(
+            customerInfo: CustomerCenterViewModelTests.customerInfoWithAppleSubscriptions
+        )
+
+        let viewModel = CustomerCenterViewModel(
+            customerCenterActionHandler: nil,
+            purchasesProvider: mockPurchases
+        )
+
+        await viewModel.loadPurchaseInformation()
+
+        expect(mockPurchases.customerInfoFetchPolicy) == .fetchCurrent
+    }
+
     func testLoadHasSubscriptionsApple() async throws {
         let mockPurchases =
         MockCustomerCenterPurchases(customerInfo: CustomerCenterViewModelTests.customerInfoWithAppleSubscriptions)

diff --git a/Tests/RevenueCatUITests/CustomerCenter/MockCustomerCenterPurchases.swift b/Tests/RevenueCatUITests/CustomerCenter/MockCustomerCenterPurchases.swift
@@ -50,7 +50,10 @@ final class MockCustomerCenterPurchases: @unchecked Sendable, CustomerCenterPurc
         self.beginRefundShouldFail = beginRefundShouldFail
     }
 
-    func customerInfo() async throws -> RevenueCat.CustomerInfo {
+    var customerInfoFetchPolicy: CacheFetchPolicy?
+    func customerInfo(fetchPolicy: CacheFetchPolicy) async throws -> RevenueCat.CustomerInfo {
+        customerInfoFetchPolicy = fetchPolicy
+
         if let customerInfoError {
             throw customerInfoError
         }