Skip to content

Commit 0d2ac7c

Browse files
zakdmazakdma
committed
MC-24548: collect-totals API call not updating the selected shippingMethod
1 parent 5dcb5a5 commit 0d2ac7c

File tree

3 files changed

+180
-1
lines changed

3 files changed

+180
-1
lines changed

‎app/code/Magento/Quote/Model/ShippingMethodManagement.php

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,15 @@ public function apply($cartId, $carrierCode, $methodCode)
224224
$this->quoteAddressResource->delete($shippingAddress);
225225
throw new StateException(__('The shipping address is missing. Set the address and try again.'));
226226
}
227-
$shippingAddress->setShippingMethod($carrierCode . '_' . $methodCode);
227+
$shippingMethod = $carrierCode . '_' . $methodCode;
228+
$shippingAddress->setShippingMethod($shippingMethod);
229+
$shippingAssignments = $quote->getExtensionAttributes()->getShippingAssignments();
230+
if (!empty($shippingAssignments)) {
231+
$shippingAssignment = $shippingAssignments[0];
232+
$shipping = $shippingAssignment->getShipping();
233+
$shipping->setMethod($shippingMethod);
234+
$shippingAssignment->setShipping($shipping);
235+
}
228236
}
229237

230238
/**

‎app/code/Magento/Quote/Test/Unit/Model/ShippingMethodManagementTest.php

Lines changed: 69 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,9 @@
2727
use Magento\Store\Model\Store;
2828
use PHPUnit\Framework\MockObject\MockObject;
2929
use PHPUnit\Framework\TestCase;
30+
use Magento\Quote\Api\Data\CartExtensionInterface;
31+
use Magento\Sales\Model\Order\ShippingAssignmentBuilder;
32+
use Magento\Quote\Api\Data\ShippingInterface;
3033

3134
/**
3235
* @SuppressWarnings(PHPMD.CouplingBetweenObjects)
@@ -103,6 +106,21 @@ class ShippingMethodManagementTest extends TestCase
103106
*/
104107
private $quoteAddressResource;
105108

109+
/**
110+
* @var CartExtensionInterface|MockObject
111+
*/
112+
private $extensionAttributesMock;
113+
114+
/**
115+
* @var ShippingInterface|MockObject
116+
*/
117+
private $shippingMock;
118+
119+
/**
120+
* @var ShippingAssignmentBuilder|MockObject
121+
*/
122+
private $shippingAssignmentBuilder;
123+
106124
protected function setUp(): void
107125
{
108126
$this->objectManager = new ObjectManager($this);
@@ -135,6 +153,7 @@ protected function setUp(): void
135153
'collectTotals',
136154
'save',
137155
'__wakeup',
156+
'getExtensionAttributes'
138157
])
139158
->getMock();
140159

@@ -190,6 +209,20 @@ protected function setUp(): void
190209
'dataProcessor',
191210
$this->dataProcessor
192211
);
212+
213+
$this->extensionAttributesMock = $this->getMockBuilder(CartExtensionInterface::class)
214+
->setMethods(['getShippingAssignments'])
215+
->getMockForAbstractClass();
216+
217+
$this->shippingMock = $this->getMockForAbstractClass(ShippingInterface::class);
218+
219+
$this->shippingAssignmentBuilder = $this->getMockBuilder(ShippingAssignmentBuilder::class)
220+
->disableOriginalConstructor()
221+
->setMethods([
222+
'getShipping',
223+
'setShipping'
224+
])
225+
->getMock();
193226
}
194227

195228
public function testGetMethodWhenShippingAddressIsNotSet()
@@ -424,6 +457,24 @@ public function testSetMethodWithCouldNotSaveException()
424457
$this->shippingAddress->expects($this->once())
425458
->method('setShippingMethod')
426459
->with($carrierCode . '_' . $methodCode);
460+
$this->quote->expects($this->once())
461+
->method('getExtensionAttributes')
462+
->willReturn($this->extensionAttributesMock);
463+
464+
$this->extensionAttributesMock->expects($this->once())->method('getShippingAssignments')
465+
->willReturn([$this->shippingAssignmentBuilder]);
466+
467+
$this->shippingAssignmentBuilder->expects($this->once())->method('getShipping')
468+
->willReturn($this->shippingMock);
469+
470+
$this->shippingMock->expects($this->once())
471+
->method('setMethod')
472+
->with($carrierCode . '_' . $methodCode);
473+
474+
$this->shippingAssignmentBuilder->expects($this->once())
475+
->method('setShipping')
476+
->with($this->shippingMock);
477+
427478
$exception = new \Exception('Custom Error');
428479
$this->quote->expects($this->once())->method('collectTotals')->willReturnSelf();
429480
$this->quoteRepository->expects($this->once())
@@ -477,6 +528,24 @@ public function testSetMethod()
477528
->method('getCountryId')->willReturn($countryId);
478529
$this->shippingAddress->expects($this->once())
479530
->method('setShippingMethod')->with($carrierCode . '_' . $methodCode);
531+
$this->quote->expects($this->once())
532+
->method('getExtensionAttributes')
533+
->willReturn($this->extensionAttributesMock);
534+
535+
$this->extensionAttributesMock->expects($this->once())->method('getShippingAssignments')
536+
->willReturn([$this->shippingAssignmentBuilder]);
537+
538+
$this->shippingAssignmentBuilder->expects($this->once())->method('getShipping')
539+
->willReturn($this->shippingMock);
540+
541+
$this->shippingMock->expects($this->once())
542+
->method('setMethod')
543+
->with($carrierCode . '_' . $methodCode);
544+
545+
$this->shippingAssignmentBuilder->expects($this->once())
546+
->method('setShipping')
547+
->with($this->shippingMock);
548+
480549
$this->quote->expects($this->once())->method('collectTotals')->willReturnSelf();
481550
$this->quoteRepository->expects($this->once())->method('save')->with($this->quote);
482551

