add image code

JIRA: DOCTOR-93

Change-Id: I6551d5e31799b967ce442cb05a664c0d64a3dcf8
Signed-off-by: dongwenjuan <dong.wenjuan@zte.com.cn>

1 files changed, 77 insertions, 0 deletions

diff --git a/tests/image.py b/tests/image.py
new file mode 100644
index 00000000..0b4a3d72
--- /dev/null
+++ b/tests/image.py
@@ -0,0 +1,77 @@
+##############################################################################
+# Copyright (c) 2017 ZTE Corporation and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+import os
+import urllib2
+
+from oslo_config import cfg
+
+from identity_auth import get_session
+from os_clients import glance_client
+import logger as doctor_log
+
+IMAGE_OPTS = [
+    cfg.StrOpt('name',
+               default=os.environ.get('IMAGE_NAME', 'cirros'),
+               help='the name of test image',
+               required=True),
+    cfg.StrOpt('format',
+               default='qcow2',
+               help='the format of test image',
+               required=True),
+    cfg.StrOpt('file_name',
+               default='cirros.img',
+               help='the name of image file',
+               required=True),
+    cfg.StrOpt('url',
+               default='https://launchpad.net/cirros/trunk/0.3.0/+download/cirros-0.3.0-x86_64-disk.img',
+               help='the url where to get the image',
+               required=True),
+]
+
+LOG = doctor_log.Logger('doctor').getLogger()
+
+
+class Image(object):
+
+    def __init__(self, conf):
+        self.conf = conf
+        self.glance = \
+            glance_client(conf.os_clients.glance_version,
+                          get_session())
+        self.use_existing_image = False
+        self.image = None
+
+    def create(self):
+        LOG.info('image create start......')
+
+        images = {image.name: image for image in self.glance.images.list()}
+        if self.conf.image.name not in images:
+            if not os.path.exists(self.conf.image.file_name):
+                resp = urllib2.urlopen(self.conf.image.url)
+                with open(self.conf.image.file_name, "wb") as file:
+                    file.write(resp.read())
+            self.image = self.glance.images.create(name=self.conf.image.name,
+                                                   disk_format=self.conf.image.format,
+                                                   container_format="bare",
+                                                   visibility="public")
+            self.glance.images.upload(self.image['id'],
+                                      open(self.conf.image.file_name, 'rb'))
+        else:
+            self.use_existing_image = True
+            self.image = images[self.conf.image.name]
+
+        LOG.info('image create end......')
+
+    def delete(self):
+        LOG.info('image delete start.......')
+
+        if not self.use_existing_image and self.image:
+            self.glance.images.delete(self.image['id'])
+
+        LOG.info('image delete end.......')