def authenticate(self, authentication=None):
"""
:param authentication:
:return:
"""
if self.__cyclades is not None:
return True
try:
authcl = AstakosClient(authentication['URL'], authentication['TOKEN'])
authcl.authenticate()
self.__cyclades = CycladesClient(authcl.get_service_endpoints('compute')['publicURL'],
authentication['TOKEN'])
self.__network_client = CycladesNetworkClient(authcl.get_service_endpoints('network')['publicURL'],
authentication['TOKEN'])
except ClientError:
stderr.write('Connector initialization failed')
return False
return True
def authenticate(self, authentication=None):
"""
:param authentication:
:return:
"""
if self.__cyclades is not None:
return True
try:
authcl = AstakosClient(authentication['URL'],
authentication['TOKEN'])
authcl.authenticate()
self.__cyclades = CycladesClient(
authcl.get_service_endpoints('compute')['publicURL'],
authentication['TOKEN'])
self.__network_client = CycladesNetworkClient(
authcl.get_service_endpoints('network')['publicURL'],
authentication['TOKEN'])
except ClientError:
stderr.write('Connector initialization failed')
return False
return True
class Image(livetest.Generic):
def setUp(self):
self.now = time.mktime(time.gmtime())
self.cloud = 'cloud.%s' % self['testcloud']
aurl, self.token = self[self.cloud, 'url'], self[self.cloud, 'token']
self.auth_base = AstakosCachedClient(aurl, self.token)
self.imgname = 'img_%s' % self.now
url = self.auth_base.get_service_endpoints('image')['publicURL']
self.token = self.auth_base.token
self.client = ImageClient(url, self.token)
cyclades_url = self.auth_base.get_service_endpoints(
'compute')['publicURL']
self.cyclades = CycladesClient(cyclades_url, self.token)
self._imglist = {}
self._imgdetails = {}
def test_000(self):
self._prepare_img()
super(self.__class__, self).test_000()
def _prepare_img(self):
f = open(self['image', 'local_path'], 'rb')
(token, uuid) = (self.token, self.auth_base.user_term('id'))
purl = self.auth_base.get_service_endpoints(
'object-store')['publicURL']
from kamaki.clients.pithos import PithosClient
self.pithcli = PithosClient(purl, token, uuid)
cont = 'cont_%s' % self.now
self.pithcli.container = cont
self.obj = 'obj_%s' % self.now
print('\t- Create container %s on Pithos server' % cont)
self.pithcli.container_put()
self.location = 'pithos://%s/%s/%s' % (uuid, cont, self.obj)
print('\t- Upload an image at %s...\n' % self.location)
self.pithcli.upload_object(self.obj, f)
print('\t- ok')
f.close()
r = self.client.register(
self.imgname, self.location, params=dict(is_public=True))
self._imglist[self.imgname] = dict(
name=r['name'], id=r['id'])
self._imgdetails[self.imgname] = r
def tearDown(self):
for img in self._imglist.values():
print('\tDeleting image %s' % img['id'])
self.cyclades.delete_image(img['id'])
if hasattr(self, 'pithcli'):
print('\tDeleting container %s' % self.pithcli.container)
try:
self.pithcli.del_container(delimiter='/')
self.pithcli.purge_container()
except ClientError:
pass
def _get_img_by_name(self, name):
r = self.cyclades.list_images()
for img in r:
if img['name'] == name:
return img
return None
def test_list_public(self):
"""Test list_public"""
self._test_list_public()
def _test_list_public(self):
r = self.client.list_public()
r0 = self.client.list_public(order='-')
self.assertTrue(len(r) > 0)
for img in r:
for term in (
'status',
'name',
'container_format',
'disk_format',
'id',
'size'):
self.assertTrue(term in img)
self.assertTrue(r, r0)
r0.reverse()
for i, img in enumerate(r):
self.assert_dicts_are_equal(img, r0[i])
r1 = self.client.list_public(detail=True)
for img in r1:
for term in (
'status',
'name',
'checksum',
'created_at',
'disk_format',
'updated_at',
'id',
'location',
'container_format',
'owner',
'is_public',
'deleted_at',
'properties',
'size'):
self.assertTrue(term in img)
if len(img['properties']):
for interm in ('osfamily', 'root_partition'):
self.assertTrue(interm in img['properties'])
size_max = 1000000000000
r2 = self.client.list_public(filters=dict(size_max=size_max))
self.assertTrue(len(r2) <= len(r))
for img in r2:
self.assertTrue(int(img['size']) <= size_max)
def test_get_meta(self):
"""Test get_meta"""
self._test_get_meta()
def _test_get_meta(self):
r = self.client.get_meta(self['image', 'id'])
self.assertEqual(r['id'], self['image', 'id'])
for term in (
'status',
'name',
'checksum',
'updated-at',
'created-at',
'deleted-at',
'location',
'is-public',
'owner',
'disk-format',
'size',
'container-format'):
self.assertTrue(term in r)
for interm in (
'OSFAMILY',
'USERS',
'ROOT_PARTITION',
'OS',
'DESCRIPTION'):
self.assertTrue(interm in r['properties'])
def test_register(self):
"""Test register"""
self._prepare_img()
self._test_register()
def _test_register(self):
self.assertTrue(self._imglist)
for img in self._imglist.values():
self.assertTrue(img is not None)
r = set(self._imgdetails[img['name']].keys())
self.assertTrue(r.issubset(IMGMETA.union(['properties'])))
def test_unregister(self):
"""Test unregister"""
self._prepare_img()
self._test_unregister()
def _test_unregister(self):
try:
for img in self._imglist.values():
self.client.unregister(img['id'])
self._prepare_img()
break
except ClientError as ce:
if ce.status in (405,):
print 'IMAGE UNREGISTER is not supported by server: %s' % ce
else:
raise
def test_set_members(self):
"""Test set_members"""
self._prepare_img()
self._test_set_members()
def _test_set_members(self):
members = ['*****@*****.**' % self.now]
for img in self._imglist.values():
self.client.set_members(img['id'], members)
r = self.client.list_members(img['id'])
self.assertEqual(r[0]['member_id'], members[0])
def test_list_members(self):
"""Test list_members"""
self._test_list_members()
def _test_list_members(self):
self._test_set_members()
def test_remove_members(self):
"""Test remove_members - NO CHECK"""
self._prepare_img()
self._test_remove_members()
def _test_remove_members(self):
return
members = ['*****@*****.**' % self.now, '*****@*****.**' % self.now]
for img in self._imglist.values():
self.client.set_members(img['id'], members)
r = self.client.list_members(img['id'])
self.assertTrue(len(r) > 1)
self.client.remove_member(img['id'], members[0])
r0 = self.client.list_members(img['id'])
self.assertEqual(len(r), 1 + len(r0))
self.assertEqual(r0[0]['member_id'], members[1])
def test_list_shared(self):
"""Test list_shared - NOT CHECKED"""
self._test_list_shared()
def _test_list_shared(self):
#No way to test this, if I dont have member images
pass
class SynnefoCI(object):
"""SynnefoCI python class"""
def __init__(self, config_file=None, build_id=None, cloud=None):
""" Initialize SynnefoCI python class
Setup logger, local_dir, config and kamaki
"""
# Setup logger
self.logger = logging.getLogger('synnefo-ci')
self.logger.setLevel(logging.DEBUG)
handler1 = logging.StreamHandler(sys.stdout)
handler1.setLevel(logging.DEBUG)
handler1.addFilter(_InfoFilter())
handler1.setFormatter(_MyFormatter())
handler2 = logging.StreamHandler(sys.stderr)
handler2.setLevel(logging.WARNING)
handler2.setFormatter(_MyFormatter())
self.logger.addHandler(handler1)
self.logger.addHandler(handler2)
# Get our local dir
self.ci_dir = os.path.dirname(os.path.abspath(__file__))
self.repo_dir = os.path.dirname(self.ci_dir)
# Read config file
if config_file is None:
config_file = os.path.join(self.ci_dir, DEFAULT_CONFIG_FILE)
config_file = os.path.abspath(config_file)
self.config = ConfigParser()
self.config.optionxform = str
self.config.read(config_file)
# Read temporary_config file
self.temp_config_file = \
os.path.expanduser(self.config.get('Global', 'temporary_config'))
self.temp_config = ConfigParser()
self.temp_config.optionxform = str
self.temp_config.read(self.temp_config_file)
self.build_id = build_id
self.logger.info("Will use \"%s\" as build id" % _green(self.build_id))
# Set kamaki cloud
if cloud is not None:
self.kamaki_cloud = cloud
elif self.config.has_option("Deployment", "kamaki_cloud"):
kamaki_cloud = self.config.get("Deployment", "kamaki_cloud")
if kamaki_cloud == "":
self.kamaki_cloud = None
else:
self.kamaki_cloud = None
# Initialize variables
self.fabric_installed = False
self.kamaki_installed = False
self.cyclades_client = None
self.compute_client = None
self.image_client = None
self.astakos_client = None
def setup_kamaki(self):
"""Initialize kamaki
Setup cyclades_client, image_client and compute_client
"""
config = kamaki_config.Config()
if self.kamaki_cloud is None:
self.kamaki_cloud = config.get_global("default_cloud")
self.logger.info("Setup kamaki client, using cloud '%s'.." %
self.kamaki_cloud)
auth_url = config.get_cloud(self.kamaki_cloud, "url")
self.logger.debug("Authentication URL is %s" % _green(auth_url))
token = config.get_cloud(self.kamaki_cloud, "token")
#self.logger.debug("Token is %s" % _green(token))
self.astakos_client = AstakosClient(auth_url, token)
cyclades_url = \
self.astakos_client.get_service_endpoints('compute')['publicURL']
self.logger.debug("Cyclades API url is %s" % _green(cyclades_url))
self.cyclades_client = CycladesClient(cyclades_url, token)
self.cyclades_client.CONNECTION_RETRY_LIMIT = 2
image_url = \
self.astakos_client.get_service_endpoints('image')['publicURL']
self.logger.debug("Images API url is %s" % _green(image_url))
self.image_client = ImageClient(cyclades_url, token)
self.image_client.CONNECTION_RETRY_LIMIT = 2
compute_url = \
self.astakos_client.get_service_endpoints('compute')['publicURL']
self.logger.debug("Compute API url is %s" % _green(compute_url))
self.compute_client = ComputeClient(compute_url, token)
self.compute_client.CONNECTION_RETRY_LIMIT = 2
def _wait_transition(self, server_id, current_status, new_status):
"""Wait for server to go from current_status to new_status"""
self.logger.debug("Waiting for server to become %s" % new_status)
timeout = self.config.getint('Global', 'build_timeout')
sleep_time = 5
while True:
server = self.cyclades_client.get_server_details(server_id)
if server['status'] == new_status:
return server
elif timeout < 0:
self.logger.error(
"Waiting for server to become %s timed out" % new_status)
self.destroy_server(False)
sys.exit(-1)
elif server['status'] == current_status:
# Sleep for #n secs and continue
timeout = timeout - sleep_time
time.sleep(sleep_time)
else:
self.logger.error(
"Server failed with status %s" % server['status'])
self.destroy_server(False)
sys.exit(-1)
@_check_kamaki
def destroy_server(self, wait=True):
"""Destroy slave server"""
server_id = int(self.read_temp_config('server_id'))
self.logger.info("Destoying server with id %s " % server_id)
self.cyclades_client.delete_server(server_id)
if wait:
self._wait_transition(server_id, "ACTIVE", "DELETED")
@_check_kamaki
def create_server(self, image=None, flavor=None, ssh_keys=None):
"""Create slave server"""
self.logger.info("Create a new server..")
# Find a build_id to use
self._create_new_build_id()
# Find an image to use
image_id = self._find_image(image)
# Find a flavor to use
flavor_id = self._find_flavor(flavor)
# Create Server
server_name = self.config.get("Deployment", "server_name")
server = self.cyclades_client.create_server(
"%s(BID: %s)" % (server_name, self.build_id),
flavor_id,
image_id)
server_id = server['id']
self.write_temp_config('server_id', server_id)
self.logger.debug("Server got id %s" % _green(server_id))
server_user = server['metadata']['users']
self.write_temp_config('server_user', server_user)
self.logger.debug("Server's admin user is %s" % _green(server_user))
server_passwd = server['adminPass']
self.write_temp_config('server_passwd', server_passwd)
server = self._wait_transition(server_id, "BUILD", "ACTIVE")
self._get_server_ip_and_port(server)
self._copy_ssh_keys(ssh_keys)
# Setup Firewall
self.setup_fabric()
self.logger.info("Setup firewall")
accept_ssh_from = self.config.get('Global', 'accept_ssh_from')
if accept_ssh_from != "":
self.logger.debug("Block ssh except from %s" % accept_ssh_from)
cmd = """
local_ip=$(/sbin/ifconfig eth0 | grep 'inet addr:' | \
cut -d':' -f2 | cut -d' ' -f1)
iptables -A INPUT -s localhost -j ACCEPT
iptables -A INPUT -s $local_ip -j ACCEPT
iptables -A INPUT -s {0} -p tcp --dport 22 -j ACCEPT
iptables -A INPUT -p tcp --dport 22 -j DROP
""".format(accept_ssh_from)
_run(cmd, False)
# Setup apt, download packages
self.logger.debug("Setup apt. Install x2goserver and firefox")
cmd = """
echo 'APT::Install-Suggests "false";' >> /etc/apt/apt.conf
apt-get update
apt-get install curl --yes --force-yes
echo -e "\n\n{0}" >> /etc/apt/sources.list
# Synnefo repo's key
curl https://dev.grnet.gr/files/apt-grnetdev.pub | apt-key add -
# X2GO Key
apt-key adv --recv-keys --keyserver keys.gnupg.net E1F958385BFE2B6E
apt-get install x2go-keyring --yes --force-yes
apt-get update
apt-get install x2goserver x2goserver-xsession \
iceweasel --yes --force-yes
# xterm published application
echo '[Desktop Entry]' > /usr/share/applications/xterm.desktop
echo 'Name=XTerm' >> /usr/share/applications/xterm.desktop
echo 'Comment=standard terminal emulator for the X window system' >> \
/usr/share/applications/xterm.desktop
echo 'Exec=xterm' >> /usr/share/applications/xterm.desktop
echo 'Terminal=false' >> /usr/share/applications/xterm.desktop
echo 'Type=Application' >> /usr/share/applications/xterm.desktop
echo 'Encoding=UTF-8' >> /usr/share/applications/xterm.desktop
echo 'Icon=xterm-color_48x48' >> /usr/share/applications/xterm.desktop
echo 'Categories=System;TerminalEmulator;' >> \
/usr/share/applications/xterm.desktop
""".format(self.config.get('Global', 'apt_repo'))
_run(cmd, False)
def _find_flavor(self, flavor=None):
"""Find a suitable flavor to use
Search by name (reg expression) or by id
"""
# Get a list of flavors from config file
flavors = self.config.get('Deployment', 'flavors').split(",")
if flavor is not None:
# If we have a flavor_name to use, add it to our list
flavors.insert(0, flavor)
list_flavors = self.compute_client.list_flavors()
for flv in flavors:
flv_type, flv_value = parse_typed_option(option="flavor",
value=flv)
if flv_type == "name":
# Filter flavors by name
self.logger.debug(
"Trying to find a flavor with name \"%s\"" % flv_value)
list_flvs = \
[f for f in list_flavors
if re.search(flv_value, f['name'], flags=re.I)
is not None]
elif flv_type == "id":
# Filter flavors by id
self.logger.debug(
"Trying to find a flavor with id \"%s\"" % flv_value)
list_flvs = \
[f for f in list_flavors
if str(f['id']) == flv_value]
else:
self.logger.error("Unrecognized flavor type %s" % flv_type)
# Check if we found one
if list_flvs:
self.logger.debug("Will use \"%s\" with id \"%s\""
% (_green(list_flvs[0]['name']),
_green(list_flvs[0]['id'])))
return list_flvs[0]['id']
self.logger.error("No matching flavor found.. aborting")
sys.exit(1)
def _find_image(self, image=None):
"""Find a suitable image to use
In case of search by name, the image has to belong to one
of the `DEFAULT_SYSTEM_IMAGES_UUID' users.
In case of search by id it only has to exist.
"""
# Get a list of images from config file
images = self.config.get('Deployment', 'images').split(",")
if image is not None:
# If we have an image from command line, add it to our list
images.insert(0, image)
auth = self.astakos_client.authenticate()
user_uuid = auth["access"]["token"]["tenant"]["id"]
list_images = self.image_client.list_public(detail=True)['images']
for img in images:
img_type, img_value = parse_typed_option(option="image", value=img)
if img_type == "name":
# Filter images by name
self.logger.debug(
"Trying to find an image with name \"%s\"" % img_value)
accepted_uuids = DEFAULT_SYSTEM_IMAGES_UUID + [user_uuid]
list_imgs = \
[i for i in list_images if i['user_id'] in accepted_uuids
and
re.search(img_value, i['name'], flags=re.I) is not None]
elif img_type == "id":
# Filter images by id
self.logger.debug(
"Trying to find an image with id \"%s\"" % img_value)
list_imgs = \
[i for i in list_images
if i['id'].lower() == img_value.lower()]
else:
self.logger.error("Unrecognized image type %s" % img_type)
sys.exit(1)
# Check if we found one
if list_imgs:
self.logger.debug("Will use \"%s\" with id \"%s\""
% (_green(list_imgs[0]['name']),
_green(list_imgs[0]['id'])))
return list_imgs[0]['id']
# We didn't found one
self.logger.error("No matching image found.. aborting")
sys.exit(1)
def _get_server_ip_and_port(self, server):
"""Compute server's IPv4 and ssh port number"""
self.logger.info("Get server connection details..")
server_ip = server['attachments'][0]['ipv4']
if (".okeanos.io" in self.cyclades_client.base_url or
".demo.synnefo.org" in self.cyclades_client.base_url):
tmp1 = int(server_ip.split(".")[2])
tmp2 = int(server_ip.split(".")[3])
server_ip = "gate.okeanos.io"
server_port = 10000 + tmp1 * 256 + tmp2
else:
server_port = 22
self.write_temp_config('server_ip', server_ip)
self.logger.debug("Server's IPv4 is %s" % _green(server_ip))
self.write_temp_config('server_port', server_port)
self.logger.debug("Server's ssh port is %s" % _green(server_port))
ssh_command = "ssh -p %s %s@%s" \
% (server_port, server['metadata']['users'], server_ip)
self.logger.debug("Access server using \"%s\"" %
(_green(ssh_command)))
@_check_fabric
def _copy_ssh_keys(self, ssh_keys):
"""Upload/Install ssh keys to server"""
self.logger.debug("Check for authentication keys to use")
if ssh_keys is None:
ssh_keys = self.config.get("Deployment", "ssh_keys")
if ssh_keys != "":
ssh_keys = os.path.expanduser(ssh_keys)
self.logger.debug("Will use \"%s\" authentication keys file" %
_green(ssh_keys))
keyfile = '/tmp/%s.pub' % fabric.env.user
_run('mkdir -p ~/.ssh && chmod 700 ~/.ssh', False)
if ssh_keys.startswith("http://") or \
ssh_keys.startswith("https://") or \
ssh_keys.startswith("ftp://"):
cmd = """
apt-get update
apt-get install wget --yes --force-yes
wget {0} -O {1} --no-check-certificate
""".format(ssh_keys, keyfile)
_run(cmd, False)
elif os.path.exists(ssh_keys):
_put(ssh_keys, keyfile)
else:
self.logger.debug("No ssh keys found")
return
_run('cat %s >> ~/.ssh/authorized_keys' % keyfile, False)
_run('rm %s' % keyfile, False)
self.logger.debug("Uploaded ssh authorized keys")
else:
self.logger.debug("No ssh keys found")
def _create_new_build_id(self):
"""Find a uniq build_id to use"""
with filelocker.lock("%s.lock" % self.temp_config_file,
filelocker.LOCK_EX):
# Read temp_config again to get any new entries
self.temp_config.read(self.temp_config_file)
# Find a uniq build_id to use
if self.build_id is None:
ids = self.temp_config.sections()
if ids:
max_id = int(max(self.temp_config.sections(), key=int))
self.build_id = max_id + 1
else:
self.build_id = 1
self.logger.debug("Will use \"%s\" as build id"
% _green(self.build_id))
# Create a new section
try:
self.temp_config.add_section(str(self.build_id))
except DuplicateSectionError:
msg = ("Build id \"%s\" already in use. " +
"Please use a uniq one or cleanup \"%s\" file.\n") \
% (self.build_id, self.temp_config_file)
self.logger.error(msg)
sys.exit(1)
creation_time = \
time.strftime("%a, %d %b %Y %X", time.localtime())
self.temp_config.set(str(self.build_id),
"created", str(creation_time))
# Write changes back to temp config file
with open(self.temp_config_file, 'wb') as tcf:
self.temp_config.write(tcf)
def write_temp_config(self, option, value):
"""Write changes back to config file"""
# Acquire the lock to write to temp_config_file
with filelocker.lock("%s.lock" % self.temp_config_file,
filelocker.LOCK_EX):
# Read temp_config again to get any new entries
self.temp_config.read(self.temp_config_file)
self.temp_config.set(str(self.build_id), option, str(value))
curr_time = time.strftime("%a, %d %b %Y %X", time.localtime())
self.temp_config.set(str(self.build_id), "modified", curr_time)
# Write changes back to temp config file
with open(self.temp_config_file, 'wb') as tcf:
self.temp_config.write(tcf)
def read_temp_config(self, option):
"""Read from temporary_config file"""
# If build_id is None use the latest one
if self.build_id is None:
ids = self.temp_config.sections()
if ids:
self.build_id = int(ids[-1])
else:
self.logger.error("No sections in temporary config file")
sys.exit(1)
self.logger.debug("Will use \"%s\" as build id"
% _green(self.build_id))
# Read specified option
return self.temp_config.get(str(self.build_id), option)
def setup_fabric(self):
"""Setup fabric environment"""
self.logger.info("Setup fabric parameters..")
fabric.env.user = self.read_temp_config('server_user')
fabric.env.host_string = self.read_temp_config('server_ip')
fabric.env.port = int(self.read_temp_config('server_port'))
fabric.env.password = self.read_temp_config('server_passwd')
fabric.env.connection_attempts = 10
fabric.env.shell = "/bin/bash -c"
fabric.env.disable_known_hosts = True
fabric.env.output_prefix = None
def _check_hash_sum(self, localfile, remotefile):
"""Check hash sums of two files"""
self.logger.debug("Check hash sum for local file %s" % localfile)
hash1 = os.popen("sha256sum %s" % localfile).read().split(' ')[0]
self.logger.debug("Local file has sha256 hash %s" % hash1)
self.logger.debug("Check hash sum for remote file %s" % remotefile)
hash2 = _run("sha256sum %s" % remotefile, False)
hash2 = hash2.split(' ')[0]
self.logger.debug("Remote file has sha256 hash %s" % hash2)
if hash1 != hash2:
self.logger.error("Hashes differ.. aborting")
sys.exit(-1)
@_check_fabric
def clone_repo(self, local_repo=False):
"""Clone Synnefo repo from slave server"""
self.logger.info("Configure repositories on remote server..")
self.logger.debug("Install/Setup git")
cmd = """
apt-get install git --yes --force-yes
git config --global user.name {0}
git config --global user.email {1}
""".format(self.config.get('Global', 'git_config_name'),
self.config.get('Global', 'git_config_mail'))
_run(cmd, False)
# Find synnefo_repo and synnefo_branch to use
synnefo_repo = self.config.get('Global', 'synnefo_repo')
synnefo_branch = self.config.get("Global", "synnefo_branch")
if synnefo_branch == "":
synnefo_branch = \
subprocess.Popen(
["git", "rev-parse", "--abbrev-ref", "HEAD"],
stdout=subprocess.PIPE).communicate()[0].strip()
if synnefo_branch == "HEAD":
synnefo_branch = \
subprocess.Popen(
["git", "rev-parse", "--short", "HEAD"],
stdout=subprocess.PIPE).communicate()[0].strip()
self.logger.info("Will use branch \"%s\"" % _green(synnefo_branch))
if local_repo or synnefo_branch == "":
# Use local_repo
self.logger.debug("Push local repo to server")
# Firstly create the remote repo
_run("git init synnefo", False)
# Then push our local repo over ssh
# We have to pass some arguments to ssh command
# namely to disable host checking.
(temp_ssh_file_handle, temp_ssh_file) = tempfile.mkstemp()
os.close(temp_ssh_file_handle)
# XXX: git push doesn't read the password
cmd = """
echo 'exec ssh -o "StrictHostKeyChecking no" \
-o "UserKnownHostsFile /dev/null" \
-q "$@"' > {4}
chmod u+x {4}
export GIT_SSH="{4}"
echo "{0}" | git push --mirror ssh://{1}@{2}:{3}/~/synnefo
rm -f {4}
""".format(fabric.env.password,
fabric.env.user,
fabric.env.host_string,
fabric.env.port,
temp_ssh_file)
os.system(cmd)
else:
# Clone Synnefo from remote repo
# Currently clonning synnefo can fail unexpectedly
cloned = False
for i in range(10):
self.logger.debug("Clone synnefo from %s" % synnefo_repo)
try:
_run("git clone %s synnefo" % synnefo_repo, False)
cloned = True
break
except BaseException:
self.logger.warning(
"Clonning synnefo failed.. retrying %s" % i)
if not cloned:
self.logger.error("Can not clone Synnefo repo.")
sys.exit(-1)
# Checkout the desired synnefo_branch
self.logger.debug("Checkout \"%s\" branch/commit" % synnefo_branch)
cmd = """
cd synnefo
for branch in `git branch -a | grep remotes | \
grep -v HEAD | grep -v master`; do
git branch --track ${branch##*/} $branch
done
git checkout %s
""" % (synnefo_branch)
_run(cmd, False)
@_check_fabric
def build_synnefo(self):
"""Build Synnefo packages"""
self.logger.info("Build Synnefo packages..")
self.logger.debug("Install development packages")
cmd = """
apt-get update
apt-get install zlib1g-dev dpkg-dev debhelper git-buildpackage \
python-dev python-all python-pip --yes --force-yes
pip install -U devflow
"""
_run(cmd, False)
if self.config.get('Global', 'patch_pydist') == "True":
self.logger.debug("Patch pydist.py module")
cmd = r"""
sed -r -i 's/(\(\?P<name>\[A-Za-z\]\[A-Za-z0-9_\.)/\1\\\-/' \
/usr/share/python/debpython/pydist.py
"""
_run(cmd, False)
# Build synnefo packages
self.logger.debug("Build synnefo packages")
cmd = """
devflow-autopkg snapshot -b ~/synnefo_build-area --no-sign
"""
with fabric.cd("synnefo"):
_run(cmd, True)
# Install snf-deploy package
self.logger.debug("Install snf-deploy package")
cmd = """
dpkg -i snf-deploy*.deb
apt-get -f install --yes --force-yes
"""
with fabric.cd("synnefo_build-area"):
with fabric.settings(warn_only=True):
_run(cmd, True)
# Setup synnefo packages for snf-deploy
self.logger.debug("Copy synnefo debs to snf-deploy packages dir")
cmd = """
cp ~/synnefo_build-area/*.deb /var/lib/snf-deploy/packages/
"""
_run(cmd, False)
@_check_fabric
def build_documentation(self):
"""Build Synnefo documentation"""
self.logger.info("Build Synnefo documentation..")
_run("pip install -U Sphinx", False)
with fabric.cd("synnefo"):
_run("devflow-update-version; "
"./ci/make_docs.sh synnefo_documentation", False)
def fetch_documentation(self, dest=None):
"""Fetch Synnefo documentation"""
self.logger.info("Fetch Synnefo documentation..")
if dest is None:
dest = "synnefo_documentation"
dest = os.path.abspath(dest)
if not os.path.exists(dest):
os.makedirs(dest)
self.fetch_compressed("synnefo/synnefo_documentation", dest)
self.logger.info("Downloaded documentation to %s" %
_green(dest))
@_check_fabric
def deploy_synnefo(self, schema=None):
"""Deploy Synnefo using snf-deploy"""
self.logger.info("Deploy Synnefo..")
if schema is None:
schema = self.config.get('Global', 'schema')
self.logger.debug("Will use \"%s\" schema" % _green(schema))
schema_dir = os.path.join(self.ci_dir, "schemas/%s" % schema)
if not (os.path.exists(schema_dir) and os.path.isdir(schema_dir)):
raise ValueError("Unknown schema: %s" % schema)
self.logger.debug("Upload schema files to server")
_put(os.path.join(schema_dir, "*"), "/etc/snf-deploy/")
self.logger.debug("Change password in nodes.conf file")
cmd = """
sed -i 's/^password =.*/password = {0}/' /etc/snf-deploy/nodes.conf
""".format(fabric.env.password)
_run(cmd, False)
self.logger.debug("Run snf-deploy")
cmd = """
snf-deploy keygen --force
snf-deploy --disable-colors --autoconf all
"""
_run(cmd, True)
@_check_fabric
def unit_test(self):
"""Run Synnefo unit test suite"""
self.logger.info("Run Synnefo unit test suite")
component = self.config.get('Unit Tests', 'component')
self.logger.debug("Install needed packages")
cmd = """
pip install -U mock
pip install -U factory_boy
"""
_run(cmd, False)
self.logger.debug("Upload tests.sh file")
unit_tests_file = os.path.join(self.ci_dir, "tests.sh")
_put(unit_tests_file, ".")
self.logger.debug("Run unit tests")
cmd = """
bash tests.sh {0}
""".format(component)
_run(cmd, True)
@_check_fabric
def run_burnin(self):
"""Run burnin functional test suite"""
self.logger.info("Run Burnin functional test suite")
cmd = """
auth_url=$(grep -e '^url =' .kamakirc | cut -d' ' -f3)
token=$(grep -e '^token =' .kamakirc | cut -d' ' -f3)
images_user=$(kamaki image list -l | grep owner | \
cut -d':' -f2 | tr -d ' ')
snf-burnin --auth-url=$auth_url --token=$token \
--force-flavor=2 --image-id=all \
--system-images-user=$images_user \
{0}
BurninExitStatus=$?
log_folder=$(ls -1d /var/log/burnin/* | tail -n1)
for i in $(ls $log_folder/*/details*); do
echo -e "\\n\\n"
echo -e "***** $i\\n"
cat $i
done
exit $BurninExitStatus
""".format(self.config.get('Burnin', 'cmd_options'))
_run(cmd, True)
@_check_fabric
def fetch_compressed(self, src, dest=None):
"""Create a tarball and fetch it locally"""
self.logger.debug("Creating tarball of %s" % src)
basename = os.path.basename(src)
tar_file = basename + ".tgz"
cmd = "tar czf %s %s" % (tar_file, src)
_run(cmd, False)
if not os.path.exists(dest):
os.makedirs(dest)
tmp_dir = tempfile.mkdtemp()
fabric.get(tar_file, tmp_dir)
dest_file = os.path.join(tmp_dir, tar_file)
self._check_hash_sum(dest_file, tar_file)
self.logger.debug("Untar packages file %s" % dest_file)
cmd = """
cd %s
tar xzf %s
cp -r %s/* %s
rm -r %s
""" % (tmp_dir, tar_file, src, dest, tmp_dir)
os.system(cmd)
self.logger.info("Downloaded %s to %s" %
(src, _green(dest)))
@_check_fabric
def fetch_packages(self, dest=None):
"""Fetch Synnefo packages"""
if dest is None:
dest = self.config.get('Global', 'pkgs_dir')
dest = os.path.abspath(os.path.expanduser(dest))
if not os.path.exists(dest):
os.makedirs(dest)
self.fetch_compressed("synnefo_build-area", dest)
self.logger.info("Downloaded debian packages to %s" %
_green(dest))
def x2go_plugin(self, dest=None):
"""Produce an html page which will use the x2goplugin
Arguments:
dest -- The file where to save the page (String)
"""
output_str = """
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">
<html>
<head>
<title>X2Go SynnefoCI Service</title>
</head>
<body onload="checkPlugin()">
<div id="x2goplugin">
<object
src="location"
type="application/x2go"
name="x2goplugin"
palette="background"
height="100%"
hspace="0"
vspace="0"
width="100%"
x2goconfig="
session=X2Go-SynnefoCI-Session
server={0}
user={1}
sshport={2}
published=true
autologin=true
">
</object>
</div>
</body>
</html>
""".format(self.read_temp_config('server_ip'),
self.read_temp_config('server_user'),
self.read_temp_config('server_port'))
if dest is None:
dest = self.config.get('Global', 'x2go_plugin_file')
self.logger.info("Writting x2go plugin html file to %s" % dest)
fid = open(dest, 'w')
fid.write(output_str)
fid.close()
class Cyclades(livetest.Generic):
"""Set up a Cyclades test"""
def setUp(self):
print
with open(self['cmpimage', 'details']) as f:
self.img_details = eval(f.read())
self.img = self.img_details['id']
with open(self['flavor', 'details']) as f:
self._flavor_details = eval(f.read())
self.PROFILES = ('ENABLED', 'DISABLED', 'PROTECTED')
self.servers = {}
self.now = time.mktime(time.gmtime())
self.servname1 = 'serv' + unicode(self.now)
self.servname2 = self.servname1 + '_v2'
self.servname1 += '_v1'
self.flavorid = self._flavor_details['id']
#servers have to be created at the begining...
self.networks = {}
self.netname1 = 'net' + unicode(self.now)
self.netname2 = 'net' + unicode(self.now) + '_v2'
self.cloud = 'cloud.%s' % self['testcloud']
aurl, self.token = self[self.cloud, 'url'], self[self.cloud, 'token']
self.auth_base = AstakosClient(aurl, self.token)
curl = self.auth_base.get_service_endpoints('compute')['publicURL']
self.client = CycladesClient(curl, self.token)
def tearDown(self):
"""Destoy servers used in testing"""
for net in self.networks.keys():
self._delete_network(net)
for server in self.servers.values():
self._delete_server(server['id'])
print('DEL VM %s (%s)' % (server['id'], server['name']))
def test_000(self):
"Prepare a full Cyclades test scenario"
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self.server2 = self._create_server(
self.servname2, self.flavorid, self.img)
super(self.__class__, self).test_000()
def _create_server(self, servername, flavorid, imageid, personality=None):
server = self.client.create_server(
servername, flavorid, imageid, personality=personality)
print('CREATE VM %s (%s)' % (server['id'], server['name']))
self.servers[servername] = server
return server
def _delete_server(self, servid):
try:
current_state = self.client.get_server_details(servid)
current_state = current_state['status']
if current_state == 'DELETED':
return
self.client.delete_server(servid)
self._wait_for_status(servid, current_state)
self.client.delete_server(servid)
except:
return
def _create_network(self, netname, **kwargs):
net = self.client.create_network(netname, **kwargs)
self.networks[net['id']] = net
return net
def _delete_network(self, netid):
if not netid in self.networks:
return None
print('Disconnect nics of network %s' % netid)
self.client.disconnect_network_nics(netid)
def netwait(wait):
try:
self.client.delete_network(netid)
except ClientError:
time.sleep(wait)
self.do_with_progress_bar(
netwait,
'Delete network %s' % netid,
self._waits[:7])
return self.networks.pop(netid)
def _wait_for_network(self, netid, status):
def netwait(wait):
r = self.client.get_network_details(netid)
if r['status'] == status:
return
time.sleep(wait)
self.do_with_progress_bar(
netwait,
'Wait network %s to reach status %s' % (netid, status),
self._waits[:5])
def _wait_for_nic(self, netid, servid, in_creation=True):
self._wait_for_network(netid, 'ACTIVE')
def nicwait(wait):
nics = self.client.list_server_nics(servid)
for net in nics:
found_nic = net['network_id'] == netid
if (in_creation and found_nic) or not (
in_creation or found_nic):
return
time.sleep(wait)
self.do_with_progress_bar(
nicwait,
'Wait nic-%s-%s to %sconnect' % (
netid,
servid,
'' if in_creation else 'dis'),
self._waits[:5])
for net in self.client.list_server_nics(servid):
if netid == net['network_id']:
return True
return False
def _has_status(self, servid, status):
r = self.client.get_server_details(servid)
#print 'MY ', servid, ' STATUS IS ', r['status']
return r['status'] == status
def _wait_for_status(self, servid, status):
(wait_bar, wait_cb) = self._safe_progress_bar(
'Server %s in %s' % (servid, status))
self.client.wait_server(
servid, status, wait_cb=wait_cb, delay=2, max_wait=198)
self._safe_progress_bar_finish(wait_bar)
def test_parallel_creation(self):
"""test create with multiple threads
Do not use this in regular livetest
"""
from kamaki.clients import SilentEvent
c = []
for i in range(8):
sname = '%s_%s' % (self.servname1, i)
c.append(SilentEvent(
self._create_server, sname, self.flavorid, self.img))
for i in range(8):
c[i].start()
def test_create_server(self):
"""Test create_server"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._wait_for_status(self.server1['id'], 'BUILD')
self._test_0010_create_server()
def _test_0010_create_server(self):
self.assertEqual(self.server1["name"], self.servname1)
self.assertEqual(self.server1["flavor"]["id"], self.flavorid)
self.assertEqual(self.server1["image"]["id"], self.img)
self.assertEqual(self.server1["status"], "BUILD")
def test_list_servers(self):
"""Test list servers"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self.server2 = self._create_server(
self.servname2, self.flavorid, self.img)
self._test_0020_list_servers()
def _test_0020_list_servers(self):
servers = self.client.list_servers()
dservers = self.client.list_servers(detail=True)
"""detailed and simple are same size"""
self.assertEqual(len(dservers), len(servers))
fields = set([
'created', 'flavor', 'hostId', 'image', 'progress', 'status',
'updated'])
for i, srv in enumerate(servers):
self.assertTrue(fields.isdisjoint(srv))
self.assertTrue(fields.issubset(dservers[i]))
"""detailed and simple contain same names"""
names = sorted(map(lambda x: x["name"], servers))
dnames = sorted(map(lambda x: x["name"], dservers))
self.assertEqual(names, dnames)
def _test_0030_wait_test_servers_to_build(self):
"""Pseudo-test to wait for VMs to load"""
print('')
self._wait_for_status(self.server1['id'], 'BUILD')
self._wait_for_status(self.server2['id'], 'BUILD')
def test_get_server_details(self):
"""Test get_server_details"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._wait_for_status(self.server1['id'], 'BUILD')
self._test_0040_get_server_details()
def _test_0040_get_server_details(self):
r = self.client.get_server_details(self.server1['id'])
self.assertEqual(r["name"], self.servname1)
self.assertEqual(r["flavor"]["id"], self.flavorid)
self.assertEqual(r["image"]["id"], self.img)
self.assertEqual(r["status"], "ACTIVE")
def test_update_server_name(self):
"""Test update_server_name"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0050_update_server_name()
def _test_0050_update_server_name(self):
new_name = self.servname1 + '_new_name'
self.client.update_server_name(self.server1['id'], new_name)
r = self.client.get_server_details(
self.server1['id'], success=(200, 400))
self.assertEqual(r['name'], new_name)
changed = self.servers.pop(self.servname1)
changed['name'] = new_name
self.servers[new_name] = changed
def test_reboot_server(self):
"""Test reboot server"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._wait_for_status(self.server1['id'], 'BUILD')
self.server2 = self._create_server(
self.servname2, self.flavorid + 2, self.img)
self._wait_for_status(self.server2['id'], 'BUILD')
self._test_0060_reboot_server()
self._wait_for_status(self.server1['id'], 'REBOOT')
self._wait_for_status(self.server2['id'], 'REBOOT')
def _test_0060_reboot_server(self):
self.client.reboot_server(self.server1['id'])
self.assertTrue(self._has_status(self.server1['id'], 'REBOOT'))
self.client.reboot_server(self.server2['id'], hard=True)
self.assertTrue(self._has_status(self.server2['id'], 'REBOOT'))
def test_resize_server(self):
"""Modify the flavor of a server"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0065_resize_server()
self.delete_server(self.server1['id'])
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
def _test_0065_resize_server(self):
self.client.resize_server(self.servname1, self.flavorid + 2)
srv = self.client.get_flavor_details(self.server1['id'])
self.assertEqual(srv['flavor']['id'], self.flavorid + 2)
def _test_0070_wait_test_servers_to_reboot(self):
"""Pseudo-test to wait for VMs to load"""
print('')
self._wait_for_status(self.server1['id'], 'REBOOT')
self._wait_for_status(self.server2['id'], 'REBOOT')
def test_create_server_metadata(self):
"""Test create_server_metadata"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0080_create_server_metadata()
def _test_0080_create_server_metadata(self):
r1 = self.client.create_server_metadata(
self.server1['id'], 'mymeta', 'mymeta val')
self.assertTrue('mymeta' in r1)
r2 = self.client.get_server_metadata(self.server1['id'], 'mymeta')
self.assert_dicts_are_equal(r1, r2)
def test_get_server_metadata(self):
"""Test get server_metadata"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0090_get_server_metadata()
def _test_0090_get_server_metadata(self):
self.client.update_server_metadata(
self.server1['id'], mymeta_0='val_0')
r = self.client.get_server_metadata(self.server1['id'], 'mymeta_0')
self.assertEqual(r['mymeta_0'], 'val_0')
def test_update_server_metadata(self):
"""Test update_server_metadata"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0100_update_server_metadata()
def _test_0100_update_server_metadata(self):
r1 = self.client.update_server_metadata(
self.server1['id'], mymeta3='val2')
self.assertTrue('mymeta3'in r1)
r2 = self.client.update_server_metadata(
self.server1['id'], mymeta3='val3')
self.assertTrue(r2['mymeta3'], 'val3')
def test_delete_server_metadata(self):
"""Test delete_server_metadata"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0110_delete_server_metadata()
def _test_0110_delete_server_metadata(self):
r1 = self.client.update_server_metadata(
self.server1['id'], mymeta='val')
self.assertTrue('mymeta' in r1)
self.client.delete_server_metadata(self.server1['id'], 'mymeta')
try:
self.client.get_server_metadata(self.server1['id'], 'mymeta')
raise ClientError('Wrong Error', status=100)
except ClientError as err:
self.assertEqual(err.status, 404)
def test_list_flavors(self):
"""Test flavors_get"""
self._test_0120_list_flavors()
def _test_0120_list_flavors(self):
r = self.client.list_flavors()
self.assertTrue(len(r) > 1)
r = self.client.list_flavors(detail=True)
self.assertTrue('SNF:disk_template' in r[0])
def test_get_flavor_details(self):
"""Test test_get_flavor_details"""
self._test_0130_get_flavor_details()
def _test_0130_get_flavor_details(self):
r = self.client.get_flavor_details(self.flavorid)
self.assert_dicts_are_equal(self._flavor_details, r)
#def test_list_images(self):
# """Test list_images"""
# self._test_0140_list_images()
def _test_0140_list_images(self):
r = self.client.list_images()
self.assertTrue(len(r) > 1)
r = self.client.list_images(detail=True)
for detailed_img in r:
if detailed_img['id'] == self.img:
break
self.assert_dicts_are_equal(detailed_img, self.img_details)
def test_get_image_details(self):
"""Test image_details"""
self._test_0150_get_image_details()
def _test_0150_get_image_details(self):
r = self.client.get_image_details(self.img)
self.assert_dicts_are_equal(r, self.img_details)
def test_get_image_metadata(self):
"""Test get_image_metadata"""
self._test_0160_get_image_metadata()
def _test_0160_get_image_metadata(self):
r = self.client.get_image_metadata(self.img)
self.assert_dicts_are_equal(
self.img_details['properties'], r)
for key, val in self.img_details['properties'].items():
r = self.client.get_image_metadata(self.img, key)
self.assertEqual(r[key], val)
def test_shutdown_server(self):
"""Test shutdown_server"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._wait_for_status(self.server1['id'], 'BUILD')
self._test_0170_shutdown_server()
def _test_0170_shutdown_server(self):
self.client.shutdown_server(self.server1['id'])
self._wait_for_status(self.server1['id'], 'ACTIVE')
r = self.client.get_server_details(self.server1['id'])
self.assertEqual(r['status'], 'STOPPED')
def test_start_server(self):
"""Test start_server"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._wait_for_status(self.server1['id'], 'BUILD')
self.client.shutdown_server(self.server1['id'])
self._wait_for_status(self.server1['id'], 'ACTIVE')
self._test_0180_start_server()
def _test_0180_start_server(self):
self.client.start_server(self.server1['id'])
self._wait_for_status(self.server1['id'], 'STOPPED')
r = self.client.get_server_details(self.server1['id'])
self.assertEqual(r['status'], 'ACTIVE')
def test_get_server_console(self):
"""Test get_server_console"""
self.server2 = self._create_server(
self.servname2, self.flavorid, self.img)
self._wait_for_status(self.server2['id'], 'BUILD')
self._test_0190_get_server_console()
def _test_0190_get_server_console(self):
r = self.client.get_server_console(self.server2['id'])
self.assertTrue('host' in r)
self.assertTrue('password' in r)
self.assertTrue('port' in r)
self.assertTrue('type' in r)
def test_get_firewall_profile(self):
"""Test get_firewall_profile"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0200_get_firewall_profile()
def _test_0200_get_firewall_profile(self):
self._wait_for_status(self.server1['id'], 'BUILD')
fprofile = self.client.get_firewall_profile(self.server1['id'])
self.assertTrue(fprofile in self.PROFILES)
def test_set_firewall_profile(self):
"""Test set_firewall_profile"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0210_set_firewall_profile()
def _test_0210_set_firewall_profile(self):
self._wait_for_status(self.server1['id'], 'BUILD')
PROFILES = ['DISABLED', 'ENABLED', 'DISABLED', 'PROTECTED']
fprofile = self.client.get_firewall_profile(self.server1['id'])
print('')
count_success = 0
for counter, fprofile in enumerate(PROFILES):
npos = counter + 1
try:
nprofile = PROFILES[npos]
except IndexError:
nprofile = PROFILES[0]
print('\tprofile swap %s: %s -> %s' % (npos, fprofile, nprofile))
self.client.set_firewall_profile(self.server1['id'], nprofile)
time.sleep(0.5)
self.client.reboot_server(self.server1['id'], hard=True)
time.sleep(1)
self._wait_for_status(self.server1['id'], 'REBOOT')
time.sleep(0.5)
changed = self.client.get_firewall_profile(self.server1['id'])
try:
self.assertEqual(changed, nprofile)
except AssertionError as err:
if count_success:
print('\tFAIL in swap #%s' % npos)
break
else:
raise err
count_success += 1
def test_get_server_stats(self):
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self._test_0220_get_server_stats()
def _test_0220_get_server_stats(self):
r = self.client.get_server_stats(self.server1['id'])
it = ('cpuBar', 'cpuTimeSeries', 'netBar', 'netTimeSeries', 'refresh')
for term in it:
self.assertTrue(term in r)
def test_create_network(self):
"""Test create_network"""
self._test_0230_create_network()
def _test_0230_create_network(self):
print('\twith no params')
self.network1 = self._create_network(self.netname1)
self._wait_for_network(self.network1['id'], 'ACTIVE')
n1id = self.network1['id']
self.network1 = self.client.get_network_details(n1id)
nets = self.client.list_networks(self.network1['id'])
chosen = [net for net in nets if net['id'] == n1id][0]
chosen.pop('updated')
net1 = dict(self.network1)
net1.pop('updated')
self.assert_dicts_are_equal(chosen, net1)
full_args = dict(
cidr='192.168.1.0/24',
gateway='192.168.1.1',
type='MAC_FILTERED',
dhcp=True)
try_args = dict(all=True)
try_args.update(full_args)
for param, val in try_args.items():
print('\tdelete %s to avoid max net limit' % n1id)
self._delete_network(n1id)
kwargs = full_args if param == 'all' else {param: val}
print('\twith %s=%s' % (param, val))
self.network1 = self._create_network(self.netname1, **kwargs)
n1id = self.network1['id']
self._wait_for_network(n1id, 'ACTIVE')
self.network1 = self.client.get_network_details(n1id)
if param == 'all':
for p, v in full_args.items():
self.assertEqual(self.network1[p], v)
else:
self.assertEqual(self.network1[param], val)
def test_connect_server(self):
"""Test connect_server"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self.network1 = self._create_network(self.netname1)
self._wait_for_status(self.server1['id'], 'BUILD')
self._wait_for_network(self.network1['id'], 'ACTIVE')
self._test_0240_connect_server()
def _test_0250_connect_server(self):
self.client.connect_server(self.server1['id'], self.network1['id'])
self.assertTrue(
self._wait_for_nic(self.network1['id'], self.server1['id']))
def test_disconnect_server(self):
"""Test disconnect_server"""
self.test_connect_server()
self._test_0250_disconnect_server()
def _test_0250_disconnect_server(self):
self.client.disconnect_server(self.server1['id'], self.network1['id'])
self.assertTrue(self._wait_for_nic(
self.network1['id'], self.server1['id'], in_creation=False))
def _test_0260_wait_for_second_network(self):
self.network2 = self._create_network(self.netname2)
self._wait_for_network(self.network2['id'], 'ACTIVE')
def test_list_server_nics(self):
"""Test list_server_nics"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self.network2 = self._create_network(self.netname2)
self._wait_for_status(self.server1['id'], 'BUILD')
self._wait_for_network(self.network2['id'], 'ACTIVE')
self._test_0280_list_server_nics()
def _test_0280_list_server_nics(self):
r = self.client.list_server_nics(self.server1['id'])
len0 = len(r)
self.client.connect_server(self.server1['id'], self.network2['id'])
self.assertTrue(
self._wait_for_nic(self.network2['id'], self.server1['id']))
r = self.client.list_server_nics(self.server1['id'])
self.assertTrue(len(r) > len0)
def test_list_networks(self):
"""Test list_network"""
self.network1 = self._create_network(self.netname1)
self._wait_for_network(self.network1['id'], 'ACTIVE')
self._test_0290_list_networks()
def _test_0290_list_networks(self):
r = self.client.list_networks()
self.assertTrue(len(r) > 1)
ids = [net['id'] for net in r]
names = [net['name'] for net in r]
self.assertTrue('1' in ids)
#self.assertTrue('public' in names)
self.assertTrue(self.network1['id'] in ids)
self.assertTrue(self.network1['name'] in names)
r = self.client.list_networks(detail=True)
ids = [net['id'] for net in r]
names = [net['name'] for net in r]
for net in r:
self.assertTrue(net['id'] in ids)
self.assertTrue(net['name'] in names)
for term in ('status', 'updated', 'created'):
self.assertTrue(term in net.keys())
def test_list_network_nics(self):
"""Test list_server_nics"""
self.server1 = self._create_server(
self.servname1, self.flavorid, self.img)
self.network1 = self._create_network(self.netname1)
self.network2 = self._create_network(self.netname2)
self._wait_for_status(self.server1['id'], 'BUILD')
self._wait_for_network(self.network1['id'], 'ACTIVE')
self._wait_for_network(self.network2['id'], 'ACTIVE')
self.client.connect_server(self.server1['id'], self.network1['id'])
self.client.connect_server(self.server1['id'], self.network2['id'])
self._wait_for_nic(self.network1['id'], self.server1['id'])
self._wait_for_nic(self.network2['id'], self.server1['id'])
self._test_0293_list_network_nics()
def _test_0293_list_network_nics(self):
netid1, netid2 = self.network1['id'], self.network2['id']
r = self.client.list_network_nics(netid1)
expected = ['nic-%s-1' % self.server1['id']]
self.assertEqual(r, expected)
r = self.client.list_network_nics(netid2)
expected = ['nic-%s-2' % self.server1['id']]
self.assertEqual(r, expected)
def test_get_network_details(self):
"""Test get_network_details"""
self.network1 = self._create_network(self.netname1)
self._test_0300_get_network_details()
def _test_0300_get_network_details(self):
r = self.client.get_network_details(self.network1['id'])
net1 = dict(self.network1)
net1.pop('status')
net1.pop('updated', None)
net1.pop('attachments')
r.pop('status')
r.pop('updated', None)
r.pop('attachments')
self.assert_dicts_are_equal(net1, r)
def test_update_network_name(self):
self.network2 = self._create_network(self.netname2)
self._test_0310_update_network_name()
def _test_0310_update_network_name(self):
updated_name = self.netname2 + '_upd'
self.client.update_network_name(self.network2['id'], updated_name)
def netwait(wait):
r = self.client.get_network_details(self.network2['id'])
if r['name'] == updated_name:
return
time.sleep(wait)
self.do_with_progress_bar(
netwait,
'Network %s name is changing:' % self.network2['id'],
self._waits[:5])
r = self.client.get_network_details(self.network2['id'])
self.assertEqual(r['name'], updated_name)
""" Don't have auth to test this
__author__ = 'cmantas'
import psycopg2
from kamaki.clients.astakos import AstakosClient
from kamaki.clients.cyclades import CycladesClient
from logging import getLogger, ERROR
# init synneffo stuff
AUTHENTICATION_URL="https://accounts.okeanos.grnet.gr/identity/v2.0"
TOKEN="hYDRO-FEV5d8wFxpOID-DF3_FWhsuD8dvTdbEX2qQRQ"
synnefo_user = AstakosClient(AUTHENTICATION_URL, TOKEN)
synnefo_user.logger.setLevel(ERROR)
getLogger().setLevel(ERROR)
cyclades_endpoints = synnefo_user.get_service_endpoints('compute')
CYCLADES_URL = cyclades_endpoints['publicURL']
cyclades_client = CycladesClient(CYCLADES_URL, TOKEN)
# connect to db
db = psycopg2.connect(host="127.0.0.1", user="******", password="******", database="celardb")
cursor = db.cursor()
#clear all old values from the DB
cursor.execute("DELETE FROM \"RESOURCE_TYPE\" WHERE TRUE")
cursor.execute("DELETE FROM \"SPECS\" WHERE TRUE")
cursor.execute("DELETE FROM \"PROVIDED_RESOURCE\" WHERE TRUE")
# add 'VM_FLAVOR' entry on the RESOURCE_TYPE table
class AstakosClient(TestCase):
cached = False
def assert_dicts_are_equal(self, d1, d2):
for k, v in d1.items():
self.assertTrue(k in d2)
if isinstance(v, dict):
self.assert_dicts_are_equal(v, d2[k])
else:
self.assertEqual(unicode(v), unicode(d2[k]))
def setUp(self):
self.url = 'https://astakos.example.com'
self.token = 'ast@k0sT0k3n=='
from kamaki.clients.astakos import AstakosClient as AC
self.client = AC(self.url, self.token)
def tearDown(self):
FR.json = example
@patch('%s.LoggedAstakosClient.__init__' % astakos_pkg, return_value=None)
@patch(
'%s.LoggedAstakosClient.get_endpoints' % astakos_pkg,
return_value=example)
def _authenticate(self, get_endpoints, sac):
r = self.client.authenticate()
self.assertEqual(
sac.mock_calls[-1], call(self.token, self.url,
logger=getLogger('astakosclient')))
self.assertEqual(get_endpoints.mock_calls[-1], call())
return r
def test_authenticate(self):
r = self._authenticate()
self.assert_dicts_are_equal(r, example)
uuid = example['access']['user']['id']
self.assert_dicts_are_equal(self.client._uuids, {self.token: uuid})
self.assert_dicts_are_equal(self.client._cache, {uuid: r})
from astakosclient import AstakosClient as SAC
self.assertTrue(isinstance(self.client._astakos[uuid], SAC))
self.assert_dicts_are_equal(self.client._uuids2usernames, {uuid: {}})
self.assert_dicts_are_equal(self.client._usernames2uuids, {uuid: {}})
def test_get_client(self):
if not self.cached:
self._authenticate()
from astakosclient import AstakosClient as SNFAC
self.assertTrue(self.client.get_client(), SNFAC)
def test_get_token(self):
self._authenticate()
uuid = self.client._uuids.values()[0]
self.assertEqual(self.client.get_token(uuid), self.token)
def test_get_services(self):
if not self.cached:
self._authenticate()
slist = self.client.get_services()
self.assertEqual(slist, example['access']['serviceCatalog'])
def test_get_service_details(self):
if not self.cached:
self._authenticate()
stype = '#FAIL'
self.assertRaises(ClientError, self.client.get_service_details, stype)
stype = 'compute'
expected = [s for s in example['access']['serviceCatalog'] if (
s['type'] == stype)]
self.assert_dicts_are_equal(
self.client.get_service_details(stype), expected[0])
def test_get_service_endpoints(self):
if not self.cached:
self._authenticate()
stype, version = 'compute', 'V2'
self.assertRaises(
ClientError, self.client.get_service_endpoints, stype)
expected = [s for s in example['access']['serviceCatalog'] if (
s['type'] == stype)]
expected = [e for e in expected[0]['endpoints'] if (
e['versionId'] == version.lower())]
self.assert_dicts_are_equal(
self.client.get_service_endpoints(stype, version), expected[0])
def test_user_info(self):
if not self.cached:
self._authenticate()
self.assertTrue(set(example['access']['user'].keys()).issubset(
self.client.user_info().keys()))
def test_item(self):
if not self.cached:
self._authenticate()
for term, val in example['access']['user'].items():
self.assertEqual(self.client.term(term, self.token), val)
self.assertTrue(
example['access']['user']['email'][0] in self.client.term('email'))
def test_list_users(self):
if not self.cached:
self._authenticate()
FR.json = example
self._authenticate()
r = self.client.list_users()
self.assertTrue(len(r) == 1)
self.assertEqual(r[0]['auth_token'], self.token)
@patch(
'%s.LoggedAstakosClient.get_usernames' % astakos_pkg,
return_value={42: 'username42', 43: 'username43'})
def test_uuids2usernames(self, get_usernames):
from astakosclient import AstakosClientException
self.assertRaises(
AstakosClientException, self.client.uuids2usernames, [42, 43])
with patch(
'%s.LoggedAstakosClient.__init__' % astakos_pkg,
return_value=None) as sac:
with patch(
'%s.LoggedAstakosClient.get_endpoints' % astakos_pkg,
return_value=example) as get_endpoints:
r = self.client.uuids2usernames([42, 43])
self.assert_dicts_are_equal(
r, {42: 'username42', 43: 'username43'})
self.assertEqual(sac.mock_calls[-1], call(
self.token, self.url, logger=getLogger('astakosclient')))
self.assertEqual(get_endpoints.mock_calls[-1], call())
self.assertEqual(get_usernames.mock_calls[-1], call([42, 43]))
@patch(
'%s.LoggedAstakosClient.get_uuids' % astakos_pkg,
return_value={'username42': 42, 'username43': 43})
def test_usernames2uuids(self, get_uuids):
from astakosclient import AstakosClientException
self.assertRaises(
AstakosClientException, self.client.usernames2uuids, ['u1', 'u2'])
with patch(
'%s.LoggedAstakosClient.__init__' % astakos_pkg,
return_value=None) as sac:
with patch(
'%s.LoggedAstakosClient.get_endpoints' % astakos_pkg,
return_value=example) as get_endpoints:
r = self.client.usernames2uuids(['u1', 'u2'])
self.assert_dicts_are_equal(
r, {'username42': 42, 'username43': 43})
self.assertEqual(sac.mock_calls[-1], call(
self.token, self.url, logger=getLogger('astakosclient')))
self.assertEqual(get_endpoints.mock_calls[-1], call())
self.assertEqual(get_uuids.mock_calls[-1], call(['u1', 'u2']))
def main():
"""Entry point for icaas-monitord"""
args = get_args()
manifest = read_manifest(args.manifest)
if 'status' not in manifest['service']:
sys.stderr.write('"status" is missing from the service section of the '
'manifest')
sys.exit(3)
if 'insecure' in manifest and manifest['insecure'].lower() == 'true':
verify = False
else:
verify = True
report = Report(manifest['service']['status'], verify=verify,
log=sys.stderr)
def missing_key(key, section):
"""missing key message"""
return "`%s' is missing from the `%s' section of the manifest" % \
(key, section)
# Validate the manifest
for key in 'url', 'token', 'log', 'status':
if key not in manifest['service']:
report.error(missing_key(key, 'service'))
sys.exit(3)
for key in 'url', 'name', 'object':
if key not in manifest['image']:
report.error(missing_key(key, 'image'))
sys.exit(3)
service = manifest['service']
try:
container, logname = service['log'].split('/', 1)
except ValueError:
report.error('Incorrect format for log entry in manifest file')
# Use the systems certificates
https.patch_with_certs(CERTS)
account = AstakosClient(service['url'], service['token'])
try:
account.authenticate()
except AstakosClientError as err:
report.error("Astakos: %s" % err)
sys.exit(3)
pithos = PithosClient(
account.get_service_endpoints('object-store')['publicURL'],
account.token, account.user_info['id'], container)
if args.daemonize:
daemon_context = daemon.DaemonContext(stdin=sys.stdin,
stdout=sys.stdout,
stderr=sys.stderr)
daemon_context.open()
with open(PID, 'w') as pid:
pid.write("%d\n" % os.getpid())
try:
# Export manifest to environment variables
for section in manifest:
for key, value in manifest[section].items():
name = "ICAAS_%s_%s" % (section.upper(), key.upper())
os.environ[name] = value
# Use SIGHUP to unblock from the sleep if necessary
signal.signal(signal.SIGHUP, lambda x, y: None)
if 'ICAAS_MONITOR_SIGSTOP' in os.environ:
# Tell service supervisor that we are ready.
syslog.syslog(
syslog.LOG_NOTICE, "Stopping with SIGSTOP as the "
"environment variable ICAAS_MONITOR_SIGSTOP is defined")
os.kill(os.getpid(), signal.SIGSTOP)
del os.environ['ICAAS_MONITOR_SIGSTOP']
if do_main_loop(args.interval, pithos, logname):
report.success()
else:
report.error("Image creation failed. Check the log for more info")
finally:
os.unlink(PID)
class Astakos(livetest.Generic):
def setUp(self):
self.cloud = 'cloud.%s' % self['testcloud']
self.client = AstakosClient(
self[self.cloud, 'url'], self[self.cloud, 'token'])
with open(self['astakos', 'details']) as f:
self._astakos_details = eval(f.read())
def test_authenticate(self):
self._test_0010_authenticate()
def _test_0010_authenticate(self):
r = self.client.authenticate()
self.assert_dicts_are_equal(r, self._astakos_details)
def test_get_services(self):
self._test_0020_get_services()
def _test_0020_get_services(self):
for args in (tuple(), (self[self.cloud, 'token'],)):
r = self.client.get_services(*args)
services = self._astakos_details['access']['serviceCatalog']
self.assertEqual(len(services), len(r))
for i, service in enumerate(services):
self.assert_dicts_are_equal(r[i], service)
self.assertRaises(ClientError, self.client.get_services, 'wrong_token')
def test_get_service_details(self):
self._test_0020_get_service_details()
def _test_0020_get_service_details(self):
parsed_services = dict()
for args in product(
self._astakos_details['access']['serviceCatalog'],
([tuple(), (self[self.cloud, 'token'],)])):
service = args[0]
if service['type'] in parsed_services:
continue
r = self.client.get_service_details(service['type'], *args[1])
self.assert_dicts_are_equal(r, service)
parsed_services[service['type']] = True
self.assertRaises(
ClientError, self.client.get_service_details, 'wrong_token')
def test_get_service_endpoints(self):
self._test_0020_get_service_endpoints()
def _test_0020_get_service_endpoints(self):
parsed_services = dict()
for args in product(
self._astakos_details['access']['serviceCatalog'],
([], [self[self.cloud, 'token']])):
service = args[0]
if service['type'] in parsed_services:
continue
for endpoint, with_id in product(
service['endpoints'], (True, False)):
vid = endpoint['versionId'] if (
with_id and endpoint['versionId']) else None
end_args = [service['type'], vid] + args[1]
r = self.client.get_service_endpoints(*end_args)
self.assert_dicts_are_equal(r, endpoint)
parsed_services[service['type']] = True
self.assertRaises(
ClientError, self.client.get_service_endpoints, 'wrong_token')
def test_user_info(self):
self._test_0020_user_info()
def _test_0020_user_info(self):
self.assertTrue(set([
'roles',
'name',
'id']).issubset(self.client.user_info().keys()))
def test_get(self):
self._test_0020_get()
def _test_0020_get(self):
for term in ('id', 'name'):
self.assertEqual(
self.client.term(term, self[self.cloud, 'token']),
self['astakos', term] or '')
def test_list_users(self):
self.client.authenticate()
self._test_0020_list_users()
def _test_0020_list_users(self):
terms = set(['name', 'id'])
uuid = 0
for r in self.client.list_users():
self.assertTrue(terms.issubset(r.keys()))
self.assertTrue(uuid != r['id'] if uuid else True)
uuid = r['id']
from sys import stderr
from os.path import abspath
from base64 import b64encode
from lib.persistance_module import *
from logging import ERROR
USER = "******"
#retrieve the credentials for the specified users
AUTHENTICATION_URL, TOKEN = get_credentials(USER)
synnefo_user = AstakosClient(AUTHENTICATION_URL, TOKEN)
synnefo_user.logger.setLevel(ERROR)
getLogger().setLevel(ERROR)
cyclades_endpoints = synnefo_user.get_service_endpoints("compute")
CYCLADES_URL = cyclades_endpoints['publicURL']
cyclades_client = CycladesClient(CYCLADES_URL, TOKEN)
cyclades_net_client = CycladesNetworkClient(CYCLADES_URL, TOKEN)
pub_keys_path = 'keys/just_a_key.pub'
priv_keys_path = 'keys/just_a_key'
#creates a "personality"
def personality(username):
"""
:param pub_keys_path: a path to the public key(s) to be used for this personality
:param ssh_keys_path: a path to the private key(s) to be used for this personality
"""
def main():
#authenticate to Astakos
print"---------------------------------------------------"
print"******* Authenticating to Astakos******************"
print"---------------------------------------------------"
try:
my_astakos_client = AstakosClient(AUTHENTICATION_URL,
TOKEN)
my_accountData = my_astakos_client.authenticate()
ACCOUNT_UUID = my_accountData['access']['user']['id']
print "Status: Authenticated"
except ClientError:
print"Failed to authenticate user token"
print"\n"
print"---------------------------------------------------"
print"**********Getting Endpoints for pithos*************"
print"---------------------------------------------------"
#get endpoint url
try:
endpoints = my_astakos_client.get_service_endpoints('object-store')
PITHOS_URL = endpoints['publicURL']
print "The public URL:", PITHOS_URL
except ClientError:
print "Failed to get endpoints for pithos"
print"\n"
print"---------------------------------------------------"
print"**********Authenticating to Pithos*****************"
print"---------------------------------------------------"
#Initialize pithos client
try:
pithos = PithosClient(PITHOS_URL, TOKEN)
pithos.account = ACCOUNT_UUID
pithos.container = ''
except ClientError:
print "Failed to initialize Pithos+ client"
print"\n"
print"---------------------------------------------------"
print"**********LIST ALL CONTAINERS IN YOUR ACCOUNT******"
print"---------------------------------------------------"
#list all containers
try:
container_list = pithos.list_containers()
containers = parse_containers(container_list)
ContNums = len(containers)
print "The number of Containers in your account:", ContNums
print "The containers are"
print ','.join(containers)
except ClientError:
print"Error in container list"
print"\n"
print"---------------------------------------------------"
print"******LIST OBJECTS OF A FOLDER IN A CONTAINER******"
print"---------------------------------------------------"
#list all containers
try:
print_container_objects(pithos, YOUR_CONTAINER,
prefixName=YOUR_FOLDER_PATH)
except ClientError:
print"Error in listing folder objects"
print"\n"
print"---------------------------------------------------"
print"**********Print objects for all containers*********"
print"---------------------------------------------------"
try:
for i in range(len(containers)):
print_container_objects(pithos, containers[i])
except ClientError as e:
print('Error: %s' % e)
if e.status:
print('- error code: %s' % e.status)
if e.details:
for detail in e.details:
print('- %s' % detail)
# Create and set a different container than pithos
print "Create a new container - my container"
CONTAINER = 'my container'
pithos.create_container(CONTAINER)
pithos.container = CONTAINER
print"\n"
print"---------------------------------------------------"
print"**********UPLOAD AND DOWNLOAD**********************"
print"---------------------------------------------------"
"""
B. UPLOAD AND DOWNLOAD
"""
print "Upload a small file to pithos"
# Upload a small file
print './test/'+SMALLFILE
with open('./test/'+SMALLFILE) as f:
pithos.upload_object(SMALLFILE, f)
print "Download a small file from pithos and store to string"
print SMALLFILE
FILETOSTRING = pithos.download_to_string(SMALLFILE,
download_cb=
create_pb('Downloading...'))
print "Small file string:", FILETOSTRING
#To optimize for large files, allow pithos client
# to use multiple threads! pithos client will
# auto-adjust the number of threads, up to a limit
pithos.MAX_THREADS = 5
print "Upload a large file to pithos"
# Now, large file upload will be optimized:
# dd if=/dev/zero of=test/large.txt count=8 bs=1073741824
with open('./test/'+BIGFILE) as f:
pithos.upload_object(BIGFILE, f,
hash_cb=create_pb('Calculating hashes...'),
upload_cb=create_pb('Uploading...'))
print "Create my own metadata for object"
tags = {}
tags['mytag'] = 12
pithos.set_object_meta(BIGFILE, tags)
myOwnMetadataObject = pithos.get_object_meta(BIGFILE)
print "Object Metatadata", myOwnMetadataObject
print "Download a large file from pithos"
# Download a file (btw, MAX_THREADS are still 5)
with open('./test/'+TMPFILE, 'wb+') as f:
pithos.download_object(BIGFILE, f,
download_cb=create_pb('Downloading...'))
# HIGHLIGHTS: If parts of the file are already uploaded or downloaded,
# corresponding methods will transfer only the missing parts!
print"\n"
print"---------------------------------------------------"
print"**********CREATE A NEW CONTAINER AND MOVE OBJECT***"
print"---------------------------------------------------"
""" Create a new container and move object"""
print "Create a new container - containerToCopy"
CONTAINERNEW = 'containertocopy'
pithos.create_container(CONTAINERNEW)
pithos.move_object(CONTAINER, SMALLFILE, CONTAINERNEW, SMALLFILE)
print"\n"
print"---------------------------------------------------"
print"**********DELETE AND RECOVER***********************"
print"---------------------------------------------------"
"""
C. DELETE AND RECOVER
"""
# Delete a file
pithos.container = CONTAINER
pithos.delete_object(BIGFILE)
# Recover file
file_versions = pithos.get_object_versionlist(BIGFILE)
print "The file versions"
for data in file_versions:
print "The value of id=", data[0], "date=", data[1]
first_version = file_versions[0]
v_id, v_date = first_version
pithos.copy_object(CONTAINER, BIGFILE,
CONTAINER, BIGFILE,
source_version=v_id)
print"\n"
print"---------------------------------------------------"
print"**********GET FILE DETAILS************************"
print"---------------------------------------------------"
objectDetails = pithos.get_object_info(BIGFILE)
for obj in objectDetails:
print "The value of", obj, "=", objectDetails.get(obj)
print"\n"
print"---------------------------------------------------"
print"**********SHARING AND PUBLISHING*******************"
print"---------------------------------------------------"
"""
D. SHARING AND PUBLISHING
"""
# Read permission to all pithos users
pithos.set_object_sharing(FILETOPUBLISH, read_permission='*')
# Publish and get public URL
pithos.publish_object(FILETOPUBLISH)
print "Get sharing and public information"
# Get sharing and public information
info = pithos.get_object_info(FILETOPUBLISH)
for data in info:
print "The value of", data, "=", info.get(data)
sharing = info.get('x-object-sharing', {})
print sharing
public = info.get('x-object-public', None)
print "The public URL=", public
print "Remove sharing and publishing"
# Remove sharing and publishing
pithos.del_object_sharing(FILETOPUBLISH)
pithos.unpublish_object(FILETOPUBLISH)
print "Get sharing and public information"
# Get sharing and public information
info = pithos.get_object_info(FILETOPUBLISH)
sharing = info.get('x-object-sharing', {})
public = info.get('x-object-public', None)
print "The public URL=", public
class Pithos(livetest.Generic):
files = []
def setUp(self):
self.cloud = 'cloud.%s' % self['testcloud']
aurl, self.token = self[self.cloud, 'url'], self[self.cloud, 'token']
self.auth_base = AstakosClient(aurl, self.token)
purl = self.auth_base.get_service_endpoints(
'object-store')['publicURL']
self.uuid = self.auth_base.user_term('id')
self.client = PithosClient(purl, self.token, self.uuid)
self.now = time.mktime(time.gmtime())
self.now_unformated = datetime.datetime.utcnow()
self._init_data()
"""Prepare an object to be shared - also its container"""
self.client.container = self.c1
self.client.object_post(
'test',
update=True,
permissions={'read': [self.client.account]})
self.create_remote_object(self.c1, 'another.test')
def _init_data(self):
self.c1 = 'c1_' + unicode(self.now)
self.c2 = 'c2_' + unicode(self.now)
self.c3 = 'c3_' + unicode(self.now)
try:
self.client.create_container(self.c2)
except ClientError:
pass
try:
self.client.create_container(self.c1)
except ClientError:
pass
try:
self.client.create_container(self.c3)
except ClientError:
pass
self.create_remote_object(self.c1, 'test')
self.create_remote_object(self.c2, 'test')
self.create_remote_object(self.c1, 'test1')
self.create_remote_object(self.c2, 'test1')
def create_remote_object(self, container, obj):
self.client.container = container
self.client.object_put(
obj,
content_type='application/octet-stream',
data='file %s that lives in %s' % (obj, container),
metadata={'incontainer': container})
def forceDeleteContainer(self, container):
self.client.container = container
try:
r = self.client.list_objects()
except ClientError:
return
for obj in r:
name = obj['name']
self.client.del_object(name)
r = self.client.container_delete()
self.container = ''
def tearDown(self):
"""Destroy test cases"""
for f in self.files:
f.close()
self.forceDeleteContainer(self.c1)
self.forceDeleteContainer(self.c2)
try:
self.forceDeleteContainer(self.c3)
except ClientError:
pass
self.client.container = ''
def test_000(self):
"""Prepare a full Pithos+ test"""
print('')
super(self.__class__, self).test_000()
def test_account_head(self):
"""Test account_HEAD"""
self._test_0010_account_head()
def _test_0010_account_head(self):
r = self.client.account_head()
self.assertEqual(r.status_code, 204)
r = self.client.account_head(until='1000000000')
self.assertEqual(r.status_code, 204)
r = self.client.get_account_info(until='1000000000')
datestring = unicode(r['x-account-until-timestamp'])
self.assertEqual(u'Sun, 09 Sep 2001 01:46:40 GMT', datestring)
r = self.client.get_account_quota()
self.assertTrue('x-account-policy-quota' in r)
#r = self.client.get_account_versioning()
#self.assertTrue('x-account-policy-versioning' in r)
"""Check if(un)modified_since"""
for format in self.client.DATE_FORMATS:
now_formated = self.now_unformated.strftime(format)
r1 = self.client.account_head(
if_modified_since=now_formated, success=(204, 304, 412))
sc1 = r1.status_code
r2 = self.client.account_head(
if_unmodified_since=now_formated, success=(204, 304, 412))
sc2 = r2.status_code
self.assertNotEqual(sc1, sc2)
def test_account_get(self):
"""Test account_GET"""
self._test_0020_account_get()
def _test_0020_account_get(self):
#r = self.client.account_get()
#self.assertEqual(r.status_code, 200)
r = self.client.list_containers()
fullLen = len(r)
self.assertTrue(fullLen > 2)
r = self.client.account_get(limit=1)
self.assertEqual(len(r.json), 1)
r = self.client.account_get(marker='c2_')
temp_c0 = r.json[0]['name']
temp_c2 = r.json[2]['name']
r = self.client.account_get(limit=2, marker='c2_')
conames = [container['name'] for container in r.json if (
container['name'].lower().startswith('c2_'))]
self.assertTrue(temp_c0 in conames)
self.assertFalse(temp_c2 in conames)
r = self.client.account_get(show_only_shared=True)
self.assertTrue(self.c1 in [c['name'] for c in r.json])
r = self.client.account_get(until=1342609206.0)
self.assertTrue(len(r.json) <= fullLen)
"""Check if(un)modified_since"""
for format in self.client.DATE_FORMATS:
now_formated = self.now_unformated.strftime(format)
r1 = self.client.account_get(
if_modified_since=now_formated, success=(200, 304, 412))
sc1 = r1.status_code
r2 = self.client.account_get(
if_unmodified_since=now_formated, success=(200, 304, 412))
sc2 = r2.status_code
self.assertNotEqual(sc1, sc2)
"""Check sharing_accounts"""
r = self.client.get_sharing_accounts()
try:
self.assertTrue(len(r) > 0)
except AssertionError as e:
print '\n\tWARNING: Are there any sharers to your account?'
self.assertEqual(len(r), 0)
print '\tIf there are, this (%s) is an error, else it is OK' % e
def test_account_post(self):
"""Test account_POST"""
self._test_0030_account_post()
def _test_0030_account_post(self):
r = self.client.account_post()
self.assertEqual(r.status_code, 202)
grpName = 'grp' + unicode(self.now)
"""Method set/del_account_meta and set_account_groupcall use
account_post internally
"""
u1 = self.client.account
# Invalid display name
u2 = '1nc0r3c7-d15p14y-n4m3'
self.assertRaises(
ClientError,
self.client.set_account_group,
grpName, [u1, u2])
self.client.set_account_group(grpName, [u1])
r = self.client.get_account_group()
self.assertEqual(r['x-account-group-' + grpName], '%s' % u1)
self.client.del_account_group(grpName)
r = self.client.get_account_group()
self.assertTrue('x-account-group-' + grpName not in r)
mprefix = 'meta' + unicode(self.now)
self.client.set_account_meta({
mprefix + '1': 'v1', mprefix + '2': 'v2'})
r = self.client.get_account_meta()
self.assertEqual(r['x-account-meta-' + mprefix + '1'], 'v1')
self.assertEqual(r['x-account-meta-' + mprefix + '2'], 'v2')
self.client.del_account_meta(mprefix + '1')
r = self.client.get_account_meta()
self.assertTrue('x-account-meta-' + mprefix + '1' not in r)
self.client.del_account_meta(mprefix + '2')
r = self.client.get_account_meta()
self.assertTrue('x-account-meta-' + mprefix + '2' not in r)
"""Missing testing for quota, versioning, because normally
you don't have permissions to modify those at account level
"""
#newquota = 1000000
#self.client.set_account_quota(newquota)
#r = self.client.get_account_info()
#print(unicode(r))
#r = self.client.get_account_quota()
#self.assertEqual(r['x-account-policy-quota'], newquota)
#self.client.set_account_versioning('auto')
def test_container_head(self):
"""Test container_HEAD"""
self._test_0040_container_head()
def _test_0040_container_head(self):
self.client.container = self.c1
r = self.client.container_head()
self.assertEqual(r.status_code, 204)
"""Check until"""
r = self.client.container_head(until=1000000, success=(204, 404))
self.assertEqual(r.status_code, 404)
"""Check and if(un)modified_since"""
for format in self.client.DATE_FORMATS:
now_formated = self.now_unformated.strftime(format)
r1 = self.client.container_head(
if_modified_since=now_formated, success=(204, 304, 412))
sc1 = r1.status_code
r2 = self.client.container_head(
if_unmodified_since=now_formated, success=(204, 304, 412))
sc2 = r2.status_code
self.assertNotEqual(sc1, sc2)
"""Check container object meta"""
r = self.client.get_container_object_meta()
self.assertEqual(r['x-container-object-meta'], 'Incontainer')
def test_container_get(self):
"""Test container_GET"""
self._test_0050_container_get()
def _test_0050_container_get(self):
self.client.container = self.c1
r = self.client.container_get()
self.assertEqual(r.status_code, 200)
fullLen = len(r.json)
r = self.client.container_get(prefix='test')
lalobjects = [obj for obj in r.json if obj['name'].startswith('test')]
self.assertTrue(len(r.json) > 1)
self.assertEqual(len(r.json), len(lalobjects))
r = self.client.container_get(limit=1)
self.assertEqual(len(r.json), 1)
r = self.client.container_get(marker='another')
self.assertTrue(len(r.json) > 1)
neobjects = [obj for obj in r.json if obj['name'] > 'another']
self.assertEqual(len(r.json), len(neobjects))
r = self.client.container_get(prefix='another.test', delimiter='.')
self.assertTrue(fullLen > len(r.json))
r = self.client.container_get(path='/')
self.assertEqual(fullLen, len(r.json))
r = self.client.container_get(format='xml')
self.assertEqual(r.text.split()[4], 'name="' + self.c1 + '">')
r = self.client.container_get(meta=['incontainer'])
self.assertTrue(len(r.json) > 0)
r = self.client.container_get(show_only_shared=True)
self.assertTrue(len(r.json) < fullLen)
try:
r = self.client.container_get(until=1000000000)
datestring = unicode(r.headers['x-account-until-timestamp'])
self.assertEqual(u'Sun, 09 Sep 2001 01:46:40 GMT', datestring)
except ClientError:
pass
"""Check and if un/modified_since"""
for format in self.client.DATE_FORMATS:
now_formated = self.now_unformated.strftime(format)
r1 = self.client.container_get(
if_modified_since=now_formated, success=(200, 304, 412))
sc1 = r1.status_code
r2 = self.client.container_get(
if_unmodified_since=now_formated, success=(200, 304, 412))
sc2 = r2.status_code
self.assertNotEqual(sc1, sc2)
def test_container_put(self):
"""Test container_PUT"""
self._test_0050_container_put()
def _test_0050_container_put(self):
self.client.container = self.c2
r = self.client.create_container()
self.assertTrue(isinstance(r, dict))
r = self.client.get_container_limit(self.client.container)
cquota = r.values()[0]
newquota = 2 * int(cquota)
r = self.client.create_container(sizelimit=newquota)
self.assertTrue(isinstance(r, dict))
r = self.client.get_container_limit(self.client.container)
xquota = int(r.values()[0])
self.assertEqual(newquota, xquota)
r = self.client.create_container(versioning='auto')
self.assertTrue(isinstance(r, dict))
r = self.client.get_container_versioning(self.client.container)
nvers = r.values()[0]
self.assertEqual('auto', nvers)
r = self.client.container_put(versioning='none')
self.assertEqual(r.status_code, 202)
r = self.client.get_container_versioning(self.client.container)
nvers = r.values()[0]
self.assertEqual('none', nvers)
r = self.client.create_container(metadata={'m1': 'v1', 'm2': 'v2'})
self.assertTrue(isinstance(r, dict))
r = self.client.get_container_meta(self.client.container)
self.assertTrue('x-container-meta-m1' in r)
self.assertEqual(r['x-container-meta-m1'], 'v1')
self.assertTrue('x-container-meta-m2' in r)
self.assertEqual(r['x-container-meta-m2'], 'v2')
r = self.client.container_put(metadata={'m1': '', 'm2': 'v2a'})
self.assertEqual(r.status_code, 202)
r = self.client.get_container_meta(self.client.container)
self.assertTrue('x-container-meta-m1' not in r)
self.assertTrue('x-container-meta-m2' in r)
self.assertEqual(r['x-container-meta-m2'], 'v2a')
self.client.del_container_meta(self.client.container)
def test_container_post(self):
"""Test container_POST"""
self._test_0060_container_post()
def _test_0060_container_post(self):
self.client.container = self.c2
"""Simple post"""
r = self.client.container_post()
self.assertEqual(r.status_code, 202)
"""post meta"""
self.client.set_container_meta({'m1': 'v1', 'm2': 'v2'})
r = self.client.get_container_meta(self.client.container)
self.assertTrue('x-container-meta-m1' in r)
self.assertEqual(r['x-container-meta-m1'], 'v1')
self.assertTrue('x-container-meta-m2' in r)
self.assertEqual(r['x-container-meta-m2'], 'v2')
"""post/2del meta"""
r = self.client.del_container_meta('m1')
r = self.client.set_container_meta({'m2': 'v2a'})
r = self.client.get_container_meta(self.client.container)
self.assertTrue('x-container-meta-m1' not in r)
self.assertTrue('x-container-meta-m2' in r)
self.assertEqual(r['x-container-meta-m2'], 'v2a')
"""check quota"""
r = self.client.get_container_limit(self.client.container)
cquota = r.values()[0]
newquota = 2 * int(cquota)
r = self.client.set_container_limit(newquota)
r = self.client.get_container_limit(self.client.container)
xquota = int(r.values()[0])
self.assertEqual(newquota, xquota)
r = self.client.set_container_limit(cquota)
r = self.client.get_container_limit(self.client.container)
xquota = r.values()[0]
self.assertEqual(cquota, xquota)
"""Check versioning"""
self.client.set_container_versioning('auto')
r = self.client.get_container_versioning(self.client.container)
nvers = r.values()[0]
self.assertEqual('auto', nvers)
self.client.set_container_versioning('none')
r = self.client.get_container_versioning(self.client.container)
nvers = r.values()[0]
self.assertEqual('none', nvers)
"""put_block uses content_type and content_length to
post blocks of data 2 container. All that in upload_object"""
"""Change a file at fs"""
f = self.create_large_file(1024 * 1024 * 100)
"""Upload it at a directory in container"""
self.client.create_directory('dir')
r = self.client.upload_object('/dir/sample.file', f)
for term in ('content-length', 'content-type', 'x-object-version'):
self.assertTrue(term in r)
"""Check if file has been uploaded"""
r = self.client.get_object_info('/dir/sample.file')
self.assertTrue(int(r['content-length']) > 100000000)
"""What is tranfer_encoding? What should I check about it? """
#TODO
"""Check update=False"""
r = self.client.object_post(
'test',
update=False,
metadata={'newmeta': 'newval'})
r = self.client.get_object_info('test')
self.assertTrue('x-object-meta-newmeta' in r)
self.assertFalse('x-object-meta-incontainer' in r)
r = self.client.del_container_meta('m2')
def test_container_delete(self):
"""Test container_DELETE"""
self._test_0070_container_delete()
def _test_0070_container_delete(self):
"""Fail to delete a non-empty container"""
self.client.container = self.c2
r = self.client.container_delete(success=409)
self.assertEqual(r.status_code, 409)
"""Fail to delete c3 (empty) container"""
self.client.container = self.c3
r = self.client.container_delete(until='1000000000')
self.assertEqual(r.status_code, 204)
"""Delete c3 (empty) container"""
r = self.client.container_delete()
self.assertEqual(r.status_code, 204)
"""Purge container(empty a container), check versionlist"""
self.client.container = self.c1
r = self.client.object_head('test', success=(200, 404))
self.assertEqual(r.status_code, 200)
self.client.del_container(delimiter='/')
r = self.client.object_head('test', success=(200, 404))
self.assertEqual(r.status_code, 404)
r = self.client.get_object_versionlist('test')
self.assertTrue(len(r) > 0)
self.assertTrue(len(r[0]) > 1)
self.client.purge_container()
self.assertRaises(
ClientError, self.client.get_object_versionlist, 'test')
def _test_0080_recreate_deleted_data(self):
self._init_data()
def test_object_head(self):
"""Test object_HEAD"""
self._test_0090_object_head()
def _test_0090_object_head(self):
self.client.container = self.c2
obj = 'test'
r = self.client.object_head(obj)
self.assertEqual(r.status_code, 200)
etag = r.headers['etag']
real_version = r.headers['x-object-version']
self.assertRaises(
ClientError, self.client.object_head, obj, version=-10)
r = self.client.object_head(obj, version=real_version)
self.assertEqual(r.headers['x-object-version'], real_version)
r = self.client.object_head(obj, if_etag_match=etag)
self.assertEqual(r.status_code, 200)
r = self.client.object_head(
obj, if_etag_not_match=etag, success=(200, 412, 304))
self.assertNotEqual(r.status_code, 200)
r = self.client.object_head(
obj, version=real_version, if_etag_match=etag, success=200)
self.assertEqual(r.status_code, 200)
"""Check and if(un)modified_since"""
for format in self.client.DATE_FORMATS:
now_formated = self.now_unformated.strftime(format)
r1 = self.client.object_head(
obj, if_modified_since=now_formated, success=(200, 304, 412))
sc1 = r1.status_code
r2 = self.client.object_head(
obj, if_unmodified_since=now_formated, success=(200, 304, 412))
sc2 = r2.status_code
self.assertNotEqual(sc1, sc2)
def test_object_get(self):
"""Test object_GET"""
self._test_0100_object_get()
def _test_0100_object_get(self):
self.client.container = self.c1
obj = 'test'
r = self.client.object_get(obj)
self.assertEqual(r.status_code, 200)
osize = int(r.headers['content-length'])
etag = r.headers['etag']
r = self.client.object_get(obj, hashmap=True)
for term in ('hashes', 'block_hash', 'block_hash', 'bytes'):
self.assertTrue(term in r.json)
r = self.client.object_get(obj, format='xml', hashmap=True)
self.assertEqual(len(r.text.split('hash>')), 3)
rangestr = 'bytes=%s-%s' % (osize / 3, osize / 2)
r = self.client.object_get(
obj, data_range=rangestr, success=(200, 206))
partsize = int(r.headers['content-length'])
self.assertTrue(0 < partsize and partsize <= 1 + osize / 3)
rangestr = 'bytes=%s-%s' % (osize / 3, osize / 2)
r = self.client.object_get(
obj, data_range=rangestr, if_range=True, success=(200, 206))
partsize = int(r.headers['content-length'])
self.assertTrue(0 < partsize and partsize <= 1 + osize / 3)
r = self.client.object_get(obj, if_etag_match=etag)
self.assertEqual(r.status_code, 200)
r = self.client.object_get(obj, if_etag_not_match=etag + 'LALALA')
self.assertEqual(r.status_code, 200)
"""Check and if(un)modified_since"""
for format in self.client.DATE_FORMATS:
now_formated = self.now_unformated.strftime(format)
r1 = self.client.object_get(
obj, if_modified_since=now_formated, success=(200, 304, 412))
sc1 = r1.status_code
r2 = self.client.object_get(
obj, if_unmodified_since=now_formated, success=(200, 304, 412))
sc2 = r2.status_code
self.assertNotEqual(sc1, sc2)
"""Upload an object to download"""
container_info_cache = dict()
trg_fname = 'remotefile_%s' % self.now
f_size = 59247824
src_f = self.create_large_file(f_size)
print('\tUploading...')
r = self.client.upload_object(
trg_fname, src_f, container_info_cache=container_info_cache)
print('\tDownloading...')
self.files.append(NamedTemporaryFile())
dnl_f = self.files[-1]
self.client.download_object(trg_fname, dnl_f)
print('\tCheck if files match...')
for pos in (0, f_size / 2, f_size - 128):
src_f.seek(pos)
dnl_f.seek(pos)
self.assertEqual(src_f.read(64), dnl_f.read(64))
print('\tDownload KiBs to string and check again...')
for pos in (0, f_size / 2, f_size - 256):
src_f.seek(pos)
tmp_s = self.client.download_to_string(
trg_fname, range_str='%s-%s' % (pos, (pos + 128)))
self.assertEqual(tmp_s, src_f.read(len(tmp_s)))
print('\tUploading KiBs as strings...')
trg_fname = 'fromString_%s' % self.now
src_size = 2 * 1024
src_f.seek(0)
src_str = src_f.read(src_size)
self.client.upload_from_string(trg_fname, src_str)
print('\tDownload as string and check...')
tmp_s = self.client.download_to_string(trg_fname)
self.assertEqual(tmp_s, src_str)
"""Upload a boring file"""
trg_fname = 'boringfile_%s' % self.now
src_f = self.create_boring_file(42)
print('\tUploading boring file...')
self.client.upload_object(
trg_fname, src_f, container_info_cache=container_info_cache)
print('\tDownloading boring file...')
self.files.append(NamedTemporaryFile())
dnl_f = self.files[-1]
self.client.download_object(trg_fname, dnl_f)
print('\tCheck if files match...')
for i in range(42):
self.assertEqual(sample_block(src_f, i), sample_block(dnl_f, i))
def test_object_put(self):
"""Test object_PUT"""
self._test_0150_object_put()
def _test_0150_object_put(self):
self.client.container = self.c2
obj = 'another.test'
self.client.create_object(obj + '.FAKE')
r = self.client.get_object_info(obj + '.FAKE')
self.assertEqual(r['content-type'], 'application/octet-stream')
"""create the object"""
r = self.client.object_put(
obj,
data='a',
content_type='application/octer-stream',
permissions=dict(
read=['accX:groupA', 'u1', 'u2'],
write=['u2', 'u3']),
metadata=dict(key1='val1', key2='val2'),
content_encoding='UTF-8',
content_disposition='attachment; filename="fname.ext"')
self.assertEqual(r.status_code, 201)
etag = r.headers['etag']
"""Check content-disposition"""
r = self.client.get_object_info(obj)
self.assertTrue('content-disposition' in r)
"""Check permissions"""
r = self.client.get_object_sharing(obj)
self.assertTrue('accx:groupa' in r['read'])
self.assertTrue('u1' in r['read'])
self.assertTrue('u2' in r['write'])
self.assertTrue('u3' in r['write'])
"""Check metadata"""
r = self.client.get_object_meta(obj)
self.assertEqual(r['x-object-meta-key1'], 'val1')
self.assertEqual(r['x-object-meta-key2'], 'val2')
"""Check public and if_etag_match"""
r = self.client.object_put(
obj,
if_etag_match=etag,
data='b',
content_type='application/octet-stream',
public=True)
r = self.client.object_get(obj)
self.assertTrue('x-object-public' in r.headers)
vers2 = int(r.headers['x-object-version'])
etag = r.headers['etag']
self.assertEqual(r.text, 'b')
"""Check if_etag_not_match"""
r = self.client.object_put(
obj,
if_etag_not_match=etag,
data='c',
content_type='application/octet-stream',
success=(201, 412))
self.assertEqual(r.status_code, 412)
"""Check content_type and content_length"""
tmpdir = 'dir' + unicode(self.now)
r = self.client.object_put(
tmpdir, content_type='application/directory', content_length=0)
r = self.client.get_object_info(tmpdir)
self.assertEqual(r['content-type'], 'application/directory')
"""Check copy_from, content_encoding"""
r = self.client.object_put(
'%s/%s' % (tmpdir, obj),
format=None,
copy_from='/%s/%s' % (self.client.container, obj),
content_encoding='application/octet-stream',
source_account=self.client.account,
content_length=0,
success=201)
self.assertEqual(r.status_code, 201)
"""Test copy_object for cross-conctainer copy"""
self.client.copy_object(
src_container=self.c2,
src_object='%s/%s' % (tmpdir, obj),
dst_container=self.c1,
dst_object=obj)
self.client.container = self.c1
r1 = self.client.get_object_info(obj)
self.client.container = self.c2
r2 = self.client.get_object_info('%s/%s' % (tmpdir, obj))
self.assertEqual(r1['x-object-hash'], r2['x-object-hash'])
"""Check cross-container copy_from, content_encoding"""
self.client.container = self.c1
fromstr = '/%s/%s/%s' % (self.c2, tmpdir, obj)
r = self.client.object_put(
obj,
format=None,
copy_from=fromstr,
content_encoding='application/octet-stream',
source_account=self.client.account,
content_length=0,
success=201)
self.assertEqual(r.status_code, 201)
r = self.client.get_object_info(obj)
self.assertEqual(r['etag'], etag)
"""Check source_account"""
self.client.container = self.c2
fromstr = '/%s/%s' % (self.c1, obj)
r = self.client.object_put(
'%sv2' % obj,
format=None,
move_from=fromstr,
content_encoding='application/octet-stream',
source_account='*****@*****.**',
content_length=0,
success=(201, 403))
self.assertEqual(r.status_code, 403)
"""Check cross-container move_from"""
self.client.container = self.c1
r1 = self.client.get_object_info(obj)
self.client.container = self.c2
self.client.move_object(
src_container=self.c1,
src_object=obj,
dst_container=self.c2,
dst_object=obj + 'v0')
r0 = self.client.get_object_info(obj + 'v0')
self.assertEqual(r1['x-object-hash'], r0['x-object-hash'])
"""Check move_from"""
r = self.client.object_put(
'%sv1' % obj,
format=None,
move_from='/%s/%s' % (self.c2, obj),
source_version=vers2,
content_encoding='application/octet-stream',
content_length=0, success=201)
"""Check manifest"""
mobj = 'manifest.test'
txt = ''
for i in range(10):
txt += '%s' % i
r = self.client.object_put(
'%s/%s' % (mobj, i),
data='%s' % i,
content_length=1,
success=201,
content_type='application/octet-stream',
content_encoding='application/octet-stream')
r = self.client.object_put(
mobj,
content_length=0,
content_type='application/octet-stream',
manifest='%s/%s' % (self.client.container, mobj))
r = self.client.object_get(mobj)
self.assertEqual(r.text, txt)
"""Upload a local file with one request"""
newf = self.create_large_file(1024 * 10)
self.client.upload_object('sample.file', newf)
"""Check if file has been uploaded"""
r = self.client.get_object_info('sample.file')
self.assertEqual(int(r['content-length']), 10240)
"""Some problems with transfer-encoding?"""
def test_object_copy(self):
"""Test object_COPY"""
self._test_0110_object_copy()
def _test_0110_object_copy(self):
# TODO: check with source_account option
self.client.container = self.c2
obj = 'test2'
data = '{"key1":"val1", "key2":"val2"}'
r = self.client.object_put(
'%sorig' % obj,
content_type='application/octet-stream',
data=data,
metadata=dict(mkey1='mval1', mkey2='mval2'),
permissions=dict(
read=['accX:groupA', 'u1', 'u2'],
write=['u2', 'u3']),
content_disposition='attachment; filename="fname.ext"')
r = self.client.object_copy(
'%sorig' % obj,
destination='/%s/%s' % (self.client.container, obj),
ignore_content_type=False, content_type='application/json',
metadata={'mkey2': 'mval2a', 'mkey3': 'mval3'},
permissions={'write': ['u5', 'accX:groupB']})
self.assertEqual(r.status_code, 201)
"""Check content-disposition"""
r = self.client.get_object_info(obj)
self.assertTrue('content-disposition' in r)
"""Check Metadata"""
r = self.client.get_object_meta(obj)
self.assertEqual(r['x-object-meta-mkey1'], 'mval1')
self.assertEqual(r['x-object-meta-mkey2'], 'mval2a')
self.assertEqual(r['x-object-meta-mkey3'], 'mval3')
"""Check permissions"""
r = self.client.get_object_sharing(obj)
self.assertFalse('read' in r or 'u2' in r['write'])
self.assertTrue('accx:groupb' in r['write'])
"""Check destination account"""
r = self.client.object_copy(
obj,
destination='/%s/%s' % (self.c1, obj),
content_encoding='utf8',
content_type='application/json',
destination_account='*****@*****.**',
success=(201, 404))
self.assertEqual(r.status_code, 404)
"""Check destination being another container
and also content_type and content encoding"""
r = self.client.object_copy(
obj,
destination='/%s/%s' % (self.c1, obj),
content_encoding='utf8',
content_type='application/json')
self.assertEqual(r.status_code, 201)
self.assertEqual(
r.headers['content-type'],
'application/json; charset=UTF-8')
"""Check ignore_content_type and content_type"""
r = self.client.object_get(obj)
etag = r.headers['etag']
ctype = r.headers['content-type']
self.assertEqual(ctype, 'application/json')
r = self.client.object_copy(
'%sorig' % obj,
destination='/%s/%s0' % (self.client.container, obj),
ignore_content_type=True,
content_type='application/json')
self.assertEqual(r.status_code, 201)
self.assertNotEqual(r.headers['content-type'], 'application/json')
"""Check if_etag_(not_)match"""
r = self.client.object_copy(
obj,
destination='/%s/%s1' % (self.client.container, obj),
if_etag_match=etag)
self.assertEqual(r.status_code, 201)
r = self.client.object_copy(
obj,
destination='/%s/%s2' % (self.client.container, obj),
if_etag_not_match='lalala')
self.assertEqual(r.status_code, 201)
vers2 = r.headers['x-object-version']
"""Check source_version, public and format """
r = self.client.object_copy(
'%s2' % obj,
destination='/%s/%s3' % (self.client.container, obj),
source_version=vers2,
format='xml',
public=True)
self.assertEqual(r.status_code, 201)
self.assertTrue(r.headers['content-type'].index('xml') > 0)
r = self.client.get_object_info(obj + '3')
self.assertTrue('x-object-public' in r)
def test_object_move(self):
"""Test object_MOVE"""
self._test_0120_object_move()
def _test_0120_object_move(self):
self.client.container = self.c2
obj = 'test2'
data = '{"key1": "val1", "key2": "val2"}'
r = self.client.object_put(
'%sorig' % obj,
content_type='application/octet-stream',
data=data,
metadata=dict(mkey1='mval1', mkey2='mval2'),
permissions=dict(
read=['accX:groupA', 'u1', 'u2'],
write=['u2', 'u3']))
r = self.client.object_move(
'%sorig' % obj,
destination='/%s/%s' % (self.client.container, obj),
ignore_content_type=False,
content_type='application/json',
metadata=dict(mkey2='mval2a', mkey3='mval3'),
permissions=dict(write=['u5', 'accX:groupB']))
self.assertEqual(r.status_code, 201)
"""Check Metadata"""
r = self.client.get_object_meta(obj)
self.assertEqual(r['x-object-meta-mkey1'], 'mval1')
self.assertEqual(r['x-object-meta-mkey2'], 'mval2a')
self.assertEqual(r['x-object-meta-mkey3'], 'mval3')
"""Check permissions"""
r = self.client.get_object_sharing(obj)
self.assertFalse('read' in r)
self.assertTrue('u5' in r['write'])
self.assertTrue('accx:groupb' in r['write'])
"""Check destination account"""
r = self.client.object_move(
obj,
destination='/%s/%s' % (self.c1, obj),
content_encoding='utf8',
content_type='application/json',
destination_account='*****@*****.**',
success=(201, 404))
self.assertEqual(r.status_code, 404)
"""Check destination being another container and also
content_type, content_disposition and content encoding"""
r = self.client.object_move(
obj,
destination='/%s/%s' % (self.c1, obj),
content_encoding='utf8',
content_type='application/json',
content_disposition='attachment; filename="fname.ext"')
self.assertEqual(r.status_code, 201)
self.assertEqual(
r.headers['content-type'],
'application/json; charset=UTF-8')
self.client.container = self.c1
r = self.client.get_object_info(obj)
self.assertTrue('content-disposition' in r)
self.assertTrue('fname.ext' in r['content-disposition'])
etag = r['etag']
ctype = r['content-type']
self.assertEqual(ctype, 'application/json')
"""Check ignore_content_type and content_type"""
r = self.client.object_move(
obj,
destination='/%s/%s' % (self.c2, obj),
ignore_content_type=True,
content_type='application/json')
self.assertEqual(r.status_code, 201)
self.assertNotEqual(r.headers['content-type'], 'application/json')
"""Check if_etag_(not_)match"""
self.client.container = self.c2
r = self.client.object_move(
obj,
destination='/%s/%s0' % (self.client.container, obj),
if_etag_match=etag)
self.assertEqual(r.status_code, 201)
r = self.client.object_move(
'%s0' % obj,
destination='/%s/%s1' % (self.client.container, obj),
if_etag_not_match='lalala')
self.assertEqual(r.status_code, 201)
"""Check public and format """
r = self.client.object_move(
'%s1' % obj,
destination='/%s/%s2' % (self.client.container, obj),
format='xml',
public=True)
self.assertEqual(r.status_code, 201)
self.assertTrue(r.headers['content-type'].index('xml') > 0)
r = self.client.get_object_info(obj + '2')
self.assertTrue('x-object-public' in r)
def test_object_post(self):
"""Test object_POST"""
self._test_0130_object_post()
def _test_0130_object_post(self):
self.client.container = self.c2
obj = 'test2'
"""create a filesystem file"""
self.files.append(NamedTemporaryFile())
newf = self.files[-1]
newf.writelines([
'ello!\n',
'This is a test line\n',
'inside a test file\n'])
"""create a file on container"""
r = self.client.object_put(
obj,
content_type='application/octet-stream',
data='H',
metadata=dict(mkey1='mval1', mkey2='mval2'),
permissions=dict(
read=['accX:groupA', 'u1', 'u2'],
write=['u2', 'u3']))
"""Append livetest update, content_[range|type|length]"""
newf.seek(0)
self.client.append_object(obj, newf)
r = self.client.object_get(obj)
self.assertTrue(r.text.startswith('Hello!'))
"""Overwrite livetest update,
content_type, content_length, content_range
"""
newf.seek(0)
r = self.client.overwrite_object(obj, 0, 10, newf)
r = self.client.object_get(obj)
self.assertTrue(r.text.startswith('ello!'))
"""Truncate livetest update,
content_range, content_type, object_bytes and source_object"""
r = self.client.truncate_object(obj, 5)
r = self.client.object_get(obj)
self.assertEqual(r.text, 'ello!')
"""Check metadata"""
self.client.set_object_meta(obj, {'mkey2': 'mval2a', 'mkey3': 'mval3'})
r = self.client.get_object_meta(obj)
self.assertEqual(r['x-object-meta-mkey1'], 'mval1')
self.assertEqual(r['x-object-meta-mkey2'], 'mval2a')
self.assertEqual(r['x-object-meta-mkey3'], 'mval3')
self.client.del_object_meta(obj, 'mkey1')
r = self.client.get_object_meta(obj)
self.assertFalse('x-object-meta-mkey1' in r)
"""Check permissions"""
self.client.set_object_sharing(
obj, read_permission=['u4', 'u5'], write_permission=['u4'])
r = self.client.get_object_sharing(obj)
self.assertTrue('read' in r)
self.assertTrue('u5' in r['read'])
self.assertTrue('write' in r)
self.assertTrue('u4' in r['write'])
self.client.del_object_sharing(obj)
r = self.client.get_object_sharing(obj)
self.assertTrue(len(r) == 0)
"""Check publish"""
self.client.publish_object(obj)
r = self.client.get_object_info(obj)
self.assertTrue('x-object-public' in r)
self.client.unpublish_object(obj)
r = self.client.get_object_info(obj)
self.assertFalse('x-object-public' in r)
"""Check if_etag_(not)match"""
etag = r['etag']
r = self.client.object_post(
obj,
update=True,
public=True,
if_etag_not_match=etag,
success=(412, 202, 204))
#self.assertEqual(r.status_code, 412)
r = self.client.object_post(
obj,
update=True,
public=True,
if_etag_match=etag,
content_encoding='application/json')
r = self.client.get_object_info(obj)
helloVersion = r['x-object-version']
self.assertTrue('x-object-public' in r)
self.assertEqual(r['content-encoding'], 'application/json')
"""Check source_version and source_account and content_disposition"""
r = self.client.object_post(
obj,
update=True,
content_type='application/octet-srteam',
content_length=5,
content_range='bytes 1-5/*',
source_object='/%s/%s' % (self.c2, obj),
source_account='*****@*****.**',
source_version=helloVersion,
data='12345',
success=(403, 202, 204))
self.assertEqual(r.status_code, 403)
r = self.client.object_post(
obj,
update=True,
content_type='application/octet-srteam',
content_length=5,
content_range='bytes 1-5/*',
source_object='/%s/%s' % (self.c2, obj),
source_account=self.client.account,
source_version=helloVersion,
data='12345',
content_disposition='attachment; filename="fname.ext"')
r = self.client.object_get(obj)
self.assertEqual(r.text, 'eello!')
self.assertTrue('content-disposition' in r.headers)
self.assertTrue('fname.ext' in r.headers['content-disposition'])
"""Check manifest"""
mobj = 'manifest.test'
txt = ''
for i in range(10):
txt += '%s' % i
r = self.client.object_put(
'%s/%s' % (mobj, i),
data='%s' % i,
content_length=1,
success=201,
content_encoding='application/octet-stream',
content_type='application/octet-stream')
self.client.create_object_by_manifestation(
mobj, content_type='application/octet-stream')
r = self.client.object_post(
mobj, manifest='%s/%s' % (self.client.container, mobj))
r = self.client.object_get(mobj)
self.assertEqual(r.text, txt)
"""We need to check transfer_encoding """
def test_object_delete(self):
"""Test object_DELETE"""
self._test_0140_object_delete()
def _test_0140_object_delete(self):
self.client.container = self.c2
obj = 'test2'
"""create a file on container"""
r = self.client.object_put(
obj,
content_type='application/octet-stream',
data='H',
metadata=dict(mkey1='mval1', mkey2='mval2'),
permissions=dict(
read=['accX:groupA', 'u1', 'u2'],
write=['u2', 'u3']))
"""Check with false until"""
r = self.client.object_delete(obj, until=1000000)
r = self.client.object_get(obj, success=(200, 404))
self.assertEqual(r.status_code, 200)
"""Check normal case"""
r = self.client.object_delete(obj)
self.assertEqual(r.status_code, 204)
r = self.client.object_get(obj, success=(200, 404))
self.assertEqual(r.status_code, 404)
def create_large_file(self, size):
"""Create a large file at fs"""
print
self.files.append(NamedTemporaryFile())
f = self.files[-1]
Ki = size / 8
bytelist = [b * Ki for b in range(size / Ki)]
def append2file(step):
f.seek(step)
f.write(urandom(Ki))
f.flush()
self.do_with_progress_bar(
append2file,
' create rand file %s (%sB): ' % (f.name, size),
bytelist)
f.seek(0)
return f
def create_boring_file(self, num_of_blocks):
"""Create a file with some blocks being the same"""
self.files.append(NamedTemporaryFile())
tmpFile = self.files[-1]
block_size = 4 * 1024 * 1024
print('\n\tCreate boring file of %s blocks' % num_of_blocks)
chars = chargen()
while num_of_blocks:
fslice = 3 if num_of_blocks > 3 else num_of_blocks
tmpFile.write(fslice * block_size * chars.next())
num_of_blocks -= fslice
print('\t\tDone')
tmpFile.seek(0)
return tmpFile
