glance/rally-jobs/plugins/plugin_sample.py
Boris Pavlovic 903075b3f9 Update rally-jobs files
* Rename rally-scenarios directory to rally-jobs. Because actually
  this directory contains files related to job and not scenarios.

* Upadte main README file

* Add readme files to rally-jobs/extra and rally-jobs/plugins

Change-Id: I111c3c3f86ab4886d3a7b66c7bf7c050c1d9ca1a
2014-11-25 01:18:12 +04:00

92 lines
3.1 KiB
Python

# Copyright 2014 Mirantis Inc.
# All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
""" Sample of plugin for Glance.
For more Glance related benchmarks take a look here:
github.com/stackforge/rally/blob/master/rally/benchmark/scenarios/glance/
About plugins: https://rally.readthedocs.org/en/latest/plugins.html
Rally concepts https://wiki.openstack.org/wiki/Rally/Concepts
"""
import os
from rally.benchmark.scenarios import base
from rally.benchmark import utils as bench_utils
class GlancePlugin(base.Scenario):
@base.atomic_action_timer("glance.create_image_label")
def _create_image(self, image_name, container_format,
image_location, disk_format, **kwargs):
"""Create a new image.
:param image_name: String used to name the image
:param container_format: Container format of image.
Acceptable formats: ami, ari, aki, bare, and ovf.
:param image_location: image file location used to upload
:param disk_format: Disk format of image. Acceptable formats:
ami, ari, aki, vhd, vmdk, raw, qcow2, vdi, and iso.
:param **kwargs: optional parameters to create image
returns: object of image
"""
kw = {
"name": image_name,
"container_format": container_format,
"disk_format": disk_format,
}
kw.update(kwargs)
try:
if os.path.isfile(os.path.expanduser(image_location)):
kw["data"] = open(os.path.expanduser(image_location))
else:
kw["copy_from"] = image_location
image = self.clients("glance").images.create(**kw)
image = bench_utils.wait_for(
image,
is_ready=bench_utils.resource_is("active"),
update_resource=bench_utils.get_from_manager(),
timeout=100,
check_interval=0.5)
finally:
if "data" in kw:
kw["data"].close()
return image
@base.atomic_action_timer("glance.list_images_label")
def _list_images(self):
return list(self.clients("glance").images.list())
@base.scenario(context={"cleanup": ["glance"]})
def create_and_list(self, container_format,
image_location, disk_format, **kwargs):
self._create_image(self._generate_random_name(),
container_format,
image_location,
disk_format,
**kwargs)
self._list_images()