MC-39864: [Magento Cloud] - Tax Miscalculation

Author: Serhii Balko

app/code/Magento/Sales/Test/Unit/Model/Order/Creditmemo/Total/TaxTest.php

@@ -192,7 +192,7 @@ public function collectDataProvider()
                     'base_shipping_amount' => 30,
                     'tax_amount' => 0.82,
                     'base_tax_amount' => 0.82,
-                    'invoice' => new MagentoObject(
+                    'invoice' => $this->createInvoiceMock(
                         [
                             'tax_amount' => 24.33,
                             'base_tax_amount' => 24.33,
@@ -283,7 +283,7 @@ public function collectDataProvider()
                     'base_shipping_amount' => 30,
                     'tax_amount' => 0.82 * $currencyRatio,
                     'base_tax_amount' => 0.82,
-                    'invoice' => new MagentoObject(
+                    'invoice' => $this->createInvoiceMock(
                         [
                             'tax_amount' => 24.33 * $currencyRatio,
                             'base_tax_amount' => 24.33,
@@ -360,7 +360,7 @@ public function collectDataProvider()
                     'base_shipping_amount' => 30,
                     'tax_amount' => 1.65,
                     'base_tax_amount' => 1.65,
-                    'invoice' => new MagentoObject(
+                    'invoice' => $this->createInvoiceMock(
                         [
                             'tax_amount' => 11.14,
                             'base_tax_amount' => 11.14,
@@ -438,7 +438,7 @@ public function collectDataProvider()
                     'base_shipping_amount' => 0,
                     'tax_amount' => 0.82,
                     'base_tax_amount' => 0.82,
-                    'invoice' => new MagentoObject(
+                    'invoice' => $this->createInvoiceMock(
                         [
                             'tax_amount' => 16.09,
                             'base_tax_amount' => 16.09,
@@ -587,7 +587,7 @@ public function collectDataProvider()
                     'base_grand_total' => 60.82,
                     'tax_amount' => 0.82,
                     'base_tax_amount' => 0.82,
-                    'invoice' => new MagentoObject(
+                    'invoice' => $this->createInvoiceMock(
                         [
                             'tax_amount' => 16.09,
                             'base_tax_amount' => 16.09,
@@ -779,4 +779,40 @@ protected function getCreditmemoItem($creditmemoItemData)
         $creditmemoItem->setData('qty', $creditmemoItemData['qty']);
         return $creditmemoItem;
     }
+
+    /**
+     * Create invoice mock object
+     *
+     * @param array $data
+     * @return MockObject|Invoice
+     */
+    private function createInvoiceMock(array $data): MockObject
+    {
+        /** @var MockObject|Invoice $invoice */
+        $invoice = $this->getMockBuilder(Invoice::class)
+            ->disableOriginalConstructor()
+            ->disableOriginalClone()
+            ->disableArgumentCloning()
+            ->disallowMockingUnknownTypes()
+            ->addMethods(['getBaseShippingDiscountTaxCompensationAmount'])
+            ->onlyMethods([
+                'getTaxAmount',
+                'getBaseTaxAmount',
+                'getShippingTaxAmount',
+                'getBaseShippingTaxAmount',
+                'getShippingDiscountTaxCompensationAmount'
+            ])
+            ->getMock();
+
+        $invoice->method('getTaxAmount')->willReturn($data['tax_amount'] ?? 0);
+        $invoice->method('getBaseTaxAmount')->willReturn($data['base_tax_amount'] ?? 0);
+        $invoice->method('getShippingTaxAmount')->willReturn($data['shipping_tax_amount'] ?? 0);
+        $invoice->method('getBaseShippingTaxAmount')->willReturn($data['base_shipping_tax_amount'] ?? 0);
+        $invoice->method('getShippingDiscountTaxCompensationAmount')
+            ->willReturn($data['shipping_discount_tax_compensation_amount'] ?? 0);
+        $invoice->method('getBaseShippingDiscountTaxCompensationAmount')
+            ->willReturn($data['base_shipping_discount_tax_compensation_amount'] ?? 0);
+
+        return $invoice;
+    }
 }