BugFix: Have rollback determine correct file system (#218)

Author: balmasea

pyntc/devices/ios_device.py

@@ -816,8 +816,9 @@ def redundancy_state(self):
         return processor_redundancy_state
 
     def rollback(self, rollback_to):
+
         try:
-            self.show("configure replace flash:%s force" % rollback_to)
+            self.show("configure replace %s%s force" % (self._get_file_system(), rollback_to))
         except CommandError:
             raise RollbackError("Rollback unsuccessful. %s may not exist." % rollback_to)
 

pyproject.toml

@@ -48,5 +48,6 @@ exclude = '''
 | dist
 | blib2to3
 | tests/data
+| venv
 )/
 '''

test/unit/test_devices/test_ios_device.py

@@ -233,7 +233,8 @@ def test_boot_options_show_run(self, mock_boot):
         self.assertEqual(boot_options, {"sys": BOOT_IMAGE})
         self.device.native.send_command.assert_called_with("show run | inc boot")
 
-    def test_rollback(self):
+    @mock.patch.object(IOSDevice, "_get_file_system", return_value="flash:")
+    def test_rollback(self, mock_boot):
         self.device.rollback("good_checkpoint")
         self.device.native.send_command.assert_called_with("configure replace flash:good_checkpoint force")