[MBL-1328] Pledge Button Active/Inactive States

Library/ViewModels/PostCampaignCheckoutViewModel.swift

@@ -86,8 +86,9 @@ public class PostCampaignCheckoutViewModel: PostCampaignCheckoutViewModelType,
     .skipNil()
 
     let context = initialData.map(\.context)
+    let project = initialData.map(\.project)
     let checkoutId = initialData.map(\.checkoutId)
-    let baseReward = initialData.map(\.rewards).map(\.first)
+    let baseReward = initialData.map(\.rewards).map(\.first).skipNil()
 
     let configurePaymentMethodsData = Signal.merge(
       initialData,
@@ -109,14 +110,40 @@ public class PostCampaignCheckoutViewModel: PostCampaignCheckoutViewModelType,
       .map { _ in AppEnvironment.current.currentUser }
       .map(isNotNil)
 
+    let notChangingPaymentMethod = context.map { context in
+      context.isUpdating && context != .changePaymentMethod
+    }
+    .filter(isTrue)
+
+    let paymentMethodSelected = self.creditCardSelectedProperty.signal.skipNil()
+
+    let paymentMethodChangedAndValid = Signal.merge(
+      notChangingPaymentMethod.mapConst(false),
+      Signal.combineLatest(
+        project,
+        baseReward,
+        paymentMethodSelected.map { (source: PaymentSourceSelected, _, _) in source },
+        context
+      )
+      .map(paymentMethodValid)
+    )
+
+    let isEnabled = Signal.merge(
+      self.viewDidLoadProperty.signal.mapConst(false)
+        .take(until: paymentMethodChangedAndValid.ignoreValues()),
+      paymentMethodChangedAndValid
+    )
+    .skipRepeats()
+
     self.configurePledgeViewCTAContainerView = Signal.combineLatest(
       isLoggedIn,
+      isEnabled,
       context
     )
-    .map { isLoggedIn, context in
+    .map { isLoggedIn, isEnabled, context in
       PledgeViewCTAContainerViewData(
         isLoggedIn: isLoggedIn,
-        isEnabled: true, // Pledge button never needs to be disabled on checkout page.
+        isEnabled: isEnabled,
         context: context,
         willRetryPaymentMethod: false // Only retry in the `fixPaymentMethod` context.
       )
@@ -394,9 +421,7 @@ public class PostCampaignCheckoutViewModel: PostCampaignCheckoutViewModelType,
 
     let thanksPageData = Signal.combineLatest(initialData, baseReward)
       .map { initialData, baseReward -> ThanksPageData? in
-        guard let reward = baseReward else { return nil }
-
-        return (initialData.project, reward, nil, initialData.total)
+        (initialData.project, baseReward, nil, initialData.total)
       }
 
     self.checkoutComplete = thanksPageData.skipNil()
@@ -514,3 +539,26 @@ public class PostCampaignCheckoutViewModel: PostCampaignCheckoutViewModelType,
   public var inputs: PostCampaignCheckoutViewModelInputs { return self }
   public var outputs: PostCampaignCheckoutViewModelOutputs { return self }
 }
+
+private func paymentMethodValid(
+  project: Project,
+  reward: Reward,
+  paymentSource: PaymentSourceSelected,
+  context: PledgeViewContext
+) -> Bool {
+  guard
+    let backedPaymentSourceId = project.personalization.backing?.paymentSource?.id,
+    context.isUpdating,
+    userIsBacking(reward: reward, inProject: project)
+  else {
+    return true
+  }
+
+  if project.personalization.backing?.status == .errored {
+    return true
+  } else if backedPaymentSourceId != paymentSource.savedCreditCardId {
+    return true
+  }
+
+  return false
+}

Library/ViewModels/PostCampaignCheckoutViewModelTests.swift

@@ -14,13 +14,24 @@ final class PostCampaignCheckoutViewModelTests: TestCase {
   fileprivate let checkoutComplete = TestObserver<ThanksPageData, Never>()
   fileprivate let processingViewIsHidden = TestObserver<Bool, Never>()
   fileprivate let validateCheckoutSuccess = TestObserver<PaymentSourceValidation, Never>()
+
+  private let configurePledgeViewCTAContainerViewIsLoggedIn = TestObserver<Bool, Never>()
+  private let configurePledgeViewCTAContainerViewIsEnabled = TestObserver<Bool, Never>()
+  private let configurePledgeViewCTAContainerViewContext = TestObserver<PledgeViewContext, Never>()
 
   override func setUp() {
     super.setUp()
     self.vm.goToApplePayPaymentAuthorization.observe(self.goToApplePayPaymentAuthorization.observer)
     self.vm.checkoutComplete.observe(self.checkoutComplete.observer)
     self.vm.processingViewIsHidden.observe(self.processingViewIsHidden.observer)
     self.vm.validateCheckoutSuccess.observe(self.validateCheckoutSuccess.observer)
+
+    self.vm.outputs.configurePledgeViewCTAContainerView.map { $0.0 }
+      .observe(self.configurePledgeViewCTAContainerViewIsLoggedIn.observer)
+    self.vm.outputs.configurePledgeViewCTAContainerView.map { $0.1 }
+      .observe(self.configurePledgeViewCTAContainerViewIsEnabled.observer)
+    self.vm.outputs.configurePledgeViewCTAContainerView.map { $0.2 }
+      .observe(self.configurePledgeViewCTAContainerViewContext.observer)
   }
 
   func testApplePayAuthorization_noReward_isCorrect() {
@@ -385,8 +396,14 @@ final class PostCampaignCheckoutViewModelTests: TestCase {
       self.vm.viewDidLoad()
 
       let paymentSource = PaymentSourceSelected.paymentIntentClientSecret("123")
+
       self.vm.inputs
         .creditCardSelected(source: paymentSource, paymentMethodId: "123", isNewPaymentMethod: true)
+
+      self.configurePledgeViewCTAContainerViewIsLoggedIn.assertValues([false, false])
+      self.configurePledgeViewCTAContainerViewIsEnabled.assertValues([false, true])
+      self.configurePledgeViewCTAContainerViewContext.assertValues([.latePledge, .latePledge])
+
       self.vm.inputs.submitButtonTapped()
 
       self.processingViewIsHidden.assertLastValue(false)
@@ -400,6 +417,7 @@ final class PostCampaignCheckoutViewModelTests: TestCase {
 
       self.checkoutComplete.assertDidEmitValue()
       self.processingViewIsHidden.assertLastValue(true)
+
     }
   }
 
@@ -431,6 +449,11 @@ final class PostCampaignCheckoutViewModelTests: TestCase {
       let paymentSource = PaymentSourceSelected.paymentIntentClientSecret("123")
       self.vm.inputs
         .creditCardSelected(source: paymentSource, paymentMethodId: "123", isNewPaymentMethod: true)
+
+      self.configurePledgeViewCTAContainerViewIsLoggedIn.assertValues([false, false])
+      self.configurePledgeViewCTAContainerViewIsEnabled.assertValues([false, true])
+      self.configurePledgeViewCTAContainerViewContext.assertValues([.latePledge, .latePledge])
+
       self.vm.inputs.submitButtonTapped()
 
       self.processingViewIsHidden.assertLastValue(false)
@@ -471,6 +494,11 @@ final class PostCampaignCheckoutViewModelTests: TestCase {
       let paymentSource = PaymentSourceSelected.paymentIntentClientSecret("123")
       self.vm.inputs
         .creditCardSelected(source: paymentSource, paymentMethodId: "123", isNewPaymentMethod: true)
+
+      self.configurePledgeViewCTAContainerViewIsLoggedIn.assertValues([false, false])
+      self.configurePledgeViewCTAContainerViewIsEnabled.assertValues([false, true])
+      self.configurePledgeViewCTAContainerViewContext.assertValues([.latePledge, .latePledge])
+
       self.vm.inputs.submitButtonTapped()
 
       self.processingViewIsHidden.assertLastValue(false)
@@ -485,4 +513,36 @@ final class PostCampaignCheckoutViewModelTests: TestCase {
       self.processingViewIsHidden.assertLastValue(true)
     }
   }
+
+  func testPledgeViewCTA_LoggedIn_State() {
+    let mockService = MockService(serverConfig: ServerConfig.staging)
+
+    withEnvironment(apiService: mockService, currentUser: .template) {
+      let project = Project.cosmicSurgery
+      let reward = Reward.noReward |> Reward.lens.minimum .~ 5
+
+      let data = PostCampaignCheckoutData(
+        project: project,
+        rewards: [reward],
+        selectedQuantities: [:],
+        bonusAmount: 0,
+        total: 5,
+        shipping: nil,
+        refTag: nil,
+        context: .latePledge,
+        checkoutId: "0"
+      )
+
+      self.vm.inputs.configure(with: data)
+      self.vm.inputs.viewDidLoad()
+
+      let paymentSource = PaymentSourceSelected.paymentIntentClientSecret("123")
+      self.vm.inputs
+        .creditCardSelected(source: paymentSource, paymentMethodId: "123", isNewPaymentMethod: true)
+
+      self.configurePledgeViewCTAContainerViewIsLoggedIn.assertValues([true, true])
+      self.configurePledgeViewCTAContainerViewIsEnabled.assertValues([false, true])
+      self.configurePledgeViewCTAContainerViewContext.assertValues([.latePledge, .latePledge])
+    }
+  }
 }