Add test cases for CardReaderSettingsConnectedViewModel.connectedReaderModel.

Author: jaclync

WooCommerce/Classes/ViewRelated/Orders/Collect Payments/CollectOrderPaymentUseCase.swift

@@ -53,6 +53,7 @@ final class CollectOrderPaymentUseCase: NSObject, CollectOrderPaymentProtocol {
     ///
     private var readerSubscription: AnyCancellable?
 
+    /// Stores the connected card reader for analytics.
     private var connectedReader: CardReader?
 
     /// Closure to inform when the full flow has been completed, after receipt management.

WooCommerce/WooCommerceTests/Mocks/MockCardReader.swift

@@ -33,4 +33,15 @@ struct MockCardReader {
                    readerType: .chipper,
                    locationId: "st_simulated")
     }
+
+    static func wisePad3() -> CardReader {
+        CardReader(serial: "WPC323026000412",
+                   vendorIdentifier: "SIMULATOR",
+                   name: "Simulated WISEPAD 3",
+                   status: .init(connected: false, remembered: false),
+                   softwareVersion: nil,
+                   batteryLevel: 0.5,
+                   readerType: .wisepad3,
+                   locationId: nil)
+    }
 }

WooCommerce/WooCommerceTests/ViewRelated/Dashboard/CardReaderSettings/CardReaderSettingsConnectedViewModelTests.swift

@@ -410,6 +410,50 @@ final class CardReaderSettingsConnectedViewModelTests: XCTestCase {
         // Then
         XCTAssertTrue(viewModel.optionalReaderUpdateAvailable)
     }
+
+    func test_when_connected_to_one_reader_it_sets_connectedReaderModel() {
+        XCTAssertEqual(viewModel.connectedReaderModel, MockCardReader.bbposChipper2XBT().readerType.model)
+    }
+
+    func test_when_connected_to_two_readers_it_sets_connectedReaderModel_from_the_first_reader() {
+        // Given
+        mockStoresManager = MockCardPresentPaymentsStoresManager(
+            connectedReaders: [MockCardReader.wisePad3(), MockCardReader.bbposChipper2XBT()],
+            discoveredReaders: [],
+            sessionManager: SessionManager.testingInstance
+        )
+        ServiceLocator.setStores(mockStoresManager)
+
+        analyticsProvider = MockAnalyticsProvider()
+        ServiceLocator.setAnalytics(WooAnalytics(analyticsProvider: analyticsProvider))
+
+        viewModel = CardReaderSettingsConnectedViewModel(didChangeShouldShow: nil,
+                                                         configuration: Mocks.configuration,
+                                                         delayToShowUpdateSuccessMessage: .milliseconds(1))
+
+        // Then
+        XCTAssertEqual(viewModel.connectedReaderModel, "WISEPAD_3")
+    }
+
+    func test_when_not_connected_to_any_readers_it_sets_connectedReaderModel_to_nil() {
+        // Given
+        mockStoresManager = MockCardPresentPaymentsStoresManager(
+            connectedReaders: [],
+            discoveredReaders: [],
+            sessionManager: SessionManager.testingInstance
+        )
+        ServiceLocator.setStores(mockStoresManager)
+
+        analyticsProvider = MockAnalyticsProvider()
+        ServiceLocator.setAnalytics(WooAnalytics(analyticsProvider: analyticsProvider))
+
+        viewModel = CardReaderSettingsConnectedViewModel(didChangeShouldShow: nil,
+                                                         configuration: Mocks.configuration,
+                                                         delayToShowUpdateSuccessMessage: .milliseconds(1))
+
+        // Then
+        XCTAssertNil(viewModel.connectedReaderModel)
+    }
 }
 
 private extension CardReaderSettingsConnectedViewModelTests {