added tests and route for server:power:rescue

Author: allmightyspiff

SoftLayer/CLI/hardware/power.py

@@ -77,3 +77,20 @@ def power_cycle(env, identifier):
         raise exceptions.CLIAbort('Aborted.')
 
     env.client['Hardware_Server'].powerCycle(id=hw_id)
+
+
+@click.command()
+@click.argument('identifier')
+@environment.pass_env
+def rescue(env, identifier):
+    """Reboot server into a rescue image."""
+
+    mgr = SoftLayer.HardwareManager(env.client)
+    hw_id = helpers.resolve_id(mgr.resolve_ids, identifier, 'hardware')
+
+    if not (env.skip_confirmations or
+            formatting.confirm("This action will reboot this server. Continue?")):
+
+        raise exceptions.CLIAbort('Aborted')
+
+    env.client['Hardware_Server'].bootToRescueLayer(id=hw_id)

SoftLayer/CLI/hardware/rescue.py

@@ -210,6 +210,7 @@
     ('hardware:reload', 'SoftLayer.CLI.hardware.reload:cli'),
     ('hardware:credentials', 'SoftLayer.CLI.hardware.credentials:cli'),
     ('hardware:update-firmware', 'SoftLayer.CLI.hardware.update_firmware:cli'),
+    ('hardware:rescue', 'SoftLayer.CLI.hardware.power:rescue'),
 
     ('securitygroup', 'SoftLayer.CLI.securitygroup'),
     ('securitygroup:list', 'SoftLayer.CLI.securitygroup.list:cli'),

SoftLayer/CLI/routes.py

@@ -445,3 +445,12 @@ def test_edit(self):
             args=(100,),
             identifier=100,
         )
+
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    def test_rescue(self, confirm_mock):
+        confirm_mock.return_value = True
+        result = self.run_command(['server', 'rescue', '1000'])
+
+        self.assert_no_fail(result)
+        self.assertEqual(result.output, "")
+        self.assert_called_with('SoftLayer_Hardware_Server', 'bootToRescueLayer', identifier=1000)