1 # You may obtain a copy of the License at:
3 # http://www.apache.org/licenses/LICENSE-2.0
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.
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
17 __author__ = 'spisarski'
20 class FlavorSettingsUnitTests(unittest.TestCase):
22 Tests the construction of the FlavorSettings class
25 def test_no_params(self):
26 with self.assertRaises(Exception):
29 def test_empty_config(self):
30 with self.assertRaises(Exception):
31 FlavorSettings(config=dict())
33 def test_name_only(self):
34 with self.assertRaises(Exception):
35 FlavorSettings(name='foo')
37 def test_config_with_name_only(self):
38 with self.assertRaises(Exception):
39 FlavorSettings(config={'name': 'foo'})
41 def test_name_ram_only(self):
42 with self.assertRaises(Exception):
43 FlavorSettings(name='foo', ram=1)
45 def test_config_with_name_ram_only(self):
46 with self.assertRaises(Exception):
47 FlavorSettings(config={'name': 'foo', 'ram': 1})
49 def test_name_ram_disk_only(self):
50 with self.assertRaises(Exception):
51 FlavorSettings(name='foo', ram=1, disk=1)
53 def test_config_with_name_ram_disk_only(self):
54 with self.assertRaises(Exception):
55 FlavorSettings(config={'name': 'foo', 'ram': 1, 'disk': 1})
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,
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})
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)
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})
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,
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})
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)
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})
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,
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})
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,
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})
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)
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})
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,
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})
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)
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})
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)
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})
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')
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'})
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)
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)
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)
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)
219 class CreateFlavorTests(OSComponentTestCase):
221 Test for the CreateSecurityGroup class defined in create_security_group.py
226 Instantiates the CreateSecurityGroup object that is responsible for downloading and creating an OS image file
229 guid = self.__class__.__name__ + '-' + str(uuid.uuid4())
230 self.flavor_name = guid + 'name'
232 self.nova = nova_utils.nova_client(self.os_creds)
234 # Initialize for cleanup
235 self.flavor_creator = None
239 Cleans the image and downloaded image file
241 if self.flavor_creator:
242 self.flavor_creator.clean()
244 def test_create_flavor(self):
246 Tests the creation of an OpenStack 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))
254 def test_create_flavor_existing(self):
256 Tests the creation of an OpenStack flavor then starts another creator to ensure it has not been done twice.
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))
264 flavor_creator_2 = OpenStackFlavor(self.os_creds, flavor_settings)
265 flavor2 = flavor_creator_2.create()
267 self.assertEquals(flavor.id, flavor2.id)
269 def test_create_clean_flavor(self):
271 Tests the creation and cleanup of an OpenStack 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))
280 self.flavor_creator.clean()
282 self.assertIsNone(self.flavor_creator.get_flavor())
283 self.assertIsNone(nova_utils.get_flavor_by_name(self.nova, flavor_settings.name))
285 def test_create_delete_flavor(self):
287 Tests the creation of an OpenStack Security Group, the deletion, then cleanup to ensure clean() does not
288 raise any exceptions.
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))
297 nova_utils.delete_flavor(self.nova, flavor)
298 self.assertIsNone(nova_utils.get_flavor_by_name(self.nova, flavor_settings.name))
301 self.flavor_creator.clean()
303 self.assertIsNone(self.flavor_creator.get_flavor())
305 # TODO - Add more tests to exercise all configuration options
308 def validate_flavor(flavor_settings, flavor):
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
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