‎dev/tests/api-functional/testsuite/Magento/Quote/Api/GuestCollectRatesTest.php

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,102 @@
1+
<?php
2+
/**
3+
* Copyright © Magento, Inc. All rights reserved.
4+
* See COPYING.txt for license details.
5+
*/
6+
namespace Magento\Quote\Api;
7+
8+
use Magento\Framework\Webapi\Rest\Request;
9+
use Magento\Quote\Model\Quote;
10+
use Magento\Quote\Model\QuoteIdMaskFactory;
11+
use Magento\TestFramework\Helper\Bootstrap;
12+
use Magento\TestFramework\ObjectManager;
13+
use Magento\TestFramework\TestCase\WebapiAbstract;
14+
use Magento\Quote\Model\QuoteIdMask;
15+
16+
/**
17+
* Class GuestCollectRatesTest checks that totals will be recollected properly with new shipping method
18+
*/
19+
class GuestCollectRatesTest extends WebapiAbstract
20+
{
21+
const SERVICE_VERSION = 'V1';
22+
const SERVICE_NAME = 'quoteGuestCartTotalManagementV1';
23+
const RESOURCE_PATH = '/V1/guest-carts/';
24+
25+
/**
26+
* @var ObjectManager
27+
*/
28+
private $objectManager;
29+
30+
/**
31+
* @var Quote
32+
*/
33+
protected $quote;
34+
35+
protected function setUp(): void
36+
{
37+
$this->objectManager = Bootstrap::getObjectManager();
38+
$this->quote = $this->objectManager->create(Quote::class);
39+
}
40+
41+
/**
42+
* Checks that totals are properly recollected after changing shipping method
43+
*
44+
* @magentoApiDataFixture Magento/Checkout/_files/quote_with_shipping_method.php
45+
*/
46+
public function testCollectRatesWithChangedShippingMethod()
47+
{
48+
$this->quote->load('test_order_1', 'reserved_order_id');
49+
$cartId = $this->quote->getId();
50+
51+
/** @var QuoteIdMask $quoteIdMask */
52+
$quoteIdMask = $this->objectManager
53+
->create(QuoteIdMaskFactory::class)
54+
->create();
55+
$quoteIdMask->load($cartId, 'quote_id');
56+
57+
$cartId = $quoteIdMask->getMaskedId();
58+
$requestData = [
59+
"shippingMethodCode" => "freeshipping",
60+
"shippingCarrierCode" => "freeshipping",
61+
"paymentMethod" => [
62+
"method" => "checkmo",
63+
],
64+
];
65+
66+
if (TESTS_WEB_API_ADAPTER === self::ADAPTER_SOAP) {
67+
$requestData['cartId'] = $cartId;
68+
}
69+
70+
$totals = $this->_webApiCall($this->getListServiceInfo($cartId), $requestData);
71+
$this->assertEquals(20, $totals['grand_total']);
72+
$this->assertEquals(0, $totals['shipping_amount']);
73+
74+
$requestData['shippingMethodCode'] = 'flatrate';
75+
$requestData['shippingCarrierCode'] = 'flatrate';
76+
77+
$totals = $this->_webApiCall($this->getListServiceInfo($cartId), $requestData);
78+
$this->assertEquals(30, $totals['grand_total']);
79+
$this->assertEquals(10, $totals['shipping_amount']);
80+
}
81+
82+
/**
83+
* Service info
84+
*
85+
* @param int $cartId
86+
* @return array
87+
*/
88+
protected function getListServiceInfo($cartId)
89+
{
90+
return [
91+
'rest' => [
92+
'resourcePath' => self::RESOURCE_PATH . $cartId . '/collect-totals',
93+
'httpMethod' => Request::HTTP_METHOD_PUT,
94+
],
95+
'soap' => [
96+
'service' => self::SERVICE_NAME,
97+
'serviceVersion' => self::SERVICE_VERSION,
98+
'operation' => 'quoteGuestCartTotalManagementV1CollectTotals',
99+
],
100+
];
101+
}
102+
}

0 commit comments

Comments
 (0)