ACP2E-2910: Order Rest API call is taking a long time to execute

Author: aplapana

app/code/Magento/GiftMessage/Test/Unit/Model/Plugin/OrderGetTest.php

@@ -86,15 +86,21 @@ protected function setUp(): void
             ->disableOriginalConstructor()
             ->addMethods(['getGiftMessageId'])
             ->getMockForAbstractClass();
-        $this->orderExtensionMock = $this->createMock(OrderExtension::class);
+        $this->orderExtensionMock = $this->getMockBuilder(OrderExtension::class)
+            ->disableOriginalConstructor()
+            ->addMethods(['getGiftMessage', 'setGiftMessage'])
+            ->getMock();
         $this->giftMessageMock = $this->createMock(
             MessageInterface::class
         );
         $this->orderItemMock = $this->getMockBuilder(OrderItemInterface::class)
             ->disableOriginalConstructor()
             ->addMethods(['getGiftMessageId'])
             ->getMockForAbstractClass();
-        $this->orderItemExtensionMock = $this->createMock(OrderItemExtension::class);
+        $this->orderItemExtensionMock = $this->getMockBuilder(OrderItemExtension::class)
+            ->disableOriginalConstructor()
+            ->addMethods(['getGiftMessage', 'setGiftMessage'])
+            ->getMock();
         $this->orderRepositoryMock = $this->createMock(
             \Magento\Sales\Api\OrderRepositoryInterface::class
         );

app/code/Magento/Sales/Test/Unit/Model/OrderRepositoryTest.php

@@ -139,6 +139,15 @@ public function testGetList()
             ->setMethods(['setKey', 'setValue'])->getMockForAbstractClass();
 
         $extensionAttributes = $this->getMockBuilder(OrderExtensionInterface::class)
+            ->addMethods(
+                [
+                    'getShippingAssignments',
+                    'setAppliedTaxes',
+                    'setConvertingFromQuote',
+                    'setItemAppliedTaxes',
+                    'setPaymentAdditionalInfo'
+                ]
+            )
             ->disableOriginalConstructor()
             ->getMockForAbstractClass();
         $shippingAssignmentBuilder = $this->createMock(
@@ -183,6 +192,7 @@ public function testSave()
             ->getMock();
         $orderEntity = $this->createMock(Order::class);
         $extensionAttributes = $this->getMockBuilder(OrderExtensionInterface::class)
+            ->addMethods(['getShippingAssignments'])
             ->disableOriginalConstructor()
             ->getMockForAbstractClass();
         $shippingAssignment = $this->getMockBuilder(ShippingAssignment::class)
@@ -230,6 +240,16 @@ public function testGet()
         $paymentMock->expects($this->once())->method('getAdditionalInformation')->willReturn($paymentInfo);
 
         $orderExtension = $this->getMockBuilder(OrderExtensionInterface::class)
+            ->addMethods(
+                [
+                    'getShippingAssignments',
+                    'setAppliedTaxes',
+                    'setConvertingFromQuote',
+                    'setItemAppliedTaxes',
+                    'setPaymentAdditionalInfo',
+                    'setShippingAssignments'
+                ]
+            )
             ->disableOriginalConstructor()
             ->getMockForAbstractClass();
         $orderExtension->expects($this->once())->method('getShippingAssignments')->willReturn(null);