snaps/openstack/tests/create_flavor_tests.py

   1 # You may obtain a copy of the License at:
   2 #
   3 #     http://www.apache.org/licenses/LICENSE-2.0
   4 #
   5 # Unless required by applicable law or agreed to in writing, software
   6 # distributed under the License is distributed on an "AS IS" BASIS,
   7 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   8 # See the License for the specific language governing permissions and
   9 # limitations under the License.
  10 import uuid
  11 import unittest
  12
  13 from snaps.openstack.create_flavor import FlavorSettings, OpenStackFlavor
  14 from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
  15 from snaps.openstack.utils import nova_utils
  16
  17 __author__ = 'spisarski'
  18
  19
  20 class FlavorSettingsUnitTests(unittest.TestCase):
  21     """
  22     Tests the construction of the FlavorSettings class
  23     """
  24
  25     def test_no_params(self):
  26         with self.assertRaises(Exception):
  27             FlavorSettings()
  28
  29     def test_empty_config(self):
  30         with self.assertRaises(Exception):
  31             FlavorSettings(config=dict())
  32
  33     def test_name_only(self):
  34         with self.assertRaises(Exception):
  35             FlavorSettings(name='foo')
  36
  37     def test_config_with_name_only(self):
  38         with self.assertRaises(Exception):
  39             FlavorSettings(config={'name': 'foo'})
  40
  41     def test_name_ram_only(self):
  42         with self.assertRaises(Exception):
  43             FlavorSettings(name='foo', ram=1)
  44
  45     def test_config_with_name_ram_only(self):
  46         with self.assertRaises(Exception):
  47             FlavorSettings(config={'name': 'foo', 'ram': 1})
  48
  49     def test_name_ram_disk_only(self):
  50         with self.assertRaises(Exception):
  51             FlavorSettings(name='foo', ram=1, disk=1)
  52
  53     def test_config_with_name_ram_disk_only(self):
  54         with self.assertRaises(Exception):
  55             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 1})
  56
  57     def test_ram_string(self):
  58         with self.assertRaises(Exception):
  59             FlavorSettings(name='foo', ram='bar', disk=2, vcpus=3, ephemeral=4, swap=5, rxtx_factor=6.0,
  60                            is_public=False)
  61
  62     def test_config_ram_string(self):
  63         with self.assertRaises(Exception):
  64             FlavorSettings(config={'name': 'foo', 'ram': 'bar', 'disk': 2, 'vcpus': 3, 'ephemeral': 4, 'swap': 5,
  65                                    'rxtx_factor': 6.0, 'is_public': False})
  66
  67     def test_ram_float(self):
  68         with self.assertRaises(Exception):
  69             FlavorSettings(name='foo', ram=1.5, disk=2, vcpus=3, ephemeral=4, swap=5, rxtx_factor=6.0, is_public=False)
  70
  71     def test_config_ram_float(self):
  72         with self.assertRaises(Exception):
  73             FlavorSettings(config={'name': 'foo', 'ram': 1.5, 'disk': 2, 'vcpus': 3, 'ephemeral': 4, 'swap': 5,
  74                                    'rxtx_factor': 6.0, 'is_public': False})
  75
  76     def test_disk_string(self):
  77         with self.assertRaises(Exception):
  78             FlavorSettings(name='foo', ram=1, disk='bar', vcpus=3, ephemeral=4, swap=5, rxtx_factor=6.0,
  79                            is_public=False)
  80
  81     def test_config_disk_string(self):
  82         with self.assertRaises(Exception):
  83             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 'bar', 'vcpus': 3, 'ephemeral': 4, 'swap': 5,
  84                                    'rxtx_factor': 6.0, 'is_public': False})
  85
  86     def test_disk_float(self):
  87         with self.assertRaises(Exception):
  88             FlavorSettings(name='foo', ram=1, disk=2.5, vcpus=3, ephemeral=4, swap=5, rxtx_factor=6.0, is_public=False)
  89
  90     def test_config_disk_float(self):
  91         with self.assertRaises(Exception):
  92             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2.5, 'vcpus': 3, 'ephemeral': 4, 'swap': 5,
  93                                    'rxtx_factor': 6.0, 'is_public': False})
  94
  95     def test_vcpus_string(self):
  96         with self.assertRaises(Exception):
  97             FlavorSettings(name='foo', ram=1, disk=2, vcpus='bar', ephemeral=4, swap=5, rxtx_factor=6.0,
  98                            is_public=False)
  99
 100     def test_config_vcpus_string(self):
 101         with self.assertRaises(Exception):
 102             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 'bar', 'ephemeral': 4, 'swap': 5,
 103                                    'rxtx_factor': 6.0, 'is_public': False})
 104
 105     def test_ephemeral_string(self):
 106         with self.assertRaises(Exception):
 107             FlavorSettings(name='foo', ram=1, disk=2, vcpus=3, ephemeral='bar', swap=5, rxtx_factor=6.0,
 108                            is_public=False)
 109
 110     def test_config_ephemeral_string(self):
 111         with self.assertRaises(Exception):
 112             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3, 'ephemeral': 'bar', 'swap': 5,
 113                                    'rxtx_factor': 6.0, 'is_public': False})
 114
 115     def test_ephemeral_float(self):
 116         with self.assertRaises(Exception):
 117             FlavorSettings(name='foo', ram=1, disk=2, vcpus=3, ephemeral=4.5, swap=5, rxtx_factor=6.0, is_public=False)
 118
 119     def test_config_ephemeral_float(self):
 120         with self.assertRaises(Exception):
 121             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3, 'ephemeral': 4.5, 'swap': 5,
 122                                    'rxtx_factor': 6.0, 'is_public': False})
 123
 124     def test_swap_string(self):
 125         with self.assertRaises(Exception):
 126             FlavorSettings(name='foo', ram=1, disk=2, vcpus=3, ephemeral=4, swap='bar', rxtx_factor=6.0,
 127                            is_public=False)
 128
 129     def test_config_swap_string(self):
 130         with self.assertRaises(Exception):
 131             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3, 'ephemeral': 4, 'swap': 'bar',
 132                                    'rxtx_factor': 6.0, 'is_public': False})
 133
 134     def test_swap_float(self):
 135         with self.assertRaises(Exception):
 136             FlavorSettings(name='foo', ram=1, disk=2, vcpus=3, ephemeral=4, swap=5.5, rxtx_factor=6.0, is_public=False)
 137
 138     def test_config_swap_float(self):
 139         with self.assertRaises(Exception):
 140             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3, 'ephemeral': 4, 'swap': 5.5,
 141                                    'rxtx_factor': 6.0, 'is_public': False})
 142
 143     def test_rxtx_string(self):
 144         with self.assertRaises(Exception):
 145             FlavorSettings(name='foo', ram=1, disk=2, vcpus=3, ephemeral=4, swap=5, rxtx_factor='bar', is_public=False)
 146
 147     def test_config_rxtx_string(self):
 148         with self.assertRaises(Exception):
 149             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3, 'ephemeral': 4, 'swap': 5,
 150                                    'rxtx_factor': 'bar', 'is_public': False})
 151
 152     def test_is_pub_string(self):
 153         with self.assertRaises(Exception):
 154             FlavorSettings(name='foo', ram=1, disk=2, vcpus=3, ephemeral=4, swap=5, rxtx_factor=6.0, is_public='bar')
 155
 156     def test_config_is_pub_string(self):
 157         with self.assertRaises(Exception):
 158             FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3, 'ephemeral': 4, 'swap': 5,
 159                                    'rxtx_factor': 6.0, 'is_public': 'bar'})
 160
 161     def test_name_ram_disk_vcpus_only(self):
 162         settings = FlavorSettings(name='foo', ram=1, disk=2, vcpus=3)
 163         self.assertEquals('foo', settings.name)
 164         self.assertEquals('auto', settings.flavor_id)
 165         self.assertEquals(1, settings.ram)
 166         self.assertEquals(2, settings.disk)
 167         self.assertEquals(3, settings.vcpus)
 168         self.assertEquals(0, settings.ephemeral)
 169         self.assertEquals(0, settings.swap)
 170         self.assertEquals(1.0, settings.rxtx_factor)
 171         self.assertEquals(True, settings.is_public)
 172
 173     def test_config_with_name_ram_disk_vcpus_only(self):
 174         settings = FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3})
 175         self.assertEquals('foo', settings.name)
 176         self.assertEquals('auto', settings.flavor_id)
 177         self.assertEquals(1, settings.ram)
 178         self.assertEquals(2, settings.disk)
 179         self.assertEquals(3, settings.vcpus)
 180         self.assertEquals(0, settings.ephemeral)
 181         self.assertEquals(0, settings.swap)
 182         self.assertEquals(1.0, settings.rxtx_factor)
 183         self.assertEquals(True, settings.is_public)
 184
 185     def test_all(self):
 186         settings = FlavorSettings(name='foo', flavor_id='bar', ram=1, disk=2, vcpus=3, ephemeral=4, swap=5, rxtx_factor=6.0,
 187                                   is_public=False)
 188         self.assertEquals('foo', settings.name)
 189         self.assertEquals('bar', settings.flavor_id)
 190         self.assertEquals(1, settings.ram)
 191         self.assertEquals(2, settings.disk)
 192         self.assertEquals(3, settings.vcpus)
 193         self.assertEquals(4, settings.ephemeral)
 194         self.assertEquals(5, settings.swap)
 195         self.assertEquals(6.0, settings.rxtx_factor)
 196         self.assertEquals(False, settings.is_public)
 197
 198     def test_config_all(self):
 199         settings = FlavorSettings(config={'name': 'foo', 'flavor_id': 'bar', 'ram': 1, 'disk': 2, 'vcpus': 3, 'ephemeral': 4,
 200                                           'swap': 5, 'rxtx_factor': 6.0, 'is_public': False})
 201         self.assertEquals('foo', settings.name)
 202         self.assertEquals('bar', settings.flavor_id)
 203         self.assertEquals(1, settings.ram)
 204         self.assertEquals(2, settings.disk)
 205         self.assertEquals(3, settings.vcpus)
 206         self.assertEquals(4, settings.ephemeral)
 207         self.assertEquals(5, settings.swap)
 208         self.assertEquals(6.0, settings.rxtx_factor)
 209         self.assertEquals(False, settings.is_public)
 210
 211
 212 class CreateFlavorTests(OSComponentTestCase):
 213     """
 214     Test for the CreateSecurityGroup class defined in create_security_group.py
 215     """
 216
 217     def setUp(self):
 218         """
 219         Instantiates the CreateSecurityGroup object that is responsible for downloading and creating an OS image file
 220         within OpenStack
 221         """
 222         guid = self.__class__.__name__ + '-' + str(uuid.uuid4())
 223         self.flavor_name = guid + 'name'
 224
 225         self.nova = nova_utils.nova_client(self.os_creds)
 226
 227         # Initialize for cleanup
 228         self.flavor_creator = None
 229
 230     def tearDown(self):
 231         """
 232         Cleans the image and downloaded image file
 233         """
 234         if self.flavor_creator:
 235             self.flavor_creator.clean()
 236
 237     def test_create_flavor(self):
 238         """
 239         Tests the creation of an OpenStack flavor.
 240         """
 241         # Create Flavor
 242         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 243         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 244         flavor = self.flavor_creator.create()
 245         self.assertTrue(validate_flavor(flavor_settings, flavor))
 246
 247     def test_create_flavor_existing(self):
 248         """
 249         Tests the creation of an OpenStack flavor then starts another creator to ensure it has not been done twice.
 250         """
 251         # Create Flavor
 252         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 253         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 254         flavor = self.flavor_creator.create()
 255         self.assertTrue(validate_flavor(flavor_settings, flavor))
 256
 257         flavor_creator_2 = OpenStackFlavor(self.os_creds, flavor_settings)
 258         flavor2 = flavor_creator_2.create()
 259
 260         self.assertEquals(flavor.id, flavor2.id)
 261
 262     def test_create_clean_flavor(self):
 263         """
 264         Tests the creation and cleanup of an OpenStack flavor.
 265         """
 266         # Create Flavor
 267         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 268         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 269         flavor = self.flavor_creator.create()
 270         self.assertTrue(validate_flavor(flavor_settings, flavor))
 271
 272         # Clean Flavor
 273         self.flavor_creator.clean()
 274
 275         self.assertIsNone(self.flavor_creator.get_flavor())
 276         self.assertIsNone(nova_utils.get_flavor_by_name(self.nova, flavor_settings.name))
 277
 278     def test_create_delete_flavor(self):
 279         """
 280         Tests the creation of an OpenStack Security Group, the deletion, then cleanup to ensure clean() does not
 281         raise any exceptions.
 282         """
 283         # Create Flavor
 284         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 285         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 286         flavor = self.flavor_creator.create()
 287         self.assertTrue(validate_flavor(flavor_settings, flavor))
 288
 289         # Delete Flavor
 290         nova_utils.delete_flavor(self.nova, flavor)
 291         self.assertIsNone(nova_utils.get_flavor_by_name(self.nova, flavor_settings.name))
 292
 293         # Attempt to cleanup
 294         self.flavor_creator.clean()
 295
 296         self.assertIsNone(self.flavor_creator.get_flavor())
 297
 298         # TODO - Add more tests to exercise all configuration options
 299
 300
 301 def validate_flavor(flavor_settings, flavor):
 302     """
 303     Validates the flavor_settings against the OpenStack flavor object
 304     :param flavor_settings: the settings used to create the flavor
 305     :param flavor: the OpenStack flavor object
 306     """
 307     return flavor is not None \
 308            and flavor_settings.name == flavor.name \
 309            and flavor_settings.ram == flavor.ram \
 310            and flavor_settings.disk == flavor.disk \
 311            and flavor_settings.vcpus == flavor.vcpus