Merge branch 'MAGETWO-59564' into 2.0-develop-pr1

Author: lestare

setup/src/Magento/Setup/Model/ConfigGenerator.php

@@ -237,7 +237,10 @@ public function createXFrameConfig()
     public function createModeConfig()
     {
         $configData = new ConfigData(ConfigFilePool::APP_ENV);
-        $configData->set(State::PARAM_MODE, State::MODE_DEFAULT);
+        if ($this->deploymentConfig->get(State::PARAM_MODE) === null) {
+            $configData->set(State::PARAM_MODE, State::MODE_DEFAULT);
+        }
+
         return $configData;
     }
 

setup/src/Magento/Setup/Test/Unit/Model/ConfigGeneratorTest.php

@@ -8,6 +8,7 @@
 
 
 use Magento\Framework\Config\ConfigOptionsListConstants;
+use Magento\Framework\App\State;
 
 class ConfigGeneratorTest extends \PHPUnit_Framework_TestCase
 {
@@ -58,4 +59,24 @@ public function testCreateCacheHostsConfig()
         $configData = $this->model->createCacheHostsConfig($data);
         $this->assertEquals($expectedData, $configData->getData()[ConfigOptionsListConstants::CONFIG_PATH_CACHE_HOSTS]);
     }
+
+    public function testCreateModeConfig()
+    {
+        $this->deploymentConfigMock->expects($this->once())
+            ->method('get')
+            ->with(State::PARAM_MODE)
+            ->willReturn(null);
+        $configData = $this->model->createModeConfig();
+        $this->assertSame(State::MODE_DEFAULT, $configData->getData()[State::PARAM_MODE]);
+    }
+
+    public function testCreateModeConfigIfAlreadySet()
+    {
+        $this->deploymentConfigMock->expects($this->once())
+            ->method('get')
+            ->with(State::PARAM_MODE)
+            ->willReturn(State::MODE_PRODUCTION);
+        $configData = $this->model->createModeConfig();
+        $this->assertSame([], $configData->getData());
+    }
 }