AC-2044::Cart Price Rule with shipping method set as condition is not working Magento - fixed unit tests against changes

Author: mahesh-singh-rajawat

app/code/Magento/Checkout/Test/Unit/Model/TotalsInformationManagementTest.php

@@ -85,6 +85,7 @@ public function testCalculate(?string $carrierCode, ?string $carrierMethod, int
                     'setCollectShippingRates',
                 ]
             )
+            ->onlyMethods(['save'])
             ->disableOriginalConstructor()
             ->getMock();
 
@@ -97,6 +98,9 @@ public function testCalculate(?string $carrierCode, ?string $carrierMethod, int
             ->method('setCollectShippingRates')->with(true)->willReturn($addressMock);
         $addressMock->expects($this->exactly($methodSetCount))
             ->method('setShippingMethod')->with($carrierCode . '_' . $carrierMethod);
+        $addressMock->expects($this->exactly($methodSetCount))
+            ->method('save')
+            ->willReturnSelf();
         $cartMock->expects($this->once())->method('collectTotals');
 
         $this->totalsInformationManagement->calculate($cartId, $addressInformationMock);