Add tests for the usage of created_via across layers

Author: iamgabrielma

Modules/Tests/NetworkingTests/Remote/OrdersRemoteTests.swift

@@ -48,6 +48,14 @@ final class OrdersRemoteTests: XCTestCase {
         XCTAssertFalse(fieldValues.contains(" "))
     }
 
+    func test_order_fields_parameter_includes_created_via_field() throws {
+        // When
+        let fieldValues = OrdersRemote.ParameterValues.fieldValues
+
+        // Then
+        XCTAssertTrue(fieldValues.contains("created_via"), "fieldValues should include 'created_via' field")
+    }
+
     // MARK: - Load All Orders Tests
 
     /// Verifies that loadAllOrders properly parses the `orders-load-all` sample response.

Modules/Tests/YosemiteTests/Stores/Order/OrdersUpsertUseCaseTests.swift

@@ -151,6 +151,36 @@ final class OrdersUpsertUseCaseTests: XCTestCase {
         XCTAssertEqual(storageTaxLine.toReadOnly(), taxLine)
     }
 
+    func test_order_with_createdVia_field_when_upsert_to_storage_then_field_is_persisted_correctly() throws {
+        // Given
+        let order = makeOrder().copy(siteID: defaultSiteID, orderID: 123, createdVia: "pos-rest-api")
+        let useCase = OrdersUpsertUseCase(storage: viewStorage)
+
+        // When
+        useCase.upsert([order])
+
+        // Then
+        let persistedOrder = try XCTUnwrap(viewStorage.loadOrder(siteID: defaultSiteID, orderID: 123))
+        XCTAssertEqual(persistedOrder.createdVia, "pos-rest-api")
+        XCTAssertEqual(persistedOrder.toReadOnly().createdVia, "pos-rest-api")
+    }
+
+    func test_order_with_createdVia_field_when_updated_then_field_is_updated_correctly() throws {
+        // Given
+        let originalOrder = makeOrder().copy(siteID: defaultSiteID, orderID: 123, createdVia: nil)
+        let useCase = OrdersUpsertUseCase(storage: viewStorage)
+        useCase.upsert([originalOrder])
+
+        // When
+        let updatedOrder = originalOrder.copy(createdVia: "pos-rest-api")
+        useCase.upsert([updatedOrder])
+
+        // Then
+        let persistedOrder = try XCTUnwrap(viewStorage.loadOrder(siteID: defaultSiteID, orderID: 123))
+        XCTAssertEqual(persistedOrder.createdVia, "pos-rest-api")
+        XCTAssertEqual(persistedOrder.toReadOnly().createdVia, "pos-rest-api")
+    }
+
     func test_it_persists_order_item_attributes_in_storage() throws {
         // Given
         let attributes = [

WooCommerce/WooCommerceTests/ViewRelated/Orders/OrderListCellViewModelTests.swift

@@ -61,4 +61,37 @@ final class OrderListCellViewModelTests: XCTestCase {
         XCTAssertEqual(accessoryView.image, expectedImage)
         XCTAssertEqual(accessoryView.tintColor, .tertiaryLabel)
     }
+
+    func test_salesChannel_when_createdVia_is_pos_rest_api_then_returns_POS() {
+        // Given
+        let order = MockOrders().sampleOrder().copy(createdVia: "pos-rest-api")
+
+        // When
+        let viewModel = OrderListCellViewModel(order: order, currencySettings: ServiceLocator.currencySettings)
+
+        // Then
+        XCTAssertEqual(viewModel.salesChannel, "POS")
+    }
+
+    func test_salesChannel_when_createdVia_is_nil_then_returns_nil() {
+        // Given
+        let order = MockOrders().sampleOrder().copy(createdVia: nil)
+
+        // When
+        let viewModel = OrderListCellViewModel(order: order, currencySettings: ServiceLocator.currencySettings)
+
+        // Then
+        XCTAssertNil(viewModel.salesChannel)
+    }
+
+    func test_salesChannel_when_createdVia_is_not_pos_rest_api_then_returns_nil() {
+        // Given
+        let order = MockOrders().sampleOrder().copy(createdVia: "checkout")
+
+        // When
+        let viewModel = OrderListCellViewModel(order: order, currencySettings: ServiceLocator.currencySettings)
+
+        // Then
+        XCTAssertNil(viewModel.salesChannel)
+    }
 }