MAGETWO-50972: Move to Shopping Cart action does not work inside creation offline order

- Add unit test on call getForCustomer() method after get()

Author: viktym

app/code/Magento/Quote/Test/Unit/Model/QuoteRepositoryTest.php

@@ -11,6 +11,7 @@
 
 use Magento\Framework\Api\SortOrder;
 use Magento\Framework\Api\ExtensionAttribute\JoinProcessorInterface;
+use Magento\Quote\Model\QuoteRepository\LoadHandler;
 
 class QuoteRepositoryTest extends \PHPUnit_Framework_TestCase
 {
@@ -54,6 +55,11 @@ class QuoteRepositoryTest extends \PHPUnit_Framework_TestCase
      */
     private $extensionAttributesJoinProcessorMock;
 
+    /**
+     * @var LoadHandler|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $loadHandlerMock;
+
     protected function setUp()
     {
         $objectManager = new \Magento\Framework\TestFramework\Unit\Helper\ObjectManager($this);
@@ -107,6 +113,12 @@ protected function setUp()
                 'extensionAttributesJoinProcessor' => $this->extensionAttributesJoinProcessorMock
             ]
         );
+
+        $this->loadHandlerMock = $this->getMock(LoadHandler::class, [], [], '', false);
+        $reflection = new \ReflectionClass(get_class($this->model));
+        $reflectionProperty = $reflection->getProperty('loadHandler');
+        $reflectionProperty->setAccessible(true);
+        $reflectionProperty->setValue($this->model, $this->loadHandlerMock);
     }
 
     /**
@@ -132,23 +144,74 @@ public function testGetWithExceptionById()
 
     public function testGet()
     {
-        $this->markTestSkipped('MAGETWO-48531');
         $cartId = 15;
 
-        $this->quoteFactoryMock->expects($this->once())->method('create')->willReturn($this->quoteMock);
-        $this->storeManagerMock->expects($this->once())->method('getStore')->willReturn($this->storeMock);
-        $this->storeMock->expects($this->once())->method('getId')->willReturn(1);
-        $this->quoteMock->expects($this->never())->method('setSharedStoreIds');
-        $this->quoteMock->expects($this->once())
+        $this->quoteFactoryMock->expects(static::once())
+            ->method('create')
+            ->willReturn($this->quoteMock);
+        $this->storeManagerMock->expects(static::once())
+            ->method('getStore')
+            ->willReturn($this->storeMock);
+        $this->storeMock->expects(static::once())
+            ->method('getId')
+            ->willReturn(1);
+        $this->quoteMock->expects(static::never())
+            ->method('setSharedStoreIds');
+        $this->quoteMock->expects(static::once())
             ->method('load')
             ->with($cartId)
             ->willReturn($this->storeMock);
-        $this->quoteMock->expects($this->once())->method('getId')->willReturn($cartId);
+        $this->quoteMock->expects(static::once())
+            ->method('getId')
+            ->willReturn($cartId);
+        $this->quoteMock->expects(static::never())
+            ->method('getCustomerId');
+        $this->loadHandlerMock->expects(static::once())
+            ->method('load')
+            ->with($this->quoteMock);
 
         $this->assertEquals($this->quoteMock, $this->model->get($cartId));
         $this->assertEquals($this->quoteMock, $this->model->get($cartId));
     }
 
+    public function testGetForCustomerAfterGet()
+    {
+        $cartId = 15;
+        $customerId = 23;
+
+        $this->quoteFactoryMock->expects(static::exactly(2))
+            ->method('create')
+            ->willReturn($this->quoteMock);
+        $this->storeManagerMock->expects(static::exactly(2))
+            ->method('getStore')
+            ->willReturn($this->storeMock);
+        $this->storeMock->expects(static::exactly(2))
+            ->method('getId')
+            ->willReturn(1);
+        $this->quoteMock->expects(static::never())
+            ->method('setSharedStoreIds');
+        $this->quoteMock->expects(static::once())
+            ->method('load')
+            ->with($cartId)
+            ->willReturn($this->storeMock);
+        $this->quoteMock->expects(static::once())
+            ->method('loadByCustomer')
+            ->with($customerId)
+            ->willReturn($this->storeMock);
+        $this->quoteMock->expects(static::exactly(3))
+            ->method('getId')
+            ->willReturn($cartId);
+        $this->quoteMock->expects(static::any())
+            ->method('getCustomerId')
+            ->willReturn($customerId);
+        $this->loadHandlerMock->expects(static::exactly(2))
+            ->method('load')
+            ->with($this->quoteMock);
+
+        $this->assertEquals($this->quoteMock, $this->model->get($cartId));
+        $this->assertEquals($this->quoteMock, $this->model->getForCustomer($customerId));
+    }
+
     public function testGetWithSharedStoreIds()
     {
         $this->markTestSkipped('MAGETWO-48531');
@@ -174,19 +237,31 @@ public function testGetWithSharedStoreIds()
 
     public function testGetForCustomer()
     {
-        $this->markTestSkipped('MAGETWO-48531');
         $cartId = 17;
         $customerId = 23;
 
-        $this->quoteFactoryMock->expects($this->once())->method('create')->willReturn($this->quoteMock);
-        $this->storeManagerMock->expects($this->once())->method('getStore')->willReturn($this->storeMock);
-        $this->storeMock->expects($this->once())->method('getId')->willReturn(1);
-        $this->quoteMock->expects($this->never())->method('setSharedStoreIds');
-        $this->quoteMock->expects($this->once())
+        $this->quoteFactoryMock->expects(static::once())
+            ->method('create')
+            ->willReturn($this->quoteMock);
+        $this->storeManagerMock->expects(static::once())
+            ->method('getStore')
+            ->willReturn($this->storeMock);
+        $this->storeMock->expects(static::once())
+            ->method('getId')
+            ->willReturn(1);
+        $this->quoteMock->expects(static::never())
+            ->method('setSharedStoreIds');
+        $this->quoteMock->expects(static::once())
             ->method('loadByCustomer')
             ->with($customerId)
             ->willReturn($this->storeMock);
-        $this->quoteMock->expects($this->exactly(2))->method('getId')->willReturn($cartId);
+        $this->quoteMock->expects(static::exactly(2))
+            ->method('getId')
+            ->willReturn($cartId);
+
+        $this->loadHandlerMock->expects(static::once())
+            ->method('load')
+            ->with($this->quoteMock);
 
         $this->assertEquals($this->quoteMock, $this->model->getForCustomer($customerId));
         $this->assertEquals($this->quoteMock, $this->model->getForCustomer($customerId));