MAGETWO-31933: Can't change shipping address when creating order in backend

 - enhanced unit test

Author: Mike Weis

app/code/Magento/Backend/Test/Unit/Model/Session/QuoteTest.php

@@ -231,9 +231,10 @@ public function testGetQuote()
                 'setCustomerGroupId',
                 'setIsActive',
                 'getId',
-                'assignCustomer',
+                'assignCustomerWithAddressChange',
                 'setIgnoreOldQty',
                 'setIsSuperMode',
+                'getAllAddresses',
                 '__wakeup'
             ],
             [],
@@ -290,14 +291,17 @@ public function testGetQuote()
             ->with($customerId)
             ->willReturn($dataCustomerMock);
         $quoteMock->expects($this->once())
-            ->method('assignCustomer')
+            ->method('assignCustomerWithAddressChange')
             ->with($dataCustomerMock);
         $quoteMock->expects($this->once())
             ->method('setIgnoreOldQty')
             ->with(true);
         $quoteMock->expects($this->once())
             ->method('setIsSuperMode')
             ->with(true);
+        $quoteMock->expects($this->any())
+            ->method('getAllAddresses')
+            ->will($this->returnValue([]));
 
         $this->assertEquals($quoteMock, $this->quote->getQuote());
     }
@@ -320,9 +324,10 @@ public function testGetQuoteGet()
                 'setCustomerGroupId',
                 'setIsActive',
                 'getId',
-                'assignCustomer',
+                'assignCustomerWithAddressChange',
                 'setIgnoreOldQty',
                 'setIsSuperMode',
+                'getAllAddresses',
                 '__wakeup'
             ],
             [],
@@ -360,14 +365,17 @@ public function testGetQuoteGet()
             ->with($customerId)
             ->willReturn($dataCustomerMock);
         $quoteMock->expects($this->once())
-            ->method('assignCustomer')
+            ->method('assignCustomerWithAddressChange')
             ->with($dataCustomerMock);
         $quoteMock->expects($this->once())
             ->method('setIgnoreOldQty')
             ->with(true);
         $quoteMock->expects($this->once())
             ->method('setIsSuperMode')
             ->with(true);
+        $quoteMock->expects($this->any())
+            ->method('getAllAddresses')
+            ->will($this->returnValue([]));
 
         $this->assertEquals($quoteMock, $this->quote->getQuote());
     }