Add unit tests

Author: hichamboushaba

WooCommerce/src/test/kotlin/com/woocommerce/android/ui/orders/OrderTestUtils.kt

@@ -213,7 +213,7 @@ object OrderTestUtils {
             "    \"subtotal_tax\":\"0.00\",\n" +
             "    \"total\":\"10.00\",\n" +
             "    \"total_tax\":\"0.00\",\n" +
-            "    \"taxes\":[],\n" +
+            "    \"taxes\":[{\"id\":1, \"total\":2}, {\"id\":2, \"total\":5}],\n" +
             "    \"meta_data\":[],\n" +
             "    \"sku\":null,\n" +
             "    \"price\":10\n" +
@@ -241,7 +241,7 @@ object OrderTestUtils {
                 "   \"instance_id\":\"0\",\n" +
                 "   \"total\":\"30.00\",\n" +
                 "   \"total_tax\":\"0.00\",\n" +
-                "   \"taxes\":[],\n" +
+                "   \"taxes\":[{\"id\":1, \"total\":2}, {\"id\":2, \"total\":5}],\n" +
                 "   \"meta_data\":[]}]",
         )
     }
@@ -258,7 +258,7 @@ object OrderTestUtils {
                 "    \"subtotal_tax\":\"0.00\",\n" +
                 "    \"total\":\"10.00\",\n" +
                 "    \"total_tax\":\"0.00\",\n" +
-                "    \"taxes\":[],\n" +
+                "    \"taxes\":[{\"id\":1, \"total\":2}, {\"id\":2, \"total\":5}],\n" +
                 "    \"meta_data\":[],\n" +
                 "    \"sku\":null,\n" +
                 "    \"price\":10\n" +
@@ -305,7 +305,7 @@ object OrderTestUtils {
                 "    \"subtotal_tax\":\"0.00\",\n" +
                 "    \"total\":\"10.00\",\n" +
                 "    \"total_tax\":\"0.00\",\n" +
-                "    \"taxes\":[],\n" +
+                "    \"taxes\":[{\"id\":1, \"total\":2}, {\"id\":2, \"total\":5}],\n" +
                 "    \"meta_data\":[],\n" +
                 "    \"sku\":null,\n" +
                 "    \"price\":10\n" +
@@ -319,7 +319,7 @@ object OrderTestUtils {
                 "   \"instance_id\":\"0\",\n" +
                 "   \"total\":\"30.00\",\n" +
                 "   \"total_tax\":\"0.00\",\n" +
-                "   \"taxes\":[],\n" +
+                "   \"taxes\":[{\"id\":1, \"total\":2}, {\"id\":2, \"total\":5}],\n" +
                 "   \"meta_data\":[]},\n" +
                 "{  " +
                 "\"id\":120,\n" +
@@ -328,7 +328,7 @@ object OrderTestUtils {
                 "   \"instance_id\":\"0\",\n" +
                 "   \"total\":\"20.00\",\n" +
                 "   \"total_tax\":\"0.00\",\n" +
-                "   \"taxes\":[],\n" +
+                "   \"taxes\":[{\"id\":1, \"total\":3}, {\"id\":2, \"total\":4}],\n" +
                 "   \"meta_data\":[]\n" +
                 "}]",
         )
@@ -377,7 +377,8 @@ object OrderTestUtils {
     fun generateTestOrderItems(
         count: Int = 1,
         productId: Long = -1,
-        quantity: Float = 1F
+        quantity: Float = 1F,
+        taxes: (Int) -> List<Order.LineTaxEntry> = { emptyList() }
     ): List<Item> {
         val list = mutableListOf<Item>()
         for (i in 1..count) {
@@ -394,7 +395,8 @@ object OrderTestUtils {
                     totalTax = BigDecimal.ZERO,
                     total = BigDecimal("10"),
                     variationId = 0,
-                    attributesList = emptyList()
+                    attributesList = emptyList(),
+                    taxes = taxes(i),
                 )
             )
         }

WooCommerce/src/test/kotlin/com/woocommerce/android/ui/payments/refunds/IssueRefundViewModelTest.kt

@@ -15,6 +15,7 @@ import com.woocommerce.android.ui.orders.OrderTestUtils
 import com.woocommerce.android.ui.orders.details.OrderDetailRepository
 import com.woocommerce.android.ui.payments.refunds.IssueRefundViewModel.RefundByItemsViewState
 import com.woocommerce.android.util.CurrencyFormatter
+import com.woocommerce.android.util.getOrAwaitValue
 import com.woocommerce.android.viewmodel.BaseUnitTest
 import com.woocommerce.android.viewmodel.MultiLiveEvent
 import com.woocommerce.android.viewmodel.ResourceProvider
@@ -1206,4 +1207,67 @@ class IssueRefundViewModelTest : BaseUnitTest() {
             assertThat(viewModel.refundItems.value!![1].maxQuantity).isEqualTo(2.0F)
             assertThat(viewModel.refundItems.value!![2].maxQuantity).isEqualTo(2.0F)
         }
+
+    @Test
+    fun `when preparing refund for products, then pass correct tax rates`() {
+        testBlocking {
+            val order = OrderTestUtils.generateOrderWithOneShipping()
+            whenever(orderStore.getOrderByIdAndSite(any(), any())).thenReturn(order)
+
+            initViewModel()
+            viewModel.onSelectButtonTapped()
+
+            val items = viewModel.refundItems.getOrAwaitValue()
+            items.map { it.toDataModel() }.forEach { refundItem ->
+                val item = order.getLineItemList().first { it.id == refundItem.itemId }
+                assertThat(refundItem.refundTax).allMatch { refundTaxItem ->
+                    val tax = item.taxes!!.first { it.rateId == refundTaxItem.taxRateId }
+                    tax.rateId == refundTaxItem.taxRateId &&
+                        tax.total.isEqualTo(refundTaxItem.refundTotal)
+                }
+            }
+        }
+    }
+
+    @Test
+    fun `when preparing refund for shipping, then pass correct tax rates`() {
+        testBlocking {
+            val order = OrderTestUtils.generateOrderWithOneShipping()
+            whenever(orderStore.getOrderByIdAndSite(any(), any())).thenReturn(order)
+
+            initViewModel()
+            viewModel.onSelectButtonTapped()
+
+            val shippingLines = viewModel.refundShippingLines.getOrAwaitValue()
+            shippingLines.map { it.toDataModel() }.forEach { refundItem ->
+                val shippingLine = order.getShippingLineList().first { it.id == refundItem.itemId }
+                assertThat(refundItem.refundTax).allMatch { refundTaxItem ->
+                    val tax = shippingLine.taxes!!.first { it.rateId == refundTaxItem.taxRateId }
+                    tax.rateId == refundTaxItem.taxRateId &&
+                        tax.total.isEqualTo(refundTaxItem.refundTotal)
+                }
+            }
+        }
+    }
+
+    @Test
+    fun `when preparing refund for fees, then pass correct tax rates`() {
+        testBlocking {
+            val order = OrderTestUtils.generateOrderWithOneShipping()
+            whenever(orderStore.getOrderByIdAndSite(any(), any())).thenReturn(order)
+
+            initViewModel()
+            viewModel.onSelectButtonTapped()
+
+            val feeLines = viewModel.refundFeeLines.getOrAwaitValue()
+            feeLines.map { it.toDataModel() }.forEach { refundItem ->
+                val feeLine = order.getFeeLineList().first { it.id == refundItem.itemId }
+                assertThat(refundItem.refundTax).allMatch { refundTaxItem ->
+                    val tax = feeLine.taxes!!.first { it.rateId == refundTaxItem.taxRateId }
+                    tax.rateId == refundTaxItem.taxRateId &&
+                        tax.total.isEqualTo(refundTaxItem.refundTotal)
+                }
+            }
+        }
+    }
 }