Execute purchase check on purchase or purchase restoration

Issue #238
64characters · Sep 22, 2017 · 6cf5666 · 6cf5666
1 parent 8965f00
commit 6cf5666
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/Telephone/CallHistoryViewEventTarget.swift b/Telephone/CallHistoryViewEventTarget.swift
@@ -51,3 +51,19 @@ extension CallHistoryViewEventTarget: CallHistoryEventTarget {
         purchaseCheck.execute()
     }
 }
+
+extension CallHistoryViewEventTarget: StoreEventTarget {
+    func didPurchase() {
+        purchaseCheck.execute()
+    }
+
+    func didRestorePurchases() {
+        purchaseCheck.execute()
+    }
+
+    func didStartPurchasingProduct(withIdentifier identifier: String) {}
+    func didFailPurchasing(error: String) {}
+    func didCancelPurchasing() {}
+    func didFailRestoringPurchases(error: String) {}
+    func didCancelRestoringPurchases() {}
+}
diff --git a/TelephoneTests/CallHistoryViewEventTargetTests.swift b/TelephoneTests/CallHistoryViewEventTargetTests.swift
@@ -76,6 +76,34 @@ final class CallHistoryViewEventTargetTests: XCTestCase {
         XCTAssertTrue(purchaseCheck.didCallExecute)
     }
 
+    func testExecutesPurchaseCheckUseCaseOnDidPurchase() {
+        let purchaseCheck = UseCaseSpy()
+        let sut = CallHistoryViewEventTarget(
+            recordsGet: UseCaseSpy(),
+            purchaseCheck: purchaseCheck,
+            recordRemove: CallHistoryRecordRemoveUseCaseFactorySpy(remove: UseCaseSpy()),
+            callMake: CallHistoryCallMakeUseCaseFactorySpy(callMake: UseCaseSpy())
+        )
+
+        sut.didPurchase()
+
+        XCTAssertTrue(purchaseCheck.didCallExecute)
+    }
+
+    func testExecutesPurchaseCheckUseCaseOnDidRestorePurchases() {
+        let purchaseCheck = UseCaseSpy()
+        let sut = CallHistoryViewEventTarget(
+            recordsGet: UseCaseSpy(),
+            purchaseCheck: purchaseCheck,
+            recordRemove: CallHistoryRecordRemoveUseCaseFactorySpy(remove: UseCaseSpy()),
+            callMake: CallHistoryCallMakeUseCaseFactorySpy(callMake: UseCaseSpy())
+        )
+
+        sut.didRestorePurchases()
+
+        XCTAssertTrue(purchaseCheck.didCallExecute)
+    }
+
     func testCreatesCallHistoryCallMakeUseCaseWithExpectedIdentifierOnDidPickRecord() {
         let factory = CallHistoryCallMakeUseCaseFactorySpy(callMake: UseCaseSpy())
         let sut = CallHistoryViewEventTarget(