Adding disaster recovery failover api

Author: Rajarajan

SoftLayer/CLI/block/replication/disaster_recovery_failover.py

@@ -0,0 +1,34 @@
+"""Failover an inaccessible file volume to its available replicant volume."""
+# :license: MIT, see LICENSE for more details.
+
+import click
+import SoftLayer
+from SoftLayer.CLI import environment
+from SoftLayer.CLI import formatting
+from SoftLayer.CLI import exceptions
+
+
+@click.command()
+@click.argument('volume-id')
+@click.option('--replicant-id', help="ID of the replicant volume")
+@environment.pass_env
+def cli(env, volume_id, replicant_id):
+    """Failover an inaccessible file volume to its available replicant volume."""
+    block_storage_manager = SoftLayer.BlockStorageManager(env.client)
+
+    click.secho("""WARNING:Disaster Recovery Failover a block volume to the given replicant volume.\n"""
+                """* This action cannot be undone\n"""
+                """* You will not be able to perform failback to the original\n"""
+                """* You cannot failover without replica""",fg = 'red' )
+   
+    if not (formatting.confirm('Are you sure you want to continue?')):
+        raise exceptions.CLIAbort('Aborted.')
+
+    success = block_storage_manager.disaster_recovery_failover_to_replicant(
+        volume_id,
+        replicant_id
+    )
+    if success:
+        click.echo("Disaster Recovery Failover to replicant is now in progress.")
+    else:
+        click.echo("Disaster Recovery Failover operation could not be initiated.")

SoftLayer/CLI/file/replication/disaster_recovery_failover.py

@@ -0,0 +1,34 @@
+"""Failover an inaccessible file volume to its available replicant volume."""
+# :license: MIT, see LICENSE for more details.
+
+import click
+import SoftLayer
+from SoftLayer.CLI import environment
+from SoftLayer.CLI import formatting
+from SoftLayer.CLI import exceptions
+
+
+@click.command()
+@click.argument('volume-id')
+@click.option('--replicant-id', help="ID of the replicant volume")
+@environment.pass_env
+def cli(env, volume_id, replicant_id):
+    """Failover an inaccessible file volume to its available replicant volume."""
+    file_storage_manager = SoftLayer.FileStorageManager(env.client)
+
+    click.secho("""WARNING : Disaster Recovery Failover should not be performed unless data center for the primary volume is unreachable.\n"""
+                """* This action cannot be undone\n"""
+                """* You will not be able to perform failback to the original without support intervention\n"""
+                """* You cannot failover without replica""",fg = 'red' )
+    
+    if not (formatting.confirm('Are you sure you want to continue?')):
+        raise exceptions.CLIAbort('Aborted.')
+
+    success = file_storage_manager.disaster_recovery_failover_to_replicant(
+        volume_id,
+        replicant_id
+    )
+    if success:
+        click.echo("Disaster Recovery Failover to replicant is now in progress.")
+    else:
+        click.echo("Disaster Recovery Failover operation could not be initiated.")

SoftLayer/CLI/routes.py

@@ -92,6 +92,7 @@
     ('block:subnets-remove', 'SoftLayer.CLI.block.subnets.remove:cli'),
     ('block:replica-failback', 'SoftLayer.CLI.block.replication.failback:cli'),
     ('block:replica-failover', 'SoftLayer.CLI.block.replication.failover:cli'),
+    ('block:disaster-recovery-failover', 'SoftLayer.CLI.block.replication.disaster_recovery_failover:cli'),
     ('block:replica-order', 'SoftLayer.CLI.block.replication.order:cli'),
     ('block:replica-partners', 'SoftLayer.CLI.block.replication.partners:cli'),
     ('block:replica-locations', 'SoftLayer.CLI.block.replication.locations:cli'),
@@ -127,6 +128,7 @@
     ('file:access-revoke', 'SoftLayer.CLI.file.access.revoke:cli'),
     ('file:replica-failback', 'SoftLayer.CLI.file.replication.failback:cli'),
     ('file:replica-failover', 'SoftLayer.CLI.file.replication.failover:cli'),
+    ('file:disaster-recovery-failover', 'SoftLayer.CLI.file.replication.disaster_recovery_failover:cli'),
     ('file:replica-order', 'SoftLayer.CLI.file.replication.order:cli'),
     ('file:replica-partners', 'SoftLayer.CLI.file.replication.partners:cli'),
     ('file:replica-locations', 'SoftLayer.CLI.file.replication.locations:cli'),

SoftLayer/fixtures/SoftLayer_Network_Storage.py

