MAGETWO-38151: Data Rollback

- Changes based on CR comments.

Author: safwkhan

lib/internal/Magento/Framework/Setup/Test/Unit/BackupRollbackTest.php

@@ -116,7 +116,7 @@ public function testCodeRollback()
         $this->model->codeRollback('RollbackFile_A.tgz');
     }
 
-    public function testMeidaBackup()
+    public function testMediaBackup()
     {
         $this->setupCodeBackupRollback();
         $this->filesystem->expects($this->once())

setup/src/Magento/Setup/Console/Command/RollbackCommand.php

@@ -146,19 +146,17 @@ protected function execute(InputInterface $input, OutputInterface $output)
     private function doRollback(InputInterface $input, OutputInterface $output)
     {
         $inputOptionProvided = false;
+        $rollbackHandler = $this->backupRollbackFactory->create($output);
         if ($input->getOption(self::INPUT_KEY_CODE_BACKUP_FILE)) {
-            $codeRollback = $this->backupRollbackFactory->create($output);
-            $codeRollback->codeRollback($input->getOption(self::INPUT_KEY_CODE_BACKUP_FILE));
+            $rollbackHandler->codeRollback($input->getOption(self::INPUT_KEY_CODE_BACKUP_FILE));
             $inputOptionProvided = true;
         }
         if ($input->getOption(self::INPUT_KEY_MEDIA_BACKUP_FILE)) {
-            $mediaRollback = $this->backupRollbackFactory->create($output);
-            $mediaRollback->codeRollback($input->getOption(self::INPUT_KEY_MEDIA_BACKUP_FILE), Factory::TYPE_MEDIA);
+            $rollbackHandler->codeRollback($input->getOption(self::INPUT_KEY_MEDIA_BACKUP_FILE), Factory::TYPE_MEDIA);
             $inputOptionProvided = true;
         }
         if ($input->getOption(self::INPUT_KEY_DB_BACKUP_FILE)) {
-            $dbRollback = $this->backupRollbackFactory->create($output);
-            $dbRollback->dbRollback($input->getOption(self::INPUT_KEY_DB_BACKUP_FILE));
+            $rollbackHandler->dbRollback($input->getOption(self::INPUT_KEY_DB_BACKUP_FILE));
             $inputOptionProvided = true;
         }
         if (!$inputOptionProvided) {