296 lines
8.3 KiB
Python
296 lines
8.3 KiB
Python
# Copyright 2013 Nebula Inc.
|
|
#
|
|
# 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.
|
|
|
|
import random
|
|
from unittest import mock
|
|
import uuid
|
|
|
|
from openstack.image.v2 import image
|
|
from openstack.image.v2 import member
|
|
from openstack.image.v2 import metadef_namespace
|
|
from openstack.image.v2 import metadef_resource_type
|
|
from openstack.image.v2 import service_info as _service_info
|
|
from openstack.image.v2 import task
|
|
|
|
from openstackclient.tests.unit import fakes
|
|
from openstackclient.tests.unit.identity.v3 import fakes as identity_fakes
|
|
from openstackclient.tests.unit import utils
|
|
|
|
|
|
class TestImagev2(utils.TestCommand):
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
self.app.client_manager.image = mock.Mock()
|
|
self.image_client = self.app.client_manager.image
|
|
|
|
self.app.client_manager.identity = identity_fakes.FakeIdentityv3Client(
|
|
endpoint=fakes.AUTH_URL,
|
|
token=fakes.AUTH_TOKEN,
|
|
)
|
|
|
|
|
|
def create_one_image(attrs=None):
|
|
"""Create a fake image.
|
|
|
|
:param attrs: A dictionary with all attributes of image
|
|
:type attrs: dict
|
|
:return: A fake Image object.
|
|
:rtype: `openstack.image.v2.image.Image`
|
|
"""
|
|
attrs = attrs or {}
|
|
|
|
# Set default attribute
|
|
image_info = {
|
|
'id': str(uuid.uuid4()),
|
|
'name': 'image-name' + uuid.uuid4().hex,
|
|
'owner_id': 'image-owner' + uuid.uuid4().hex,
|
|
'is_protected': bool(random.choice([0, 1])),
|
|
'visibility': random.choice(['public', 'private']),
|
|
'tags': [uuid.uuid4().hex for r in range(2)],
|
|
}
|
|
|
|
# Overwrite default attributes if there are some attributes set
|
|
image_info.update(attrs)
|
|
|
|
return image.Image(**image_info)
|
|
|
|
|
|
def create_images(attrs=None, count=2):
|
|
"""Create multiple fake images.
|
|
|
|
:param attrs: A dictionary with all attributes of image
|
|
:type attrs: dict
|
|
:param count: The number of images to be faked
|
|
:type count: int
|
|
:return: A list of fake Image objects
|
|
:rtype: list
|
|
"""
|
|
images = []
|
|
for n in range(0, count):
|
|
images.append(create_one_image(attrs))
|
|
|
|
return images
|
|
|
|
|
|
def create_one_image_member(attrs=None):
|
|
"""Create a fake image member.
|
|
|
|
:param attrs: A dictionary with all attributes of image member
|
|
:type attrs: dict
|
|
:return: A fake Member object.
|
|
:rtype: `openstack.image.v2.member.Member`
|
|
"""
|
|
attrs = attrs or {}
|
|
|
|
# Set default attribute
|
|
image_member_info = {
|
|
'member_id': 'member-id-' + uuid.uuid4().hex,
|
|
'image_id': 'image-id-' + uuid.uuid4().hex,
|
|
'status': 'pending',
|
|
}
|
|
|
|
# Overwrite default attributes if there are some attributes set
|
|
image_member_info.update(attrs)
|
|
|
|
return member.Member(**image_member_info)
|
|
|
|
|
|
def create_one_import_info(attrs=None):
|
|
"""Create a fake import info.
|
|
|
|
:param attrs: A dictionary with all attributes of import info
|
|
:type attrs: dict
|
|
:return: A fake Import object.
|
|
:rtype: `openstack.image.v2.service_info.Import`
|
|
"""
|
|
attrs = attrs or {}
|
|
|
|
import_info = {
|
|
'import-methods': {
|
|
'description': 'Import methods available.',
|
|
'type': 'array',
|
|
'value': [
|
|
'glance-direct',
|
|
'web-download',
|
|
'glance-download',
|
|
'copy-image',
|
|
],
|
|
}
|
|
}
|
|
import_info.update(attrs)
|
|
|
|
return _service_info.Import(**import_info)
|
|
|
|
|
|
def create_one_stores_info(attrs=None):
|
|
"""Create a fake stores info.
|
|
|
|
:param attrs: A dictionary with all attributes of stores
|
|
:type attrs: dict
|
|
:return: A fake Store object list.
|
|
:rtype: `openstack.image.v2.service_info.Store`
|
|
"""
|
|
attrs = attrs or {}
|
|
|
|
stores_info = {
|
|
"stores": [
|
|
{
|
|
"id": "reliable",
|
|
"description": "More expensive store with data redundancy",
|
|
},
|
|
{
|
|
"id": "fast",
|
|
"description": "Provides quick access to your image data",
|
|
"default": True,
|
|
},
|
|
{
|
|
"id": "cheap",
|
|
"description": "Less expensive store for seldom-used images",
|
|
},
|
|
]
|
|
}
|
|
stores_info.update(attrs)
|
|
|
|
return _service_info.Store(**stores_info)
|
|
|
|
|
|
def create_one_task(attrs=None):
|
|
"""Create a fake task.
|
|
|
|
:param attrs: A dictionary with all attributes of task
|
|
:type attrs: dict
|
|
:return: A fake Task object.
|
|
:rtype: `openstack.image.v2.task.Task`
|
|
"""
|
|
attrs = attrs or {}
|
|
|
|
# Set default attribute
|
|
task_info = {
|
|
'created_at': '2016-06-29T16:13:07Z',
|
|
'expires_at': '2016-07-01T16:13:07Z',
|
|
'id': str(uuid.uuid4()),
|
|
'input': {
|
|
'image_properties': {
|
|
'container_format': 'ovf',
|
|
'disk_format': 'vhd',
|
|
},
|
|
'import_from': 'https://apps.openstack.org/excellent-image',
|
|
'import_from_format': 'qcow2',
|
|
},
|
|
'message': '',
|
|
'owner': str(uuid.uuid4()),
|
|
'result': {
|
|
'image_id': str(uuid.uuid4()),
|
|
},
|
|
'schema': '/v2/schemas/task',
|
|
'status': random.choice(
|
|
[
|
|
'pending',
|
|
'processing',
|
|
'success',
|
|
'failure',
|
|
]
|
|
),
|
|
# though not documented, the API only allows 'import'
|
|
# https://github.com/openstack/glance/blob/24.0.0/glance/api/v2/tasks.py#L186-L190
|
|
'type': 'import',
|
|
'updated_at': '2016-06-29T16:13:07Z',
|
|
}
|
|
|
|
# Overwrite default attributes if there are some attributes set
|
|
task_info.update(attrs)
|
|
|
|
return task.Task(**task_info)
|
|
|
|
|
|
def create_tasks(attrs=None, count=2):
|
|
"""Create multiple fake tasks.
|
|
|
|
:param attrs: A dictionary with all attributes of Task
|
|
:type attrs: dict
|
|
:param count: The number of tasks to be faked
|
|
:type count: int
|
|
:return: A list of fake Task objects
|
|
:rtype: list
|
|
"""
|
|
tasks = []
|
|
for n in range(0, count):
|
|
tasks.append(create_one_task(attrs))
|
|
|
|
return tasks
|
|
|
|
|
|
def create_one_metadef_namespace(attrs=None):
|
|
"""Create a fake MetadefNamespace member.
|
|
|
|
:param attrs: A dictionary with all attributes of metadef_namespace member
|
|
:type attrs: dict
|
|
:return: a list of MetadefNamespace objects
|
|
:rtype: list of `metadef_namespace.MetadefNamespace`
|
|
"""
|
|
attrs = attrs or {}
|
|
|
|
metadef_namespace_list = {
|
|
'created_at': '2022-08-17T11:30:22Z',
|
|
'display_name': 'Flavor Quota',
|
|
'namespace': 'OS::Compute::Quota',
|
|
'owner': 'admin',
|
|
# 'resource_type_associations': ['OS::Nova::Flavor'],
|
|
# The part that receives the list type factor is not implemented.
|
|
'visibility': 'public',
|
|
}
|
|
|
|
# Overwrite default attributes if there are some attributes set
|
|
metadef_namespace_list.update(attrs)
|
|
return metadef_namespace.MetadefNamespace(**metadef_namespace_list)
|
|
|
|
|
|
def create_one_resource_type(attrs=None):
|
|
"""Create a fake MetadefResourceType member.
|
|
|
|
:param attrs: A dictionary with all attributes of
|
|
metadef_resource_type member
|
|
:type attrs: dict
|
|
:return: a fake MetadefResourceType object
|
|
:rtype: A `metadef_resource_type.MetadefResourceType`
|
|
"""
|
|
attrs = attrs or {}
|
|
|
|
metadef_resource_type_info = {
|
|
'name': 'OS::Compute::Quota',
|
|
'properties_target': 'image',
|
|
}
|
|
|
|
metadef_resource_type_info.update(attrs)
|
|
return metadef_resource_type.MetadefResourceType(
|
|
**metadef_resource_type_info
|
|
)
|
|
|
|
|
|
def create_resource_types(attrs=None, count=2):
|
|
"""Create multiple fake resource types.
|
|
|
|
:param attrs: A dictionary with all attributes of
|
|
metadef_resource_type member
|
|
:type attrs: dict
|
|
:return: A list of fake MetadefResourceType objects
|
|
:rtype: list
|
|
"""
|
|
metadef_resource_types = []
|
|
for n in range(0, count):
|
|
metadef_resource_types.append(create_one_resource_type(attrs))
|
|
|
|
return metadef_resource_types
|