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