Updated tests to include the MaintenanceModeEnabler

Author: jokeputs

app/code/Magento/Theme/Test/Unit/Console/Command/ThemeUninstallCommandTest.php

@@ -304,9 +304,9 @@ public function testExecute()
     {
         $this->setUpExecute();
         $this->cleanupFiles->expects($this->never())->method('clearMaterializedViewFiles');
+        $this->maintenanceMode->expects($this->once())->method('enableMaintenanceMode');
+        $this->maintenanceMode->expects($this->once())->method('disableMaintenanceMode');
         $this->tester->execute(['theme' => ['area/vendor/test']]);
-        $this->assertContains('Enabling maintenance mode', $this->tester->getDisplay());
-        $this->assertContains('Disabling maintenance mode', $this->tester->getDisplay());
         $this->assertContains('Alert: Generated static view files were not cleared.', $this->tester->getDisplay());
         $this->assertNotContains('Generated static view files cleared successfully', $this->tester->getDisplay());
     }
@@ -315,9 +315,9 @@ public function testExecuteCleanStaticFiles()
     {
         $this->setUpExecute();
         $this->cleanupFiles->expects($this->once())->method('clearMaterializedViewFiles');
+        $this->maintenanceMode->expects($this->once())->method('enableMaintenanceMode');
+        $this->maintenanceMode->expects($this->once())->method('disableMaintenanceMode');
         $this->tester->execute(['theme' => ['area/vendor/test'], '-c' => true]);
-        $this->assertContains('Enabling maintenance mode', $this->tester->getDisplay());
-        $this->assertContains('Disabling maintenance mode', $this->tester->getDisplay());
         $this->assertNotContains('Alert: Generated static view files were not cleared.', $this->tester->getDisplay());
         $this->assertContains('Generated static view files cleared successfully', $this->tester->getDisplay());
     }

setup/src/Magento/Setup/Test/Unit/Console/Command/BackupCommandTest.php

@@ -35,9 +35,14 @@ class BackupCommandTest extends \PHPUnit\Framework\TestCase
      */
     private $deploymentConfig;
 
+    /**
+     * @var \Magento\Framework\App\Console\MaintenanceModeEnabler|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $maintenanceMode;
+
     public function setUp()
     {
-        $maintenanceMode = $this->createMock(\Magento\Framework\App\Console\MaintenanceModeEnabler::class);
+        $this->maintenanceMode = $this->createMock(\Magento\Framework\App\Console\MaintenanceModeEnabler::class);
         $objectManagerProvider = $this->createMock(\Magento\Setup\Model\ObjectManagerProvider::class);
         $this->objectManager = $this->getMockForAbstractClass(
             \Magento\Framework\ObjectManagerInterface::class,
@@ -72,7 +77,7 @@ public function setUp()
             );
         $command = new BackupCommand(
             $objectManagerProvider,
-            $maintenanceMode,
+            $this->maintenanceMode,
             $this->deploymentConfig
         );
         $this->tester = new CommandTester($command);
@@ -128,10 +133,10 @@ public function testExecuteNoOptions()
         $this->deploymentConfig->expects($this->once())
             ->method('isAvailable')
             ->will($this->returnValue(false));
+        $this->maintenanceMode->expects($this->once())->method('enableMaintenanceMode');
+        $this->maintenanceMode->expects($this->once())->method('disableMaintenanceMode');
         $this->tester->execute([]);
-        $expected = 'Enabling maintenance mode' . PHP_EOL
-            . 'Not enough information provided to take backup.' . PHP_EOL
-            . 'Disabling maintenance mode' . PHP_EOL;
+        $expected = 'Not enough information provided to take backup.' . PHP_EOL;
         $this->assertSame($expected, $this->tester->getDisplay());
     }
 }

setup/src/Magento/Setup/Test/Unit/Console/Command/RollbackCommandTest.php

@@ -50,10 +50,15 @@ class RollbackCommandTest extends \PHPUnit\Framework\TestCase
      */
     private $command;
 
+    /**
+     * @var \Magento\Framework\App\Console\MaintenanceModeEnabler|\PHPUnit_Framework_MockObject_MockObject
+     */
+    private $maintenanceMode;
+
     public function setUp()
     {
         $this->deploymentConfig = $this->createMock(\Magento\Framework\App\DeploymentConfig::class);
-        $maintenanceMode = $this->createMock(\Magento\Framework\App\Console\MaintenanceModeEnabler::class);
+        $this->maintenanceMode = $this->createMock(\Magento\Framework\App\Console\MaintenanceModeEnabler::class);
         $this->objectManager = $this->getMockForAbstractClass(
             \Magento\Framework\ObjectManagerInterface::class,
             [],
@@ -95,7 +100,7 @@ public function setUp()
             ->will($this->returnValue($this->question));
         $this->command = new RollbackCommand(
             $objectManagerProvider,
-            $maintenanceMode,
+            $this->maintenanceMode,
             $this->deploymentConfig
         );
         $this->command->setHelperSet($this->helperSet);
@@ -152,10 +157,10 @@ public function testExecuteNoOptions()
         $this->deploymentConfig->expects($this->once())
             ->method('isAvailable')
             ->will($this->returnValue(true));
+        $this->maintenanceMode->expects($this->once())->method('enableMaintenanceMode');
+        $this->maintenanceMode->expects($this->once())->method('disableMaintenanceMode');
         $this->tester->execute([]);
-        $expected = 'Enabling maintenance mode' . PHP_EOL
-            . 'Not enough information provided to roll back.' . PHP_EOL
-            . 'Disabling maintenance mode' . PHP_EOL;
+        $expected = 'Not enough information provided to roll back.' . PHP_EOL;
         $this->assertSame($expected, $this->tester->getDisplay());
     }