Update for PHPUnit test scenario according to new update for merging guest cart and customer cart.

Author: ivanko-dev

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

@@ -7,11 +7,14 @@
 
 namespace Magento\Quote\Test\Unit\Model;
 
+use Magento\Customer\Api\Data\CustomerInterface;
+use Magento\Customer\Model\Customer;
 use Magento\Framework\App\RequestInterface;
 use Magento\Framework\Exception\NoSuchEntityException;
 
 use Magento\Framework\HTTP\PhpEnvironment\RemoteAddress;
 use Magento\Quote\Model\CustomerManagement;
+use Magento\Quote\Model\Quote;
 use Magento\Quote\Model\QuoteIdMaskFactory;
 use Magento\Sales\Api\Data\OrderAddressInterface;
 
@@ -199,7 +202,7 @@ protected function setUp()
         );
 
         $this->quoteMock = $this->createPartialMock(
-            \Magento\Quote\Model\Quote::class,
+            Quote::class,
             [
                 'assignCustomer',
                 'collectTotals',
@@ -275,7 +278,7 @@ public function testCreateEmptyCartAnonymous()
         $storeId = 345;
         $quoteId = 2311;
 
-        $quoteMock = $this->createMock(\Magento\Quote\Model\Quote::class);
+        $quoteMock = $this->createMock(Quote::class);
         $quoteAddress = $this->createPartialMock(
             \Magento\Quote\Model\Quote\Address::class,
             ['setCollectShippingRates']
@@ -306,14 +309,14 @@ public function testCreateEmptyCartForCustomer()
         $quoteId = 2311;
         $userId = 567;
 
-        $quoteMock = $this->createMock(\Magento\Quote\Model\Quote::class);
+        $quoteMock = $this->createMock(Quote::class);
 
         $this->quoteRepositoryMock
             ->expects($this->once())
             ->method('getActiveForCustomer')
             ->with($userId)
             ->willThrowException(new NoSuchEntityException());
-        $customer = $this->getMockBuilder(\Magento\Customer\Api\Data\CustomerInterface::class)
+        $customer = $this->getMockBuilder(CustomerInterface::class)
             ->setMethods(['getDefaultBilling'])->disableOriginalConstructor()->getMockForAbstractClass();
         $quoteAddress = $this->createPartialMock(
             \Magento\Quote\Model\Quote\Address::class,
@@ -342,14 +345,14 @@ public function testCreateEmptyCartForCustomerReturnExistsQuote()
         $storeId = 345;
         $userId = 567;
 
-        $quoteMock = $this->createMock(\Magento\Quote\Model\Quote::class);
+        $quoteMock = $this->createMock(Quote::class);
 
         $this->quoteRepositoryMock
             ->expects($this->once())
             ->method('getActiveForCustomer')
             ->with($userId)->willReturn($quoteMock);
 
-        $customer = $this->getMockBuilder(\Magento\Customer\Api\Data\CustomerInterface::class)
+        $customer = $this->getMockBuilder(CustomerInterface::class)
             ->setMethods(['getDefaultBilling'])->disableOriginalConstructor()->getMockForAbstractClass();
         $quoteAddress = $this->createPartialMock(
             \Magento\Quote\Model\Quote\Address::class,
@@ -379,8 +382,8 @@ public function testAssignCustomerFromAnotherStore()
         $customerId = 455;
         $storeId = 5;
 
-        $quoteMock = $this->createMock(\Magento\Quote\Model\Quote::class);
-        $customerMock = $this->createMock(\Magento\Customer\Api\Data\CustomerInterface::class);
+        $quoteMock = $this->createMock(Quote::class);
+        $customerMock = $this->createMock(CustomerInterface::class);
 
         $this->quoteRepositoryMock
             ->expects($this->once())
@@ -395,7 +398,7 @@ public function testAssignCustomerFromAnotherStore()
             ->willReturn($customerMock);
 
         $customerModelMock = $this->createPartialMock(
-            \Magento\Customer\Model\Customer::class,
+            Customer::class,
             ['load', 'getSharedStoreIds']
         );
         $this->customerFactoryMock->expects($this->once())->method('create')->willReturn($customerModelMock);
@@ -424,10 +427,10 @@ public function testAssignCustomerToNonanonymousCart()
         $storeId = 5;
 
         $quoteMock = $this->createPartialMock(
-            \Magento\Quote\Model\Quote::class,
+            Quote::class,
             ['getCustomerId', 'setCustomer', 'setCustomerIsGuest']
         );
-        $customerMock = $this->createMock(\Magento\Customer\Api\Data\CustomerInterface::class);
+        $customerMock = $this->createMock(CustomerInterface::class);
 
         $this->quoteRepositoryMock
             ->expects($this->once())
@@ -442,7 +445,7 @@ public function testAssignCustomerToNonanonymousCart()
             ->willReturn($customerMock);
 
         $customerModelMock = $this->createPartialMock(
-            \Magento\Customer\Model\Customer::class,
+            Customer::class,
             ['load', 'getSharedStoreIds']
         );
         $this->customerFactoryMock->expects($this->once())->method('create')->willReturn($customerModelMock);
@@ -463,7 +466,7 @@ public function testAssignCustomerToNonanonymousCart()
     }
 
     /**
-     * @expectedException \Magento\Framework\Exception\StateException
+     * @expectedException NoSuchEntityException
      */
     public function testAssignCustomerNoSuchCustomer()
     {
@@ -472,10 +475,9 @@ public function testAssignCustomerNoSuchCustomer()
         $storeId = 5;
 
         $quoteMock = $this->createPartialMock(
-            \Magento\Quote\Model\Quote::class,
+            Quote::class,
             ['getCustomerId', 'setCustomer', 'setCustomerIsGuest']
         );
-        $customerMock = $this->createMock(\Magento\Customer\Api\Data\CustomerInterface::class);
 
         $this->quoteRepositoryMock
             ->expects($this->once())
@@ -487,36 +489,14 @@ public function testAssignCustomerNoSuchCustomer()
             ->expects($this->once())
             ->method('getById')
             ->with($customerId)
-            ->willReturn($customerMock);
+            ->willThrowException(new NoSuchEntityException());
 
-        $customerModelMock = $this->createPartialMock(
-            \Magento\Customer\Model\Customer::class,
-            ['load', 'getSharedStoreIds']
+        $this->expectExceptionMessage(
+            "No such entity."
         );
-        $this->customerFactoryMock->expects($this->once())->method('create')->willReturn($customerModelMock);
-        $customerModelMock
-            ->expects($this->once())
-            ->method('load')
-            ->with($customerId)
-            ->willReturnSelf();
-
-        $customerModelMock
-            ->expects($this->once())
-            ->method('getSharedStoreIds')
-            ->willReturn([$storeId, 'some store value']);
-
-        $quoteMock->expects($this->once())->method('getCustomerId')->willReturn(null);
-
-        $this->quoteRepositoryMock
-            ->expects($this->once())
-            ->method('getForCustomer')
-            ->with($customerId);
 
         $this->model->assignCustomer($cartId, $customerId, $storeId);
 
-        $this->expectExceptionMessage(
-            "The customer can't be assigned to the cart because the customer already has an active cart."
-        );
     }
 
     public function testAssignCustomer()
@@ -525,18 +505,11 @@ public function testAssignCustomer()
         $customerId = 455;
         $storeId = 5;
 
-        $this->getPropertyValue($this->model, 'quoteIdMaskFactory')
-            ->expects($this->once())
-            ->method('create')
-            ->willReturn($this->quoteIdMock);
-        $this->quoteIdMock->expects($this->once())->method('load')->with($cartId, 'quote_id')->willReturnSelf();
-        $this->quoteIdMock->expects($this->once())->method('getId')->willReturn(10);
-        $this->quoteIdMock->expects($this->once())->method('delete');
         $quoteMock = $this->createPartialMock(
-            \Magento\Quote\Model\Quote::class,
-            ['getCustomerId', 'setCustomer', 'setCustomerIsGuest']
+            Quote::class,
+            ['getCustomerId', 'setCustomer', 'setCustomerIsGuest', 'merge']
         );
-        $customerMock = $this->createMock(\Magento\Customer\Api\Data\CustomerInterface::class);
+        $customerMock = $this->createMock(CustomerInterface::class);
 
         $this->quoteRepositoryMock
             ->expects($this->once())
@@ -551,7 +524,7 @@ public function testAssignCustomer()
             ->willReturn($customerMock);
 
         $customerModelMock = $this->createPartialMock(
-            \Magento\Customer\Model\Customer::class,
+            Customer::class,
             ['load', 'getSharedStoreIds']
         );
         $this->customerFactoryMock->expects($this->once())->method('create')->willReturn($customerModelMock);
@@ -574,10 +547,22 @@ public function testAssignCustomer()
             ->with($customerId)
             ->willThrowException(new NoSuchEntityException());
 
-        $quoteMock->expects($this->once())->method('setCustomer')->with($customerMock);
-        $quoteMock->expects($this->once())->method('setCustomerIsGuest')->with(0);
+        $activeQuoteMock = $this->createPartialMock(
+            Quote::class,
+            ['getCustomerId', 'setCustomer', 'setCustomerIsGuest', 'setStoreId', 'setIsActive', 'getIsActive', 'merge']
+        );
 
-        $this->quoteRepositoryMock->expects($this->once())->method('save')->with($quoteMock);
+        $this->quoteFactoryMock->expects($this->once())->method('create')->willReturn($activeQuoteMock);
+        $activeQuoteMock->expects($this->once())->method('setCustomer')->with($customerMock);
+        $activeQuoteMock->expects($this->once())->method('setCustomerIsGuest')->with(0);
+        $activeQuoteMock->expects($this->once())->method('setStoreId')->with($storeId);
+        $activeQuoteMock->expects($this->once())->method('setIsActive')->with(1);
+
+        $activeQuoteMock->expects($this->once())->method('getIsActive')->willReturn(1);
+        $activeQuoteMock->expects($this->once())->method('merge')->with($quoteMock)->willReturnSelf();
+        $this->quoteRepositoryMock->expects($this->once())->method('delete')->with($quoteMock);
+
+        $this->quoteRepositoryMock->expects($this->once())->method('save')->with($activeQuoteMock);
 
         $this->model->assignCustomer($cartId, $customerId, $storeId);
     }
@@ -881,7 +866,7 @@ protected function getQuote(
         \Magento\Quote\Model\Quote\Address $shippingAddress = null
     ) {
         $quote = $this->createPartialMock(
-            \Magento\Quote\Model\Quote::class,
+            Quote::class,
             [
                 'setIsActive',
                 'getCustomerEmail',
@@ -928,7 +913,7 @@ protected function getQuote(
             ->willReturn($payment);
 
         $customer = $this->createPartialMock(
-            \Magento\Customer\Model\Customer::class,
+            Customer::class,
             ['getDefaultBilling', 'getId']
         );
         $quote->expects($this->any())->method('getCustomerId')->willReturn($customerId);
@@ -1021,7 +1006,7 @@ protected function prepareOrderFactory(
     public function testGetCartForCustomer()
     {
         $customerId = 100;
-        $cartMock = $this->createMock(\Magento\Quote\Model\Quote::class);
+        $cartMock = $this->createMock(Quote::class);
         $this->quoteRepositoryMock->expects($this->once())
             ->method('getActiveForCustomer')
             ->with($customerId)