MC-37484: Cart query error when trying to switch store view

Author: cspruiell

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

@@ -136,7 +136,7 @@ public function testSetVirtualProduct()
 
         $quoteMock = $this->getMockBuilder(Quote::class)
             ->addMethods(['setTotalsCollectedFlag'])
-            ->onlyMethods(['getPayment', 'isVirtual', 'getBillingAddress', 'collectTotals', 'save'])
+            ->onlyMethods(['getPayment', 'isVirtual', 'getBillingAddress', 'collectTotals'])
             ->disableOriginalConstructor()
             ->getMock();
         $this->quoteRepositoryMock->expects($this->once())->method('get')->with($cartId)->willReturn($quoteMock);
@@ -189,7 +189,7 @@ public function testSetVirtualProduct()
             ->willReturn(true);
 
         $quoteMock->expects($this->once())->method('setTotalsCollectedFlag')->with(false)->willReturnSelf();
-        $quoteMock->expects($this->once())->method('save')->willReturnSelf();
+        $this->quoteRepositoryMock->expects($this->once())->method('save')->with($quoteMock);
 
         $paymentMock->expects($this->once())->method('getId')->willReturn($paymentId);
         $this->assertEquals($paymentId, $this->model->set($cartId, $methodMock));
@@ -267,7 +267,7 @@ public function testSetSimpleProduct()
 
         $quoteMock = $this->getMockBuilder(Quote::class)
             ->addMethods(['setTotalsCollectedFlag'])
-            ->onlyMethods(['getPayment', 'isVirtual', 'getShippingAddress', 'collectTotals', 'save'])
+            ->onlyMethods(['getPayment', 'isVirtual', 'getShippingAddress', 'collectTotals'])
             ->disableOriginalConstructor()
             ->getMock();
         $this->quoteRepositoryMock->expects($this->once())->method('get')->with($cartId)->willReturn($quoteMock);
@@ -324,7 +324,7 @@ public function testSetSimpleProduct()
             ->willReturn(true);
 
         $quoteMock->expects($this->once())->method('setTotalsCollectedFlag')->with(false)->willReturnSelf();
-        $quoteMock->expects($this->once())->method('save')->willReturnSelf();
+        $this->quoteRepositoryMock->expects($this->once())->method('save')->with($quoteMock);
 
         $paymentMock->expects($this->once())->method('getId')->willReturn($paymentId);
         $this->assertEquals($paymentId, $this->model->set($cartId, $methodMock));