#25147: Totals Information Management - setting quote shipping method when address method is null - unit test update.

Author: engcom-Foxtrot

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

@@ -36,7 +36,7 @@ class TotalsInformationManagementTest extends \PHPUnit\Framework\TestCase
      */
     private $totalsInformationManagement;
 
-    protected function setUp()
+    protected function setUp(): void
     {
         $this->objectManager = new ObjectManager($this);
         $this->cartRepositoryMock = $this->createMock(
@@ -77,13 +77,15 @@ public function testCalculate(?string $carrierCode, ?string $carrierMethod, int
         $addressInformationMock = $this->createMock(
             TotalsInformationInterface::class
         );
-        $addressMock = $this->createPartialMock(
-            Address::class,
-            [
-                'setShippingMethod',
-                'setCollectShippingRates'
-            ]
-        );
+        $addressMock = $this->getMockBuilder(Address::class)
+            ->addMethods(
+                [
+                    'setShippingMethod',
+                    'setCollectShippingRates',
+                ]
+            )
+            ->disableOriginalConstructor()
+            ->getMock();
 
         $addressInformationMock->expects($this->once())->method('getAddress')->willReturn($addressMock);
         $addressInformationMock->expects($this->any())->method('getShippingCarrierCode')->willReturn($carrierCode);