Test restoring value for the purchase, restoring, and transaction codepaths.

Author: edpizzi

packages/in_app_purchase/in_app_purchase_storekit/example/shared/RunnerTests/InAppPurchaseStoreKit2PluginTests.swift

@@ -8,10 +8,14 @@ import XCTest
 @testable import in_app_purchase_storekit
 
 final class FakeIAP2Callback: InAppPurchase2CallbackAPIProtocol {
+
+  public var lastUpdate: [in_app_purchase_storekit.SK2TransactionMessage] = []
+
   func onTransactionsUpdated(
     newTransactions newTransactionsArg: [in_app_purchase_storekit.SK2TransactionMessage],
     completion: @escaping (Result<Void, in_app_purchase_storekit.PigeonError>) -> Void
   ) {
+    lastUpdate = newTransactionsArg
     // We should only write to a flutter channel from the main thread.
     XCTAssertTrue(Thread.isMainThread)
   }
@@ -21,6 +25,7 @@ final class FakeIAP2Callback: InAppPurchase2CallbackAPIProtocol {
 final class InAppPurchase2PluginTests: XCTestCase {
   private var session: SKTestSession!
   private var plugin: InAppPurchasePlugin!
+  private var callback: FakeIAP2Callback = FakeIAP2Callback()
 
   override func setUp() async throws {
     try await super.setUp()
@@ -33,7 +38,7 @@ final class InAppPurchase2PluginTests: XCTestCase {
     plugin = InAppPurchasePluginStub(receiptManager: FIAPReceiptManagerStub()) { request in
       DefaultRequestHandler(requestHandler: FIAPRequestHandler(request: request))
     }
-    plugin.transactionCallbackAPI = FakeIAP2Callback()
+    plugin.transactionCallbackAPI = callback
     try plugin.startListeningToTransactions()
   }
 
@@ -86,11 +91,17 @@ final class InAppPurchase2PluginTests: XCTestCase {
 
     await fulfillment(of: [purchaseExpectation], timeout: 5)
 
+    XCTAssert(callback.lastUpdate.count == 1)
+    XCTAssert(
+      callback.lastUpdate.first?.restoring == false,
+      "Ordinary purchase updates should not be marked as restoring")
+
     plugin.transactions {
       result in
       switch result {
       case .success(let transactions):
         XCTAssert(transactions.count == 1)
+        XCTAssert(transactions.first?.restoring == false)
         transactionExpectation.fulfill()
       case .failure(let error):
         XCTFail("Getting transactions should NOT fail. Failed with \(error)")
@@ -376,6 +387,9 @@ final class InAppPurchase2PluginTests: XCTestCase {
     }
     await fulfillment(of: [purchaseExpectation], timeout: 5)
 
+    XCTAssert(callback.lastUpdate.count == 1)
+    XCTAssert(callback.lastUpdate.first?.restoring == false)
+
     plugin.restorePurchases { result in
       switch result {
       case .success():
@@ -385,6 +399,9 @@ final class InAppPurchase2PluginTests: XCTestCase {
       }
     }
     await fulfillment(of: [restoreExpectation], timeout: 5)
+
+    XCTAssert(callback.lastUpdate.count == 1)
+    XCTAssert(callback.lastUpdate.first?.restoring == true)
   }
 
   func testFinishTransaction() async throws {