add a new option on account orders

Author: caberos

SoftLayer/CLI/account/orders.py

@@ -10,16 +10,34 @@
 from SoftLayer import utils
 
 
+def upgrade_table(upgrades):
+    """Formats a table for upgrade orders"""
+    upgrade_table = formatting.Table(['Id', 'Maintance window', 'Status', 'Created Date',
+                                      'Case'], title="Upgrade orders")
+    upgrade_table.align['Subject'] = 'l'
+    upgrade_table.align['Impacted Resources'] = 'l'
+    for upgrade in upgrades:
+        upgrade_table.add_row([upgrade.get('id'),
+                               upgrade.get('maintenanceStartTimeUtc'),
+                               upgrade.get('statusId'),
+                               upgrade.get('createDate'),
+                               upgrade.get('ticketId') or '--'])
+    return upgrade_table
+
+
 @click.command(cls=SLCommand)
 @click.option('--limit', '-l',
               help='How many results to get in one api call',
-              default=50,
+              default=100,
               show_default=True)
+@click.option('--upgrades', is_flag=True, default=False,
+              help="Show upgrades orders.")
 @environment.pass_env
-def cli(env, limit):
+def cli(env, limit, upgrades):
     """Lists account orders. Use `slcli order lookup <ID>` to find more details about a specific order."""
     manager = AccountManager(env.client)
     orders = manager.get_account_all_billing_orders(limit)
+    upgrade = manager.get_account_upgrade_orders(limit)
 
     order_table = formatting.Table(['Id', 'State', 'User', 'Date', 'Amount', 'Item'],
                                    title="orders")
@@ -34,3 +52,5 @@ def cli(env, limit):
         order_table.add_row([order['id'], order['status'], order['userRecord']['username'], create_date,
                              order['orderTotalAmount'], utils.trim_to(' '.join(map(str, items)), 50)])
     env.fout(order_table)
+    if upgrades:
+        env.fout(upgrade_table(upgrade))

SoftLayer/fixtures/SoftLayer_Account.py

@@ -1480,3 +1480,24 @@
         "name": "GettestProv12345",
         "uri": "http://test.com"
     }]
+
+getUpgradeRequests = [{
+    "createDate": "2023-03-01T19:39:51-06:00",
+    "id": 3237486,
+    "maintenanceStartTimeUtc": "2023-03-01T19:39:51-06:00",
+    "modifyDate": "2023-03-01T19:39:53-06:00",
+    "orderId": 103127604,
+    "statusId": 3,
+    "ticketId": 152510472,
+    "userId": 167758
+},
+    {
+        "createDate": "2023-02-27T19:41:26-06:00",
+        "id": 3229998,
+        "maintenanceStartTimeUtc": "2023-02-27T19:41:26-06:00",
+        "modifyDate": "2023-02-27T19:41:29-06:00",
+        "orderId": 103029352,
+        "statusId": 3,
+        "ticketId": 152452042,
+        "userId": 167758
+    }]

SoftLayer/managers/account.py

@@ -407,3 +407,7 @@ def create_provisioning(self, name, uri):
             'uri': uri
         }
         return self.client.call('SoftLayer_Provisioning_Hook', 'createObject', template)
+
+    def get_account_upgrade_orders(self, limit=100):
+        """Gets upgrade order list"""
+        return self.client.call('SoftLayer_Account', 'getUpgradeRequests', limit=limit)

tests/CLI/modules/account_tests.py

@@ -169,3 +169,8 @@ def test_created_provisioning_hook(self):
         result = self.run_command(['account', 'hook-create', '--name', 'testslcli', '--uri', 'http://slclitest.com'])
         self.assert_no_fail(result)
         self.assert_called_with('SoftLayer_Provisioning_Hook', 'createObject')
+
+    def test_order_upgrade(self):
+        result = self.run_command(['account', 'orders', '--upgrades'])
+        self.assert_no_fail(result)
+        self.assert_called_with('SoftLayer_Account', 'getUpgradeRequests')

tests/managers/account_tests.py

@@ -184,3 +184,7 @@ def test_get_provisioning_scripts(self):
     def test_create_provisioning_scripts(self):
         self.manager.create_provisioning('testslcli', 'http://slclitest.com')
         self.assert_called_with('SoftLayer_Provisioning_Hook', 'createObject')
+
+    def test_get_upgrades_orders(self):
+        self.manager.get_account_upgrade_orders()
+        self.assert_called_with("SoftLayer_Account", "getUpgradeRequests")