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         self.assertEquals(None, settings.metadata)
 173
 174     def test_config_with_name_ram_disk_vcpus_only(self):
 175         settings = FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 2, 'vcpus': 3})
 176         self.assertEquals('foo', settings.name)
 177         self.assertEquals('auto', settings.flavor_id)
 178         self.assertEquals(1, settings.ram)
 179         self.assertEquals(2, settings.disk)
 180         self.assertEquals(3, settings.vcpus)
 181         self.assertEquals(0, settings.ephemeral)
 182         self.assertEquals(0, settings.swap)
 183         self.assertEquals(1.0, settings.rxtx_factor)
 184         self.assertEquals(True, settings.is_public)
 185         self.assertEquals(None, settings.metadata)
 186
 187     def test_all(self):
 188         metadata = {'foo': 'bar'}
 189         settings = FlavorSettings(name='foo', flavor_id='bar', ram=1, disk=2, vcpus=3, ephemeral=4, swap=5,
 190                                   rxtx_factor=6.0, is_public=False, metadata=metadata)
 191         self.assertEquals('foo', settings.name)
 192         self.assertEquals('bar', settings.flavor_id)
 193         self.assertEquals(1, settings.ram)
 194         self.assertEquals(2, settings.disk)
 195         self.assertEquals(3, settings.vcpus)
 196         self.assertEquals(4, settings.ephemeral)
 197         self.assertEquals(5, settings.swap)
 198         self.assertEquals(6.0, settings.rxtx_factor)
 199         self.assertEquals(False, settings.is_public)
 200         self.assertEquals(metadata, settings.metadata)
 201
 202     def test_config_all(self):
 203         metadata = {'foo': 'bar'}
 204         settings = FlavorSettings(config={'name': 'foo', 'flavor_id': 'bar', 'ram': 1, 'disk': 2, 'vcpus': 3,
 205                                           'ephemeral': 4, 'swap': 5, 'rxtx_factor': 6.0, 'is_public': False,
 206                                           'metadata': metadata})
 207         self.assertEquals('foo', settings.name)
 208         self.assertEquals('bar', settings.flavor_id)
 209         self.assertEquals(1, settings.ram)
 210         self.assertEquals(2, settings.disk)
 211         self.assertEquals(3, settings.vcpus)
 212         self.assertEquals(4, settings.ephemeral)
 213         self.assertEquals(5, settings.swap)
 214         self.assertEquals(6.0, settings.rxtx_factor)
 215         self.assertEquals(False, settings.is_public)
 216         self.assertEquals(metadata, settings.metadata)
 217
 218
 219 class CreateFlavorTests(OSComponentTestCase):
 220     """
 221     Test for the CreateSecurityGroup class defined in create_security_group.py
 222     """
 223
 224     def setUp(self):
 225         """
 226         Instantiates the CreateSecurityGroup object that is responsible for downloading and creating an OS image file
 227         within OpenStack
 228         """
 229         guid = self.__class__.__name__ + '-' + str(uuid.uuid4())
 230         self.flavor_name = guid + 'name'
 231
 232         self.nova = nova_utils.nova_client(self.os_creds)
 233
 234         # Initialize for cleanup
 235         self.flavor_creator = None
 236
 237     def tearDown(self):
 238         """
 239         Cleans the image and downloaded image file
 240         """
 241         if self.flavor_creator:
 242             self.flavor_creator.clean()
 243
 244     def test_create_flavor(self):
 245         """
 246         Tests the creation of an OpenStack flavor.
 247         """
 248         # Create Flavor
 249         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 250         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 251         flavor = self.flavor_creator.create()
 252         self.assertTrue(validate_flavor(flavor_settings, flavor))
 253
 254     def test_create_flavor_existing(self):
 255         """
 256         Tests the creation of an OpenStack flavor then starts another creator to ensure it has not been done twice.
 257         """
 258         # Create Flavor
 259         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 260         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 261         flavor = self.flavor_creator.create()
 262         self.assertTrue(validate_flavor(flavor_settings, flavor))
 263
 264         flavor_creator_2 = OpenStackFlavor(self.os_creds, flavor_settings)
 265         flavor2 = flavor_creator_2.create()
 266
 267         self.assertEquals(flavor.id, flavor2.id)
 268
 269     def test_create_clean_flavor(self):
 270         """
 271         Tests the creation and cleanup of an OpenStack flavor.
 272         """
 273         # Create Flavor
 274         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 275         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 276         flavor = self.flavor_creator.create()
 277         self.assertTrue(validate_flavor(flavor_settings, flavor))
 278
 279         # Clean Flavor
 280         self.flavor_creator.clean()
 281
 282         self.assertIsNone(self.flavor_creator.get_flavor())
 283         self.assertIsNone(nova_utils.get_flavor_by_name(self.nova, flavor_settings.name))
 284
 285     def test_create_delete_flavor(self):
 286         """
 287         Tests the creation of an OpenStack Security Group, the deletion, then cleanup to ensure clean() does not
 288         raise any exceptions.
 289         """
 290         # Create Flavor
 291         flavor_settings = FlavorSettings(name=self.flavor_name, ram=1, disk=1, vcpus=1)
 292         self.flavor_creator = OpenStackFlavor(self.os_creds, flavor_settings)
 293         flavor = self.flavor_creator.create()
 294         self.assertTrue(validate_flavor(flavor_settings, flavor))
 295
 296         # Delete Flavor
 297         nova_utils.delete_flavor(self.nova, flavor)
 298         self.assertIsNone(nova_utils.get_flavor_by_name(self.nova, flavor_settings.name))
 299
 300         # Attempt to cleanup
 301         self.flavor_creator.clean()
 302
 303         self.assertIsNone(self.flavor_creator.get_flavor())
 304
 305         # TODO - Add more tests to exercise all configuration options
 306
 307
 308 def validate_flavor(flavor_settings, flavor):
 309     """
 310     Validates the flavor_settings against the OpenStack flavor object
 311     :param flavor_settings: the settings used to create the flavor
 312     :param flavor: the OpenStack flavor object
 313     """
 314     return flavor is not None \
 315            and flavor_settings.name == flavor.name \
 316            and flavor_settings.ram == flavor.ram \
 317            and flavor_settings.disk == flavor.disk \
 318            and flavor_settings.vcpus == flavor.vcpus