Address flaky TapToPayEducationViewModelTests (#14873)

woocommerce · Jan 16, 2025 · bf6e6c8 · bf6e6c8
2 parents 8fcf8f9 + 304a747
commit bf6e6c8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/...Classes/ViewRelated/CardPresentPayments/BuiltInCardReaderMerchantEducationPresenter.swift b/...Classes/ViewRelated/CardPresentPayments/BuiltInCardReaderMerchantEducationPresenter.swift
@@ -6,14 +6,14 @@ protocol BuiltInCardReaderMerchantEducationPresenting {
     func presentMerchantEducation(completion: @escaping () -> Void)
 }
 
-final class BuiltInCardReaderMerchantEducationPresenter: BuiltInCardReaderMerchantEducationPresenting {
+final class BuiltInCardReaderMerchantEducationPresenter: @preconcurrency BuiltInCardReaderMerchantEducationPresenting {
     private weak var rootViewController: ViewControllerPresenting?
 
     init(rootViewController: UIViewController) {
         self.rootViewController = rootViewController
     }
 
-    func presentMerchantEducation(completion: @escaping () -> Void) {
+    @MainActor func presentMerchantEducation(completion: @escaping () -> Void) {
         let viewController = UIHostingController(rootView: TapToPayEducationView(viewModel: .init(completion: { _ in
             completion()
         })))

diff --git a/...ed/Dashboard/Settings/CardReadersV2/Tap to Pay Education/TapToPayEducationViewModel.swift b/...ed/Dashboard/Settings/CardReadersV2/Tap to Pay Education/TapToPayEducationViewModel.swift
@@ -7,6 +7,7 @@ enum TapToPayEducationResult {
     case setUpTapToPay
 }
 
+@MainActor
 final class TapToPayEducationViewModel: ObservableObject {
     struct Action {
         let title: String

diff --git a/...s/ViewRelated/Dashboard/Settings/In-Person Payments/TapToPayEducationViewModelTests.swift b/...s/ViewRelated/Dashboard/Settings/In-Person Payments/TapToPayEducationViewModelTests.swift
@@ -2,6 +2,7 @@ import Testing
 import Combine
 @testable import WooCommerce
 
+@MainActor
 struct TapToPayEducationViewModelTests {
     private let cardReaderSupportDeterminer: MockCardReaderSupportDeterminer