Merge pull request #848 from kwienken/dedicated-host

allmightyspiff · web-flow · commit af96135e123e · 2017-07-27T14:06:15.000-05:00
Add support for dedicated host instances to virtual server upgrades
diff --git a/.gitignore b/.gitignore
@@ -13,4 +13,4 @@ build/*
 dist/*
 *.egg-info
 .cache
-
+.idea
diff --git a/SoftLayer/fixtures/SoftLayer_Virtual_Guest.py b/SoftLayer/fixtures/SoftLayer_Virtual_Guest.py
@@ -252,3 +252,85 @@
 setTags = True
 createArchiveTransaction = {}
 executeRescueLayer = True
+
+getUpgradeItemPrices = [
+    {
+        'id': 1007,
+        'categories': [{'id': 80,
+                        'name': 'Computing Instance',
+                        'categoryCode': 'guest_core'}],
+        'item': {
+            'capacity': '4',
+            'units': 'PRIVATE_CORE',
+            'description': 'Computing Instance (Dedicated)',
+        }
+    },
+    {
+        'id': 1144,
+        'locationGroupId': None,
+        'categories': [{'id': 80,
+                        'name': 'Computing Instance',
+                        'categoryCode': 'guest_core'}],
+        'item': {
+            'capacity': '4',
+            'units': 'CORE',
+            'description': 'Computing Instance',
+        }
+    },
+    {
+        'id': 332211,
+        'locationGroupId': 1,
+        'categories': [{'id': 80,
+                        'name': 'Computing Instance',
+                        'categoryCode': 'guest_core'}],
+        'item': {
+            'capacity': '4',
+            'units': 'CORE',
+            'description': 'Computing Instance',
+        }
+    },
+    {
+        'id': 1122,
+        'categories': [{'id': 26,
+                        'name': 'Uplink Port Speeds',
+                        'categoryCode': 'port_speed'}],
+        'item': {
+            'capacity': '1000',
+            'description': 'Public & Private Networks',
+        }
+    },
+    {
+        'id': 1144,
+        'categories': [{'id': 26,
+                        'name': 'Uplink Port Speeds',
+                        'categoryCode': 'port_speed'}],
+        'item': {
+            'capacity': '1000',
+            'description': 'Private Networks',
+        }
+    },
+    {
+        'id': 1133,
+        'categories': [{'id': 3,
+                        'name': 'RAM',
+                        'categoryCode': 'ram'}],
+        'item': {
+            'capacity': '2',
+            'description': 'RAM',
+        }
+    },
+]
+
+DEDICATED_GET_UPGRADE_ITEM_PRICES = [
+    {
+        'id': 115566,
+        'categories': [{'id': 80,
+                        'name': 'Computing Instance',
+                        'categoryCode': 'guest_core'}],
+        'item': {
+            'capacity': '4',
+            'units': 'DEDICATED_CORE',
+            'description': 'Computing Instance (Dedicated Host)',
+        }
+    },
+]
diff --git a/SoftLayer/managers/vs.py b/SoftLayer/managers/vs.py
@@ -9,6 +9,7 @@
 import itertools
 import socket
 import time
+import warnings
 
 from SoftLayer import exceptions
 from SoftLayer.managers import ordering
@@ -773,24 +774,24 @@ def upgrade(self, instance_id, cpus=None, memory=None,
         :param int instance_id: Instance id of the VS to be upgraded
         :param int cpus: The number of virtual CPUs to upgrade to
                             of a VS instance.
-        :param bool public: CPU will be in Private/Public Node.
         :param int memory: RAM of the VS to be upgraded to.
         :param int nic_speed: The port speed to set
+        :param bool public: CPU will be in Private/Public Node.
 
         :returns: bool
         """
-        package_items = self._get_package_items()
+        upgrade_prices = self._get_upgrade_prices(instance_id)
         prices = []
 
         for option, value in {'cpus': cpus,
                               'memory': memory,
                               'nic_speed': nic_speed}.items():
             if not value:
                 continue
-            price_id = self._get_price_id_for_upgrade(package_items,
-                                                      option,
-                                                      value,
-                                                      public)
+            price_id = self._get_price_id_for_upgrade_option(upgrade_prices,
+                                                             option,
+                                                             value,
+                                                             public)
             if not price_id:
                 # Every option provided is expected to have a price
                 raise exceptions.SoftLayerError(
@@ -815,7 +816,12 @@ def upgrade(self, instance_id, cpus=None, memory=None,
         return False
 
     def _get_package_items(self):
-        """Following Method gets all the item ids related to VS."""
+        """Following Method gets all the item ids related to VS.
+
+        Deprecated in favor of _get_upgrade_prices()
+        """
+        warnings.warn("use _get_upgrade_prices() instead",
+                      DeprecationWarning)
         mask = [
             'description',
             'capacity',
@@ -833,15 +839,76 @@ def _get_package_items(self):
         package_service = self.client['Product_Package']
         return package_service.getItems(id=package['id'], mask=mask)
 
+    def _get_upgrade_prices(self, instance_id, include_downgrade_options=True):
+        """Following Method gets all the price ids related to upgrading a VS.
+
+        :param int instance_id: Instance id of the VS to be upgraded
+
+        :returns: list
+        """
+        mask = [
+            'id',
+            'locationGroupId',
+            'categories[name,id,categoryCode]',
+            'item[description,capacity,units]'
+        ]
+        mask = "mask[%s]" % ','.join(mask)
+        return self.guest.getUpgradeItemPrices(include_downgrade_options, id=instance_id, mask=mask)
+
+    def _get_price_id_for_upgrade_option(self, upgrade_prices, option, value,
+                                         public=True):
+        """Find the price id for the option and value to upgrade. This
+
+        :param list upgrade_prices: Contains all the prices related to a VS upgrade
+        :param string option: Describes type of parameter to be upgraded
+        :param int value: The value of the parameter to be upgraded
+        :param bool public: CPU will be in Private/Public Node.
+        """
+        option_category = {
+            'memory': 'ram',
+            'cpus': 'guest_core',
+            'nic_speed': 'port_speed'
+        }
+        category_code = option_category.get(option)
+        for price in upgrade_prices:
+            if price.get('categories') is None or price.get('item') is None:
+                continue
+
+            product = price.get('item')
+            is_private = (product.get('units') == 'PRIVATE_CORE'
+                          or product.get('units') == 'DEDICATED_CORE')
+
+            for category in price.get('categories'):
+                if not (category.get('categoryCode') == category_code
+                        and str(product.get('capacity')) == str(value)):
+                    continue
+
+                if option == 'cpus':
+                    # Public upgrade and public guest_core price
+                    if public and not is_private:
+                        return price.get('id')
+                    # Private upgrade and private guest_core price
+                    elif not public and is_private:
+                        return price.get('id')
+                elif option == 'nic_speed':
+                    if 'Public' in product.get('description'):
+                        return price.get('id')
+                else:
+                    return price.get('id')
+
     def _get_price_id_for_upgrade(self, package_items, option, value,
                                   public=True):
         """Find the price id for the option and value to upgrade.
 
+        Deprecated in favor of _get_price_id_for_upgrade_option()
+
         :param list package_items: Contains all the items related to an VS
         :param string option: Describes type of parameter to be upgraded
         :param int value: The value of the parameter to be upgraded
         :param bool public: CPU will be in Private/Public Node.
         """
+        warnings.warn("use _get_price_id_for_upgrade_option() instead",
+                      DeprecationWarning)
         option_category = {
             'memory': 'ram',
             'cpus': 'guest_core',
diff --git a/tests/managers/vs_tests.py b/tests/managers/vs_tests.py
@@ -7,7 +7,6 @@
 import mock
 
 import SoftLayer
-from SoftLayer import exceptions
 from SoftLayer import fixtures
 from SoftLayer import testing
 
@@ -636,13 +635,6 @@ def test_capture_additional_disks(self):
                                 identifier=1)
 
     def test_upgrade(self):
-        mock = self.set_mock('SoftLayer_Product_Package', 'getAllObjects')
-        mock.return_value = [
-            {'id': 46, 'name': 'Virtual Servers',
-             'description': 'Virtual Server Instances',
-             'type': {'keyName': 'VIRTUAL_SERVER_INSTANCE'}, 'isActive': 1},
-        ]
-
         # test single upgrade
         result = self.vs.upgrade(1, cpus=4, public=False)
 
@@ -678,10 +670,19 @@ def test_upgrade_full(self):
         self.assertIn({'id': 1122}, order_container['prices'])
         self.assertEqual(order_container['virtualGuests'], [{'id': 1}])
 
-    def test_upgrade_skips_location_based_prices(self):
-        # Test that no prices that have locationGroupId set are used
-        self.assertRaises(exceptions.SoftLayerError,
-                          self.vs.upgrade, 1, cpus=55, memory=2, public=True)
+    def test_upgrade_dedicated_host_instance(self):
+        mock = self.set_mock('SoftLayer_Virtual_Guest', 'getUpgradeItemPrices')
+        mock.return_value = fixtures.SoftLayer_Virtual_Guest.DEDICATED_GET_UPGRADE_ITEM_PRICES
+
+        # test single upgrade
+        result = self.vs.upgrade(1, cpus=4, public=False)
+
+        self.assertEqual(result, True)
+        self.assert_called_with('SoftLayer_Product_Order', 'placeOrder')
+        call = self.calls('SoftLayer_Product_Order', 'placeOrder')[0]
+        order_container = call.args[0]
+        self.assertEqual(order_container['prices'], [{'id': 115566}])
+        self.assertEqual(order_container['virtualGuests'], [{'id': 1}])
 
     def test_get_item_id_for_upgrade(self):
         item_id = 0
@@ -693,6 +694,48 @@ def test_get_item_id_for_upgrade(self):
                 break
         self.assertEqual(1133, item_id)
 
+    def test_get_package_items(self):
+        self.vs._get_package_items()
+        self.assert_called_with('SoftLayer_Product_Package', 'getItems')
+
+    def test_get_package_items_errors(self):
+        mock = self.set_mock('SoftLayer_Product_Package', 'getAllObjects')
+        mock.return_value = []
+
+        self.assertRaises(ValueError, self.vs._get_package_items)
+
+    def test_get_price_id_for_upgrade(self):
+        package_items = self.vs._get_package_items()
+
+        price_id = self.vs._get_price_id_for_upgrade(package_items=package_items,
+                                                     option='cpus',
+                                                     value='4')
+        self.assertEqual(1144, price_id)
+
+    def test_get_price_id_for_upgrade_skips_location_price(self):
+        package_items = self.vs._get_package_items()
+
+        price_id = self.vs._get_price_id_for_upgrade(package_items=package_items,
+                                                     option='cpus',
+                                                     value='55')
+        self.assertEqual(None, price_id)
+
+    def test_get_price_id_for_upgrade_finds_nic_price(self):
+        package_items = self.vs._get_package_items()
+
+        price_id = self.vs._get_price_id_for_upgrade(package_items=package_items,
+                                                     option='memory',
+                                                     value='2')
+        self.assertEqual(1133, price_id)
+
+    def test_get_price_id_for_upgrade_finds_memory_price(self):
+        package_items = self.vs._get_package_items()
+
+        price_id = self.vs._get_price_id_for_upgrade(package_items=package_items,
+                                                     option='nic_speed',
+                                                     value='1000')
+        self.assertEqual(1122, price_id)
+
 
 class VSWaitReadyGoTests(testing.TestCase):
 

-Original file line number
+Diff line change
 dist/*
 *.egg-info
 .cache
+-
 +.idea