MAGETWO-37571: [TechDebt] Covered php code after new checkout implementation by unit tests

Author: okarpenko

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

@@ -44,6 +44,9 @@ class AddressDetailsManagementTest extends \PHPUnit_Framework_TestCase
      */
     protected $dataProcessor;
 
+    /** @var \Magento\Quote\Model\QuoteRepository|\PHPUnit_Framework_MockObject_MockObject */
+    protected $quoteRepository;
+
     /**
      * @var \Magento\Framework\TestFramework\Unit\Helper\ObjectManager
      */
@@ -58,6 +61,7 @@ protected function setUp()
         $this->shippingMethodManagement = $this->getMock('\Magento\Quote\Api\ShippingMethodManagementInterface');
         $this->addressDetailsFactory = $this->getMock('\Magento\Quote\Model\AddressDetailsFactory', [], [], '', false);
         $this->dataProcessor = $this->getMock('\Magento\Quote\Model\AddressAdditionalDataProcessor', [], [], '', false);
+        $this->quoteRepository = $this->getMock('Magento\Quote\Model\QuoteRepository', [], [], '', false);
 
         $this->model = $this->objectManager->getObject(
             'Magento\Quote\Model\AddressDetailsManagement',
@@ -67,7 +71,8 @@ protected function setUp()
                 'paymentMethodManagement' => $this->paymentMethodManagement,
                 'shippingMethodManagement' => $this->shippingMethodManagement,
                 'addressDetailsFactory' => $this->addressDetailsFactory,
-                'dataProcessor' => $this->dataProcessor
+                'dataProcessor' => $this->dataProcessor,
+                'quoteRepository' =>  $this->quoteRepository,
             ]
         );
     }
@@ -125,6 +130,16 @@ public function testSaveAddresses()
             ->willReturnSelf();
         $this->dataProcessor->expects($this->once())->method('process')->with($additionalData);
 
-        $this->model->saveAddresses($cartId, $billingAddressMock, $shippingAddressMock, $additionalData);
+        $quote = $this->getMock('Magento\Quote\Model\Quote', [], [], '', false);
+        $quote->expects($this->once())
+            ->method('setCheckoutMethod')
+            ->willReturnSelf();
+
+        $this->quoteRepository
+            ->expects($this->once())
+            ->method('getActive')
+            ->willReturn($quote);
+
+        $this->model->saveAddresses($cartId, $billingAddressMock, $shippingAddressMock, $additionalData, 'register');
     }
 }

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

@@ -200,7 +200,8 @@ protected function setUp()
                 'getBillingAddress',
                 'setCustomerIsGuest',
                 'setCustomerGroupId',
-                'assignCustomer'
+                'assignCustomer',
+                'getPayment',
             ],
             [],
             '',
@@ -769,6 +770,15 @@ public function testPlaceOrder()
             ->with($cartId)
             ->willReturn($this->quoteMock);
 
+        $quotePayment = $this->getMock('Magento\Quote\Model\Quote\Payment', [], [], '', false);
+        $quotePayment->expects($this->once())
+            ->method('setQuote');
+        $quotePayment->expects($this->once())
+            ->method('importData');
+        $this->quoteMock->expects($this->atLeastOnce())
+            ->method('getPayment')
+            ->willReturn($quotePayment);
+
         $this->quoteMock->expects($this->once())
             ->method('getCheckoutMethod')
             ->willReturn(\Magento\Checkout\Model\Type\Onepage::METHOD_CUSTOMER);
@@ -787,7 +797,12 @@ public function testPlaceOrder()
         $this->checkoutSessionMock->expects($this->once())->method('setLastOrderId')->with($orderId);
         $this->checkoutSessionMock->expects($this->once())->method('setLastRealOrderId')->with($orderIncrementId);
         $this->agreementsValidatorMock->expects($this->once())->method('isValid')->willReturn(true);
-        $this->assertEquals($orderId, $service->placeOrder($cartId));
+
+        $paymentMethod = $this->getMock('Magento\Quote\Model\Quote\Payment', ['setChecks', 'getData'], [], '', false);
+        $paymentMethod->expects($this->once())->method('setChecks');
+        $paymentMethod->expects($this->once())->method('getData')->willReturn(['additional_data' => []]);
+
+        $this->assertEquals($orderId, $service->placeOrder($cartId, null, $paymentMethod));
     }
 
     /**