MC-36838: Cannot checkout with automatic customer group assignment

Author: cpartica

app/code/Magento/Quote/Test/Unit/Observer/Frontend/Quote/Address/CollectTotalsObserverTest.php

@@ -269,7 +269,7 @@ public function testDispatchWithDefaultCustomerGroupId()
         $this->quoteMock->expects($this->once())
             ->method('getCustomerGroupId')
             ->willReturn('customerGroupId');
-        $this->customerMock->expects($this->once())->method('getId')->willReturn('1');
+        $this->customerMock->expects($this->exactly(2))->method('getId')->willReturn('1');
         $this->groupManagementMock->expects($this->once())
             ->method('getDefaultGroup')
             ->willReturn($this->groupInterfaceMock);
@@ -329,6 +329,7 @@ public function testDispatchWithCustomerCountryInEU()
             ->method('setPrevQuoteCustomerGroupId')
             ->with('customerGroupId');
 
+        $this->customerMock->expects($this->once())->method('getId')->willReturn('1');
         $this->quoteMock->expects($this->once())->method('setCustomerGroupId')->with('customerGroupId');
         $this->quoteMock->expects($this->once())->method('setCustomer')->with($this->customerMock);
         $this->customerDataFactoryMock->expects($this->any())
@@ -436,6 +437,8 @@ public function testDispatchWithEmptyShippingAddress()
             ->method('setPrevQuoteCustomerGroupId')
             ->with('customerGroupId');
 
+        $this->customerMock->expects($this->once())->method('getId')->willReturn('1');
+
         $this->quoteMock->expects($this->once())->method('setCustomerGroupId')->with('customerGroupId');
         $this->quoteMock->expects($this->once())->method('setCustomer')->with($this->customerMock);
         $this->customerDataFactoryMock->expects($this->any())