@@ -221,6 +221,7 @@
 failoverToReplicant = True
 failbackFromReplicant = True
 restoreFromSnapshot = True
+disasterRecoveryFailoverToReplicant = True
 
 createSnapshot = {
     'id': 449

SoftLayer/managers/storage.py

@@ -398,13 +398,22 @@ def failover_to_replicant(self, volume_id, replicant_id):
         """
         return self.client.call('Network_Storage', 'failoverToReplicant', replicant_id, id=volume_id)
 
+    def disaster_recovery_failover_to_replicant(self, volume_id, replicant_id):
+        """Disaster Recovery Failover to a volume replicant.
+
+        :param integer volume_id: The id of the volume
+        :param integer replicant: ID of replicant to failover to
+        :return: Returns whether failover to successful or not
+        """
+        return self.client.call('Network_Storage', 'disasterRecoveryFailoverToReplicant', replicant_id, id=volume_id)    
+
     def failback_from_replicant(self, volume_id):
         """Failback from a volume replicant.
 
         :param integer volume_id: The id of the volume
         :return: Returns whether failback was successful or not
         """
-
+        
         return self.client.call('Network_Storage', 'failbackFromReplicant', id=volume_id)
 
     def cancel_volume(self, volume_id, reason='No longer needed', immediate=False):

docs/cli/block.rst

@@ -142,3 +142,7 @@ Block Commands
 .. click:: SoftLayer.CLI.block.set_note:cli
     :prog: block volume-set-note
     :show-nested:
+
+.. click:: SoftLayer.CLI.block.replication.disaster_recovery_failover:cli
+    :prog: block disaster-recovery-failover
+    :show-nested:

docs/cli/file.rst

@@ -121,4 +121,8 @@ File Commands
 
 .. click:: SoftLayer.CLI.file.set_note:cli
     :prog: file volume-set-note
+    :show-nested:
+
+.. click:: SoftLayer.CLI.file.replication.disaster_recovery_failover:cli
+    :prog: file disaster-recovery-failover
     :show-nested:

tests/CLI/modules/block_tests.py

@@ -4,8 +4,9 @@
 
     :license: MIT, see LICENSE for more details.
 """
-from SoftLayer import exceptions
+from SoftLayer import SoftLayerAPIError
 from SoftLayer import testing
+from SoftLayer.CLI import exceptions
 
 import json
 import mock
@@ -48,7 +49,7 @@ def test_volume_set_lun_id_in_range_missing_value(self):
     def test_volume_set_lun_id_not_in_range(self):
         value = '-1'
         lun_mock = self.set_mock('SoftLayer_Network_Storage', 'createOrUpdateLunId')
-        lun_mock.side_effect = exceptions.SoftLayerAPIError(
+        lun_mock.side_effect = SoftLayerAPIError(
             'SoftLayer_Exception_Network_Storage_Iscsi_InvalidLunId',
             'The LUN ID specified is out of the valid range: %s [min: 0 max: 4095]' % (value))
         result = self.run_command('block volume-set-lun-id 1234 42'.split())
@@ -498,6 +499,18 @@ def test_replicant_failover(self):
         self.assert_no_fail(result)
         self.assertEqual('Failover to replicant is now in progress.\n',
                          result.output)
+    
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    @mock.patch('SoftLayer.BlockStorageManager.disaster_recovery_failover_to_replicant')
+    def test_disaster_recovery_failover(self, disaster_recovery_failover_mock, confirm_mock):
+        confirm_mock.return_value = True
+        disaster_recovery_failover_mock.return_value = True
+        result = self.run_command(['block', 'disaster-recovery-failover', '12345678',
+                                   '--replicant-id=5678'])
+
+        self.assert_no_fail(result)
+        self.assertIn('Disaster Recovery Failover to replicant is now in progress.\n',
+                         result.output)
 
     def test_replication_locations(self):
         result = self.run_command(['block', 'replica-locations', '1234'])
@@ -558,6 +571,28 @@ def test_replicant_failover_unsuccessful(self, failover_mock):
         self.assertEqual('Failover operation could not be initiated.\n',
                          result.output)
 
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    @mock.patch('SoftLayer.BlockStorageManager.disaster_recovery_failover_to_replicant')
+    def test_disaster_recovery_failover_unsuccesful(self, disaster_recovery_failover_mock, confirm_mock):
+        confirm_mock.return_value = True
+        disaster_recovery_failover_mock.return_value = False
+
+        result = self.run_command(['block', 'disaster-recovery-failover', '12345678',
+                                   '--replicant-id=5678'])
+
+        self.assertIn('Disaster Recovery Failover operation could not be initiated.\n',
+                         result.output)
+
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    def test_disaster_recovery_failover_aborted(self, confirm_mock):
+        confirm_mock.return_value = False
+
+        result = self.run_command(['block', 'disaster-recovery-failover', '12345678',
+                                   '--replicant-id=5678'])
+
+        self.assertEqual(result.exit_code, 2)
+        self.assertIsInstance(result.exception, exceptions.CLIAbort)    
+
     def test_replicant_failback(self):
         result = self.run_command(['block', 'replica-failback', '12345678'])
 

tests/CLI/modules/file_tests.py

@@ -4,8 +4,9 @@
 
     :license: MIT, see LICENSE for more details.
 """
-from SoftLayer import exceptions
+from SoftLayer import SoftLayerError
 from SoftLayer import testing
+from SoftLayer.CLI import exceptions
 
 import json
 import mock
@@ -126,7 +127,7 @@ def test_volume_cancel_without_billing_item(self):
         result = self.run_command([
             '--really', 'file', 'volume-cancel', '1234'])
 
-        self.assertIsInstance(result.exception, exceptions.SoftLayerError)
+        self.assertIsInstance(result.exception, SoftLayerError)
 
     def test_volume_detail(self):
         result = self.run_command(['file', 'volume-detail', '1234'])
@@ -493,6 +494,41 @@ def test_replicant_failover_unsuccessful(self, failover_mock):
         self.assertEqual('Failover operation could not be initiated.\n',
                          result.output)
 
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    @mock.patch('SoftLayer.FileStorageManager.disaster_recovery_failover_to_replicant')
+    def test_disaster_recovery_failover(self, disaster_recovery_failover_mock, confirm_mock):
+        confirm_mock.return_value = True
+        disaster_recovery_failover_mock.return_value = True
+        result = self.run_command(['file', 'disaster-recovery-failover', '12345678',
+                                   '--replicant-id=5678'])
+
+        self.assert_no_fail(result)
+        self.assertEqual('Disaster Recovery Failover to replicant is now in progress.\n',
+                         result.output)
+
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    @mock.patch('SoftLayer.FileStorageManager.disaster_recovery_failover_to_replicant')
+    def test_disaster_recovery_failover_unsuccesful(self, disaster_recovery_failover_mock, confirm_mock):
+        confirm_mock.return_value = True
+        disaster_recovery_failover_mock.return_value = False
+
+        result = self.run_command(['file', 'disaster-recovery-failover', '12345678',
+                                   '--replicant-id=5678'])
+
+        self.assertEqual('Disaster Recovery Failover operation could not be initiated.\n',
+                         result.output)
+
+    @mock.patch('SoftLayer.CLI.formatting.confirm')
+    def test_disaster_recovery_failover_aborted(self, confirm_mock):
+        confirm_mock.return_value = False
+
+        result = self.run_command(['file', 'disaster-recovery-failover', '12345678',
+                                   '--replicant-id=5678'])
+
+        self.assertEqual(result.exit_code, 2)
+        self.assertIsInstance(result.exception, exceptions.CLIAbort)                          
+
+
     def test_replicant_failback(self):
         result = self.run_command(['file', 'replica-failback', '12345678'])
 

tests/managers/block_tests.py

@@ -384,6 +384,18 @@ def test_replicant_failover(self):
             identifier=1234,
         )
 
+    def test_disaster_recovery_failover(self):
+        result = self.block.disaster_recovery_failover_to_replicant(1234, 5678)
+
+        self.assertEqual(
+            SoftLayer_Network_Storage.disasterRecoveryFailoverToReplicant, result)
+        self.assert_called_with(
+            'SoftLayer_Network_Storage',
+            'disasterRecoveryFailoverToReplicant',
+            args=(5678,),
+            identifier=1234,
+        )        
+
     def test_replicant_failback(self):
         result = self.block.failback_from_replicant(1234)
 

tests/managers/file_tests.py

@@ -289,6 +289,18 @@ def test_replicant_failover(self):
             identifier=1234,
         )
 
+    def test_disaster_recovery_failover(self):
+            result = self.file.disaster_recovery_failover_to_replicant(1234, 5678)
+
+            self.assertEqual(
+                SoftLayer_Network_Storage.disasterRecoveryFailoverToReplicant, result)
+            self.assert_called_with(
+                'SoftLayer_Network_Storage',
+                'disasterRecoveryFailoverToReplicant',
+                args=(5678,),
+                identifier=1234,
+            )
+
     def test_replicant_failback(self):
         result = self.file.failback_from_replicant(1234)