MAGETWO-90496: Countries from default website set when edit order address

 - Updated unit test

Author: YevSent

app/code/Magento/Sales/Test/Unit/Block/Adminhtml/Order/Create/FormTest.php

@@ -23,6 +23,7 @@
 use Magento\Sales\Block\Adminhtml\Order\Create\Form;
 use Magento\Sales\Model\AdminOrder\Create;
 use Magento\Store\Model\Store;
+use Magento\Store\Model\StoreManagerInterface;
 use PHPUnit\Framework\TestCase;
 use PHPUnit_Framework_MockObject_MockObject as MockObject;
 
@@ -51,6 +52,11 @@ class FormTest extends TestCase
      */
     private $block;
 
+    /**
+     * @var StoreManagerInterface|MockObject
+     */
+    private $storeManager;
+
     /**
      * @inheritdoc
      */
@@ -60,6 +66,9 @@ protected function setUp()
         $context = $this->getMockBuilder(Context::class)
             ->disableOriginalConstructor()
             ->getMock();
+        $this->storeManager = $this->getMockForAbstractClass(StoreManagerInterface::class);
+        $context->method('getStoreManager')
+            ->willReturn($this->storeManager);
         $this->quoteSession = $this->getMockBuilder(QuoteSession::class)
             ->disableOriginalConstructor()
             ->setMethods(['getCustomerId', 'getQuoteId', 'getStoreId', 'getStore', 'getQuote'])
@@ -119,6 +128,8 @@ public function testGetOrderDataJson()
             'quote_id' => $quoteId
         ];
 
+        $this->storeManager->method('setCurrentStore')
+            ->with($storeId);
         $this->quoteSession->method('getCustomerId')
             ->willReturn($customerId);
         $this->quoteSession->method('getStoreId')