slcli vlan cancel should report if a vlan is automatic

Author: caberos

SoftLayer/CLI/routes.py

@@ -336,6 +336,7 @@
     ('vlan:detail', 'SoftLayer.CLI.vlan.detail:cli'),
     ('vlan:edit', 'SoftLayer.CLI.vlan.edit:cli'),
     ('vlan:list', 'SoftLayer.CLI.vlan.list:cli'),
+    ('vlan:cancel', 'SoftLayer.CLI.vlan.cancel:cli'),
 
     ('summary', 'SoftLayer.CLI.summary:cli'),
 

SoftLayer/CLI/vlan/cancel.py

@@ -0,0 +1,30 @@
+"""Cancel Network Vlan."""
+# :license: MIT, see LICENSE for more details.
+
+import click
+
+import SoftLayer
+from SoftLayer.CLI import environment
+from SoftLayer.CLI import exceptions
+from SoftLayer.CLI import formatting
+from SoftLayer.managers.billing import BillingManager
+
+
+@click.command()
+@click.argument('identifier')
+@environment.pass_env
+def cli(env, identifier):
+    """Cancel network vlan."""
+
+    mgr = SoftLayer.NetworkManager(env.client)
+    billing = BillingManager(env.client)
+    if not (env.skip_confirmations or formatting.no_going_back(identifier)):
+        raise exceptions.CLIAbort('Aborted')
+
+    item = mgr.get_vlan(identifier).get('billingItem')
+    if item:
+        billing.cancel_item(item.get('id'), 'cancel by cli command')
+        env.fout('Cancel Successfully')
+    else:
+        res = mgr.get_cancel_failure_reasons(identifier)
+        raise exceptions.ArgumentError(res)

SoftLayer/fixtures/SoftLayer_Network_Vlan.py

@@ -5,9 +5,19 @@
     },
     'id': 1234,
     'vlanNumber': 4444,
-    'firewallInterfaces': None
+    'firewallInterfaces': None,
+    'billingItem': {
+            'allowCancellationFlag': 1,
+            'categoryCode': 'network_vlan',
+            'description': 'Private Network Vlan',
+            'id': 235689,
+            'notes': 'test cli',
+            'orderItemId': 147258,
+        }
 }
 
 editObject = True
 setTags = True
 getList = [getObject]
+
+cancel = True

SoftLayer/managers/billing.py

@@ -0,0 +1,27 @@
+"""
+    SoftLayer.BillingItem
+    ~~~~~~~~~~~~~~~~~~~
+    BillingItem manager
+
+    :license: MIT, see LICENSE for more details.
+"""
+
+
+class BillingManager(object):
+    """Manager for interacting with Billing item instances."""
+
+    def __init__(self, client):
+        self.client = client
+
+    def cancel_item(self, identifier, reason_cancel):
+        """Cancel a billing item immediately, deleting all its data.
+
+        :param integer identifier: the instance ID to cancel
+        :param string reason_cancel: reason cancel
+        """
+        return self.client.call('SoftLayer_Billing_Item', 'cancelItem',
+                                True,
+                                True,
+                                reason_cancel,
+                                reason_cancel,
+                                id=identifier)

SoftLayer/managers/network.py

@@ -49,6 +49,7 @@
     'primaryRouter[id, fullyQualifiedDomainName, datacenter]',
     'totalPrimaryIpAddressCount',
     'networkSpace',
+    'billingItem',
     'hardware',
     'subnets',
     'virtualGuests',
@@ -752,3 +753,10 @@ def set_subnet_ipddress_note(self, identifier, note):
         """
         result = self.client.call('SoftLayer_Network_Subnet_IpAddress', 'editObject', note, id=identifier)
         return result
+
+    def get_cancel_failure_reasons(self, identifier):
+        """get the reasons by cannot cancel the VLAN
+
+        :param integer identifier:  the instance ID
+        """
+        return self.vlan.getCancelFailureReasons(id=identifier)

docs/api/managers/billing.rst

@@ -0,0 +1,5 @@
+.. _billing:
+
+.. automodule:: SoftLayer.managers.billing
+   :members:
+   :inherited-members:

docs/cli/vlan.rst

@@ -14,3 +14,7 @@ VLANs
 .. click:: SoftLayer.CLI.vlan.list:cli
     :prog: vlan list
     :show-nested:
+
+.. click:: SoftLayer.CLI.vlan.cancel:cli
+    :prog: vlan cancel
+    :show-nested:

tests/CLI/modules/vlan_tests.py

@@ -99,3 +99,15 @@ def test_vlan_list(self):
         result = self.run_command(['vlan', 'list'])
         self.assert_no_fail(result)
         self.assert_called_with('SoftLayer_Account', 'getNetworkVlans')
+
+    @mock.patch('SoftLayer.CLI.formatting.no_going_back')
+    def test_vlan_cancel(self, confirm_mock):
+        confirm_mock.return_value = True
+        result = self.run_command(['vlan', 'cancel', '1234'])
+        self.assert_no_fail(result)
+
+    @mock.patch('SoftLayer.CLI.formatting.no_going_back')
+    def test_vlan_cancel_fail(self, confirm_mock):
+        confirm_mock.return_value = False
+        result = self.run_command(['vlan', 'cancel', '1234'])
+        self.assertTrue(result.exit_code, 2)