AC-14690: Refresh order doesn't get latest custom attribute data

Author: wip44850

app/code/Magento/Sales/Test/Unit/Model/AdminOrder/CreateTest.php

@@ -34,6 +34,7 @@
 use Magento\Sales\Model\ResourceModel\Order\Item\Collection as ItemCollection;
 use Magento\Store\Api\Data\StoreInterface;
 use PHPUnit\Framework\MockObject\MockObject;
+use Magento\Customer\Api\CustomerRepositoryInterface;
 use PHPUnit\Framework\TestCase;
 
 /**
@@ -638,4 +639,95 @@ public static function setShippingAsBillingDataProvider(): array
             ]
         ];
     }
+    
+    public function testGetQuoteAssignsCustomerWhenCustomerIdPresent(): void
+    {
+        $quote = $this->getMockBuilder(Quote::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['assignCustomer'])
+            ->addMethods(['getCustomerId'])
+            ->getMock();
+
+        $quote->expects($this->once())
+            ->method('getCustomerId')
+            ->willReturn(self::CUSTOMER_ID);
+
+        $customerData = $this->getMockBuilder(CustomerInterface::class)
+            ->getMockForAbstractClass();
+
+        $customerRepository = $this->getMockBuilder(CustomerRepositoryInterface::class)
+            ->getMockForAbstractClass();
+        $customerRepository->expects($this->once())
+            ->method('getById')
+            ->with(self::CUSTOMER_ID)
+            ->willReturn($customerData);
+
+        $quote->expects($this->once())
+            ->method('assignCustomer')
+            ->with($customerData);
+
+        $this->sessionQuote->expects($this->once())
+            ->method('getQuote')
+            ->willReturn($quote);
+
+        $subject = $this->createAdminOrderCreateWithCustomerRepository($customerRepository);
+
+        $result = $subject->getQuote();
+
+        $this->assertSame($quote, $result);
+    }
+
+    public function testGetQuoteSkipsAssignWhenNoCustomerId(): void
+    {
+        $quote = $this->getMockBuilder(Quote::class)
+            ->disableOriginalConstructor()
+            ->onlyMethods(['assignCustomer'])
+            ->addMethods(['getCustomerId'])
+            ->getMock();
+
+        $quote->expects($this->once())
+            ->method('getCustomerId')
+            ->willReturn(0);
+
+        $quote->expects($this->never())
+            ->method('assignCustomer');
+
+        $customerRepository = $this->getMockBuilder(CustomerRepositoryInterface::class)
+            ->getMockForAbstractClass();
+        $customerRepository->expects($this->never())
+            ->method('getById');
+
+        $this->sessionQuote->expects($this->once())
+            ->method('getQuote')
+            ->willReturn($quote);
+
+        $subject = $this->createAdminOrderCreateWithCustomerRepository($customerRepository);
+
+        $result = $subject->getQuote();
+
+        $this->assertSame($quote, $result);
+    }
+
+    private function createAdminOrderCreateWithCustomerRepository(
+        CustomerRepositoryInterface $customerRepository
+    ): Create {
+        $objectManagerHelper = new ObjectManagerHelper($this);
+        return $objectManagerHelper->getObject(
+            Create::class,
+            [
+                '_objectManager' => $this->objectManager,
+                'messageManager' => $this->messageManager,
+                'quoteSession' => $this->sessionQuote,
+                'metadataFormFactory' => $this->formFactory,
+                'customerFactory' => $this->customerFactory,
+                'groupRepository' => $this->groupRepository,
+                'quoteItemUpdater' => $this->itemUpdater,
+                'customerMapper' => $this->customerMapper,
+                'dataObjectHelper' => $this->dataObjectHelper,
+                'quoteRepository' => $this->quoteRepository,
+                'quoteFactory' => $this->quoteFactory,
+                'customerRepository' => $customerRepository,
+            ]
+        );
+    }
 }