ACP2E-2213: Loading order through repository loads the order multiple times from DB

Author: o-iegorov

app/code/Magento/Sales/Test/Unit/Model/Order/ShippingAssignmentBuilderTest.php

@@ -0,0 +1,80 @@
+<?php
+/**
+ * Copyright © Magento, Inc. All rights reserved.
+ * See COPYING.txt for license details.
+ */
+declare(strict_types=1);
+
+namespace Magento\Sales\Test\Unit\Model\Order;
+
+use Magento\Sales\Api\Data\ShippingAssignmentInterface;
+use Magento\Sales\Api\Data\ShippingAssignmentInterfaceFactory;
+use Magento\Sales\Model\OrderFactory;
+use Magento\Sales\Api\Data\OrderInterface;
+use Magento\Sales\Model\Order\ShippingAssignmentBuilder;
+use Magento\Sales\Model\Order\ShippingBuilderFactory;
+use PHPUnit\Framework\MockObject\MockObject;
+use PHPUnit\Framework\TestCase;
+
+class ShippingAssignmentBuilderTest extends TestCase
+{
+    /**
+     * @var ShippingAssignmentBuilder
+     */
+    private $shippingAssignmentBuilder;
+
+    /**
+     * @var OrderFactory|MockObject
+     */
+    private $orderFactory;
+
+    /**
+     * @var ShippingAssignmentInterfaceFactory|Mockobject
+     */
+    private $shippingAssignmentFactory;
+
+    /**
+     * @var ShippingBuilderFactory|MockObject
+     */
+    private $shippingBuilderFactory;
+
+    /**
+     * @inheirtDoc
+     */
+    protected function setUp(): void
+    {
+        $this->orderFactory = $this->getMockBuilder(OrderFactory::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+        $this->shippingAssignmentFactory = $this->getMockBuilder(ShippingAssignmentInterfaceFactory::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+        $this->shippingBuilderFactory = $this->getMockBuilder(ShippingBuilderFactory::class)
+            ->disableOriginalConstructor()
+            ->getMock();
+        $this->shippingAssignmentBuilder = new ShippingAssignmentBuilder(
+            $this->orderFactory,
+            $this->shippingAssignmentFactory,
+            $this->shippingBuilderFactory
+        );
+    }
+
+    /**
+     * Test for case when order is provided instead of order_id
+     *
+     * @return void
+     */
+    public function testCreateWithOrder() : void
+    {
+        $order = $this->getMockBuilder(OrderInterface::class)
+            ->getMockForAbstractClass();
+        $this->shippingAssignmentBuilder->setOrder($order);
+        $order->expects($this->any())
+            ->method('getEntityId')
+            ->willReturn(1);
+        $this->orderFactory->expects($this->never())
+            ->method('create');
+
+        $this->shippingAssignmentBuilder->create();
+    }
+}