def test_cleanup(self, mock_clients, mock_wrap, api_versions=None):
tenants_count = 2
users_per_tenant = 5
images_per_tenant = 5
tenants = self._gen_tenants(tenants_count)
users = []
created_images = []
for tenant_id in tenants:
for i in range(users_per_tenant):
users.append({"id": i, "tenant_id": tenant_id,
"credential": mock.MagicMock()})
tenants[tenant_id].setdefault("images", [])
for j in range(images_per_tenant):
image = mock.Mock()
created_images.append(image)
tenants[tenant_id]["images"].append(image)
self.context.update({
"config": {
"users": {
"tenants": tenants_count,
"users_per_tenant": users_per_tenant,
"concurrent": 10,
},
"images": {
"image_url": "mock_url",
"image_type": "qcow2",
"image_container": "bare",
"images_per_tenant": images_per_tenant,
"image_name": "some_name",
"min_ram": 128,
"min_disk": 1,
}
},
"admin": {
"credential": mock.MagicMock()
},
"users": users,
"tenants": tenants
})
if api_versions:
self.context["config"]["api_versions"] = api_versions
images_ctx = images.ImageGenerator(self.context)
images_ctx.cleanup()
glance_client = mock_clients.return_value.glance.return_value
glance_client.images.get.assert_has_calls([mock.call(i)
for i in created_images])
wrapper_calls = []
wrapper_calls.extend([mock.call(mock_clients.return_value.glance,
images_ctx)] * tenants_count)
wrapper_calls.extend(
[mock.call().delete_image(glance_client.images.get.return_value)] *
len(created_images))
mock_wrap.assert_has_calls(wrapper_calls, any_order=True)
mock_clients.assert_has_calls(
[mock.call(mock.ANY, api_info=api_versions)] * tenants_count,
any_order=True)
def test_cleanup(self, mock_cleanup, api_versions=None):
images_per_tenant = 5
tenants = self._gen_tenants(self.tenants_num)
users = []
created_images = []
for tenant_id in tenants:
for i in range(self.users_per_tenant):
users.append({
"id": i,
"tenant_id": tenant_id,
"credential": mock.MagicMock()
})
tenants[tenant_id].setdefault("images", [])
for j in range(images_per_tenant):
image = mock.Mock()
created_images.append(image)
tenants[tenant_id]["images"].append(image)
self.context.update({
"config": {
"users": {
"tenants": self.tenants_num,
"users_per_tenant": self.users_per_tenant,
"concurrent": 10,
},
"images": {
"image_url": "mock_url",
"image_type": "qcow2",
"image_container": "bare",
"images_per_tenant": images_per_tenant,
"image_name": "some_name",
"min_ram": 128,
"min_disk": 1,
}
},
"admin": {
"credential": mock.MagicMock()
},
"users": users,
"tenants": tenants
})
images_ctx = images.ImageGenerator(self.context)
images_ctx.cleanup()
mock_cleanup.assert_called_once_with(
names=["glance.images"],
users=self.context["users"],
api_versions=self.context["config"].get("api_versions"),
superclass=images_ctx.__class__,
task_id=self.context["owner_id"])
def test_setup(self, mock_glance_scenario__create_image):
tenants_count = 2
users_per_tenant = 5
images_per_tenant = 5
tenants = self._gen_tenants(tenants_count)
users = []
for id_ in tenants:
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": id_,
"endpoint": mock.MagicMock()
})
real_context = {
"config": {
"users": {
"tenants": tenants_count,
"users_per_tenant": users_per_tenant,
"concurrent": 10,
},
"images": {
"image_url": "mock_url",
"image_type": "qcow2",
"image_container": "bare",
"images_per_tenant": images_per_tenant,
"image_name": "some_name",
"min_ram": 128,
"min_disk": 1,
}
},
"admin": {
"endpoint": mock.MagicMock()
},
"task": mock.MagicMock(),
"users": users,
"tenants": tenants
}
new_context = copy.deepcopy(real_context)
for id_ in new_context["tenants"].keys():
new_context["tenants"][id_].setdefault("images", [])
for j in range(images_per_tenant):
new_context["tenants"][id_]["images"].append("uuid")
images_ctx = images.ImageGenerator(real_context)
images_ctx.setup()
self.assertEqual(new_context, real_context)
def test_cleanup(self, mock_cleanup):
tenants_count = 2
users_per_tenant = 5
images_per_tenant = 5
tenants = self._gen_tenants(tenants_count)
users = []
for id_ in tenants:
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": id_,
"endpoint": "endpoint"
})
tenants[id_].setdefault("images", [])
for j in range(images_per_tenant):
tenants[id_]["images"].append("uuid")
context = {
"config": {
"users": {
"tenants": 2,
"users_per_tenant": 5,
"concurrent": 10,
},
"images": {
"image_url": "mock_url",
"image_type": "qcow2",
"image_container": "bare",
"images_per_tenant": 5,
"image_name": "some_name",
"min_ram": 128,
"min_disk": 1,
}
},
"admin": {
"endpoint": mock.MagicMock()
},
"task": mock.MagicMock(),
"users": users,
"tenants": tenants
}
images_ctx = images.ImageGenerator(context)
images_ctx.cleanup()
mock_cleanup.assert_called_once_with(names=["glance.images"],
users=context["users"])
def test_cleanup(self, mock_cleanup, admin=None, api_versions=None):
images_per_tenant = 5
tenants = self._gen_tenants(self.tenants_num)
users = []
created_images = []
for tenant_id in tenants:
for i in range(self.users_per_tenant):
users.append({
"id": i,
"tenant_id": tenant_id,
"credential": mock.MagicMock()
})
tenants[tenant_id].setdefault("images", [])
for j in range(images_per_tenant):
image = mock.Mock()
created_images.append(image)
tenants[tenant_id]["images"].append(image)
self.context.update({
"config": {
"users": {
"tenants": self.tenants_num,
"users_per_tenant": self.users_per_tenant,
"concurrent": 10,
},
"images": {},
"api_versions": api_versions
},
"users": mock.Mock()
})
if admin:
self.context["admin"] = {"credential": mock.MagicMock()}
else:
# ensure that there is no admin
self.context.pop("admin")
images_ctx = images.ImageGenerator(self.context)
images_ctx.cleanup()
mock_cleanup.assert_called_once_with(
names=["glance.images", "cinder.image_volumes_cache"],
admin=self.context.get("admin"),
admin_required=None if admin else False,
users=self.context["users"],
api_versions=api_versions,
superclass=images_ctx.__class__,
task_id=self.context["owner_id"])
def test_cleanup_for_predefined_name(self, mock_cleanup,
mock_make_name_matcher):
self.context.update({
"config": {
"images": {
"image_name": "foo"
}
},
"users": mock.Mock()
})
images_ctx = images.ImageGenerator(self.context)
images_ctx.cleanup()
mock_cleanup.assert_called_once_with(
names=["glance.images", "cinder.image_volumes_cache"],
admin=self.context.get("admin"),
admin_required=None,
users=self.context["users"],
api_versions=None,
superclass=mock_make_name_matcher.return_value,
task_id=self.context["owner_id"])
def test_setup(self,
mock_clients,
mock_wrap,
image_container="bare",
image_type="qcow2",
image_url="http://example.com/fake/url",
tenants=1,
users_per_tenant=1,
images_per_tenant=1,
image_name=None,
min_ram=None,
min_disk=None,
image_args=None,
api_versions=None):
tenant_data = self._gen_tenants(tenants)
users = []
for tenant_id in tenant_data:
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": tenant_id,
"credential": mock.MagicMock()
})
self.context.update({
"config": {
"users": {
"tenants": tenants,
"users_per_tenant": users_per_tenant,
"concurrent": 10,
},
"images": {
"image_url": image_url,
"image_type": image_type,
"image_container": image_container,
"images_per_tenant": images_per_tenant,
}
},
"admin": {
"credential": mock.MagicMock()
},
"users": users,
"tenants": tenant_data
})
if api_versions:
self.context["config"]["api_versions"] = api_versions
expected_image_args = {}
if image_args is not None:
self.context["config"]["images"]["image_args"] = image_args
expected_image_args.update(image_args)
if image_name is not None:
self.context["config"]["images"]["image_name"] = image_name
if min_ram is not None:
self.context["config"]["images"]["min_ram"] = min_ram
expected_image_args["min_ram"] = min_ram
if min_disk is not None:
self.context["config"]["images"]["min_disk"] = min_disk
expected_image_args["min_disk"] = min_disk
wrapper = mock_wrap.return_value
new_context = copy.deepcopy(self.context)
for tenant_id in new_context["tenants"].keys():
new_context["tenants"][tenant_id]["images"] = [
wrapper.create_image.return_value.id
] * images_per_tenant
images_ctx = images.ImageGenerator(self.context)
images_ctx.setup()
self.assertEqual(new_context, self.context)
wrapper_calls = []
wrapper_calls.extend(
[mock.call(mock_clients.return_value.glance, images_ctx)] *
tenants)
wrapper_calls.extend([
mock.call().create_image(image_container,
image_url,
image_type,
name=mock.ANY,
**expected_image_args)
] * tenants * images_per_tenant)
mock_wrap.assert_has_calls(wrapper_calls, any_order=True)
if image_name:
for args in wrapper.create_image.call_args_list:
self.assertTrue(args[1]["name"].startswith(image_name))
mock_clients.assert_has_calls(
[mock.call(mock.ANY, api_info=api_versions)] * tenants)
def test_setup(self,
mock_clients,
container_format="bare",
disk_format="qcow2",
image_url="http://example.com/fake/url",
tenants=1,
users_per_tenant=1,
images_per_tenant=1,
image_name=None,
min_ram=None,
min_disk=None,
image_args={"is_public": True},
api_versions=None,
visibility="public"):
image_service = self.mock_image.return_value
tenant_data = self._gen_tenants(tenants)
users = []
for tenant_id in tenant_data:
for i in range(users_per_tenant):
users.append({
"id": i,
"tenant_id": tenant_id,
"credential": mock.MagicMock()
})
self.context.update({
"config": {
"users": {
"tenants": tenants,
"users_per_tenant": users_per_tenant,
"concurrent": 10,
},
"images": {
"image_url": image_url,
"image_type": disk_format,
"disk_format": disk_format,
"image_container": container_format,
"container_format": container_format,
"images_per_tenant": images_per_tenant,
"is_public": visibility,
"visibility": visibility,
"image_args": image_args
}
},
"admin": {
"credential": mock.MagicMock()
},
"users": users,
"tenants": tenant_data
})
if api_versions:
self.context["config"]["api_versions"] = api_versions
expected_image_args = {}
if image_args is not None:
self.context["config"]["images"]["image_args"] = image_args
expected_image_args.update(image_args)
if image_name is not None:
self.context["config"]["images"]["image_name"] = image_name
if min_ram is not None:
self.context["config"]["images"]["min_ram"] = min_ram
expected_image_args["min_ram"] = min_ram
if min_disk is not None:
self.context["config"]["images"]["min_disk"] = min_disk
expected_image_args["min_disk"] = min_disk
new_context = copy.deepcopy(self.context)
for tenant_id in new_context["tenants"].keys():
new_context["tenants"][tenant_id]["images"] = [
image_service.create_image.return_value.id
] * images_per_tenant
images_ctx = images.ImageGenerator(self.context)
images_ctx.setup()
self.assertEqual(new_context, self.context)
wrapper_calls = []
wrapper_calls.extend(
[mock.call(mock_clients.return_value.glance, images_ctx)] *
tenants)
wrapper_calls.extend([
mock.call().create_image(container_format,
image_url,
disk_format,
name=mock.ANY,
**expected_image_args)
] * tenants * images_per_tenant)
mock_clients.assert_has_calls(
[mock.call(mock.ANY, api_info=api_versions)] * tenants)
def test_setup_with_deprecated_args(self, mock_log, mock_image):
image_type = "itype"
image_container = "icontainer"
is_public = True
d_min_ram = mock.Mock()
d_min_disk = mock.Mock()
self.context.update({
"config": {
"images": {
"image_type": image_type,
"image_container": image_container,
"image_args": {
"is_public": is_public,
"min_ram": d_min_ram,
"min_disk": d_min_disk
}
}
},
"users": [{
"tenant_id": "foo-tenant",
"credential": mock.MagicMock()
}],
"tenants": {
"foo-tenant": {}
}
})
images_ctx = images.ImageGenerator(self.context)
images_ctx.setup()
mock_image.return_value.create_image.assert_called_once_with(
image_name=None,
container_format=image_container,
image_location=None,
disk_format=image_type,
visibility="public",
min_disk=d_min_disk,
min_ram=d_min_ram)
self.assertEqual([
mock.call("The 'image_type' argument is deprecated since "
"Rally 0.10.0, use disk_format arguments instead."),
mock.call("The 'image_container' argument is deprecated since "
"Rally 0.10.0; use container_format arguments instead"),
mock.call("The 'image_args' argument is deprecated since "
"Rally 0.10.0; specify exact arguments in a root "
"section of context instead.")
], mock_log.warning.call_args_list)
mock_image.return_value.create_image.reset_mock()
mock_log.warning.reset_mock()
min_ram = mock.Mock()
min_disk = mock.Mock()
visibility = "foo"
disk_format = "dformat"
container_format = "cformat"
self.context["config"]["images"].update({
"min_ram":
min_ram,
"min_disk":
min_disk,
"visibility":
visibility,
"disk_format":
disk_format,
"container_format":
container_format
})
images_ctx = images.ImageGenerator(self.context)
images_ctx.setup()
# check that deprecated arguments are not used
mock_image.return_value.create_image.assert_called_once_with(
image_name=None,
container_format=container_format,
image_location=None,
disk_format=disk_format,
visibility=visibility,
min_disk=min_disk,
min_ram=min_ram)
# No matter will be deprecated arguments used or not, if they are
# specified, warning message should be printed.
self.assertEqual([
mock.call("The 'image_type' argument is deprecated since "
"Rally 0.10.0, use disk_format arguments instead."),
mock.call("The 'image_container' argument is deprecated since "
"Rally 0.10.0; use container_format arguments instead"),
mock.call("The 'image_args' argument is deprecated since "
"Rally 0.10.0; specify exact arguments in a root "
"section of context instead.")
], mock_log.warning.call_args_list)