def cleanup(self): super(run, self).cleanup() # Clean up all containers dc = DockerContainers(self) for cobj in dc.list_containers(): self.logwarning("Found leftover container: %s", cobj.container_name) try: dc.kill_container_by_obj(cobj) except ValueError: pass # already dead else: self.logdebug("Killed container %s, waiting up to %d seconds " "for it to exit", cobj.container_name, dc.timeout) dc.wait_by_obj(cobj) self.logdebug("Removing container %s", cobj.container_name) dc.remove_by_obj(cobj) # Clean up all non-default images fqin = DockerImage.full_name_from_defaults(self.config) di = DockerImages(self) def_img_obj = di.list_imgs_with_full_name(fqin)[0] for img_obj in di.list_imgs(): if img_obj.full_name != def_img_obj.full_name: self.logwarning("Found leftover image: %s", img_obj) di.remove_image_by_image_obj(img_obj) else: self.logdebug("Not removing default image: %s", def_img_obj)
def lookup_image_id(self, image_name, image_tag): di = DockerImages(self.parent_subtest) fqin = DockerImage.full_name_from_component(image_name, image_tag) imglst = di.list_imgs_with_full_name(fqin) try: # Don't care about duplicate ID's return imglst[0].long_id except IndexError: return None # expected by some sub-subtests
def lookup_image_id(self, image_name, image_tag): di = DockerImages(self) fqin = DockerImage.full_name_from_component(image_name, image_tag) imglst = di.list_imgs_with_full_name(fqin) try: # Don't care about duplicate ID's return imglst[0].long_id except IndexError: return None # expected by some sub-subtests
def run_once(self): fqin = self.sub_stuff['fqin'] self.sub_stuff["images"].append(fqin) di = DockerImages(self.parent_subtest) try: images = di.list_imgs_with_full_name(fqin) for i in images: di.remove_image_by_id(i.long_id) except error.CmdError: pass # removal was the goal images = di.list_imgs_with_full_name(fqin) if images != []: error.TestNAError("Unable to prepare env for test:" " image %s already/still" " exist in docker repository", fqin) self.logdebug("Existing images: %s", di.list_imgs_full_name()) self.loginfo("Executing test commands") self.run_command('run', self.sub_stuff['subargs'], 'cmdresult_run1') self.run_command('rm', [self.sub_stuff["rand_name"]], 'cmdresult_rm') self.run_command('tag', [fqin, self.sub_stuff["rand_name"].lower()], 'cmdresult_tag') self.run_command('rmi', [fqin], 'cmdresult_rmi') self.run_command('run', self.sub_stuff['subargs'], 'cmdresult_run2') self.run_command('rm', [self.sub_stuff["rand_name"]], 'cmdresult_rm2') self.run_command('rmi', [self.sub_stuff["rand_name"].lower()], 'cmdresult_rmi2')
def postprocess(self): super(pull_base, self).postprocess() self.outputcheck() if self.config["docker_expected_result"] == "PASS": self.failif(self.sub_stuff['cmdresult'].exit_status != 0, "Non-zero pull exit status: %s" % self.sub_stuff['cmdresult']) di = DockerImages(self.parent_subtest) image_list = di.list_imgs_with_full_name(self.sub_stuff["img_fn"]) self.sub_stuff['image_list'] = image_list self.failif(self.sub_stuff['image_list'] == [], "Failed to look up image ") elif self.config["docker_expected_result"] == "FAIL": self.failif(self.sub_stuff['cmdresult'].exit_status == 0, "Zero pull exit status: Command should fail due to" " wrong command arguments.")
def check_image_exists(self, full_name): di = DockerImages(self) return di.list_imgs_with_full_name(full_name)
def check_image_exists(self, full_name): di = DockerImages(self) return di.list_imgs_with_full_name(full_name)
class tag_base(SubSubtest): """ tag base class """ def __init__(self, *args, **kwargs): super(tag_base, self).__init__(*args, **kwargs) self.dkrimg = DockerImages(self.parent_subtest) self.sub_stuff['tmp_image_list'] = set() def get_images_by_name(self, full_name): """ :return: List of images with given name """ return self.dkrimg.list_imgs_with_full_name(full_name) def prep_image(self, base_image): """ Tag the dockertest image to this test name """ NoFailDockerCmd(self, "pull", [base_image], verbose=False).execute() subargs = [base_image, self.sub_stuff["image"]] tag_results = DockerCmd(self, "tag", subargs, verbose=False).execute() if tag_results.exit_status: raise xceptions.DockerTestNAError("Problems during " "initialization of" " test: %s", tag_results) img = self.get_images_by_name(self.sub_stuff["image"]) self.failif(not img, "Image %s was not created." % self.sub_stuff["image"]) self.sub_stuff['image_list'] = img def initialize(self): super(tag_base, self).initialize() config.none_if_empty(self.config) self.dkrimg.gen_lower_only = self.config['gen_lower_only'] new_img_name = self.dkrimg.get_unique_name() self.sub_stuff["image"] = new_img_name base_image = DockerImage.full_name_from_defaults(self.config) self.prep_image(base_image) def complete_docker_command_line(self): """ :return: tag subargs using new_image_name """ force = self.config["tag_force"] cmd = [] if force: cmd.append("-f") cmd.append(self.sub_stuff["image"]) cmd.append(self.sub_stuff["new_image_name"]) self.sub_stuff["tag_cmd"] = cmd return cmd def run_once(self): super(tag_base, self).run_once() subargs = self.complete_docker_command_line() self.sub_stuff["cmdresult"] = DockerCmd(self, 'tag', subargs).execute() def postprocess(self): super(tag_base, self).postprocess() if self.config["docker_expected_result"] == "PASS": # Raise exception if problems found OutputGood(self.sub_stuff['cmdresult']) self.failif(self.sub_stuff['cmdresult'].exit_status != 0, "Non-zero tag exit status: %s" % self.sub_stuff['cmdresult']) img = self.get_images_by_name(self.sub_stuff["new_image_name"]) # Needed for cleanup self.sub_stuff['image_list'] += img self.failif(len(img) < 1, "Failed to look up tagted image ") elif self.config["docker_expected_result"] == "FAIL": chck = OutputGood(self.sub_stuff['cmdresult'], ignore_error=True) exit_code = self.sub_stuff['cmdresult'].exit_status self.failif(not chck or not exit_code, "Zero tag exit status: Command should fail due to" " wrong command arguments.") else: self.failif(True, "Improper 'docker_expected_result' value %s" % self.config["docker_expected_result"]) def cleanup(self): super(tag_base, self).cleanup() # Auto-converts "yes/no" to a boolean if self.config['remove_after_test'] and 'image_list' in self.sub_stuff: for image in self.sub_stuff["image_list"]: self.logdebug("Removing image %s", image.full_name) try: self.dkrimg.remove_image_by_full_name(image.full_name) except error.CmdError, exc: err = exc.result_obj.stderr if "tagged in multiple repositories" not in err: raise self.loginfo("Successfully removed test image: %s", image.full_name) for image in self.sub_stuff['tmp_image_list']: image = self.get_images_by_name(image) if image: self.logdebug("Removing image %s", image[0].full_name) self.dkrimg.remove_image_by_full_name(image[0].full_name) self.loginfo("Successfully removed test image: %s", image[0].full_name)
class tag_base(SubSubtest): """ tag base class """ def __init__(self, *args, **kwargs): super(tag_base, self).__init__(*args, **kwargs) self.dkrimg = DockerImages(self) self.sub_stuff['tmp_image_list'] = set() self._expect_pass = True def expect_pass(self, set_to=None): """ Most of the time we expect our 'docker tag' command to succeed. This method can be used (as a setter) to set an explicit expectation or (as a getter) when checking results. """ if set_to is not None: self._expect_pass = set_to return self._expect_pass def get_images_by_name(self, full_name): """ :return: List of images with given name """ return self.dkrimg.list_imgs_with_full_name(full_name) def prep_image(self, base_image): """ Tag the dockertest image to this test name """ mustpass(DockerCmd(self, "pull", [base_image]).execute()) subargs = [base_image, self.sub_stuff["image"]] tag_results = DockerCmd(self, "tag", subargs).execute() if tag_results.exit_status: raise xceptions.DockerTestNAError("Problems during " "initialization of" " test: %s", tag_results) img = self.get_images_by_name(self.sub_stuff["image"]) self.failif(not img, "Image %s was not created." % self.sub_stuff["image"]) self.sub_stuff['image_list'] = img def initialize(self): super(tag_base, self).initialize() config.none_if_empty(self.config) self.dkrimg.gen_lower_only = self.config['gen_lower_only'] new_img_name = self.dkrimg.get_unique_name() self.sub_stuff["image"] = new_img_name base_image = DockerImage.full_name_from_defaults(self.config) self.prep_image(base_image) def complete_docker_command_line(self): """ :return: tag subargs using new_image_name """ cmd = [] if 'force_tag' in self.sub_stuff and self.sub_stuff['force_tag']: cmd.append("-f") cmd.append(self.sub_stuff["image"]) cmd.append(self.sub_stuff["new_image_name"]) self.sub_stuff["tag_cmd"] = cmd return cmd def run_once(self): super(tag_base, self).run_once() subargs = self.complete_docker_command_line() self.sub_stuff["cmdresult"] = DockerCmd(self, 'tag', subargs).execute() def postprocess(self): super(tag_base, self).postprocess() expect_pass = self.expect_pass() OutputGood(self.sub_stuff['cmdresult'], ignore_error=not expect_pass) if expect_pass: # Raise exception if problems found self.failif_ne(self.sub_stuff['cmdresult'].exit_status, 0, "Non-zero tag exit status: %s" % self.sub_stuff['cmdresult']) img = self.get_images_by_name(self.sub_stuff["new_image_name"]) # Needed for cleanup self.sub_stuff['image_list'] += img self.failif(len(img) < 1, "Failed to look up tagged image ") else: self.failif(self.sub_stuff['cmdresult'].exit_status == 0, "Was expecting tag command to fail, but it exited OK") def cleanup(self): super(tag_base, self).cleanup() # Auto-converts "yes/no" to a boolean if self.config['remove_after_test'] and 'image_list' in self.sub_stuff: for image in self.sub_stuff["image_list"]: self.logdebug("Removing image %s", image.full_name) try: self.dkrimg.remove_image_by_full_name(image.full_name) except error.CmdError, exc: err = exc.result_obj.stderr if "tagged in multiple repositories" not in err: raise self.loginfo("Successfully removed test image: %s", image.full_name) for image_name in self.sub_stuff['tmp_image_list']: image = self.get_images_by_name(image_name) if image: self.logdebug("Removing tmp image %s", image[0].full_name) self.dkrimg.remove_image_by_full_name(image[0].full_name) self.loginfo("Successfully removed test image: %s", image[0].full_name)
class tag_base(SubSubtest): """ tag base class """ def __init__(self, *args, **kwargs): super(tag_base, self).__init__(*args, **kwargs) self.dkrimg = DockerImages(self) self.sub_stuff['tmp_image_list'] = set() self._expect_pass = True def expect_pass(self, set_to=None): """ Most of the time we expect our 'docker tag' command to succeed. This method can be used (as a setter) to set an explicit expectation or (as a getter) when checking results. """ if set_to is not None: self._expect_pass = set_to return self._expect_pass def get_images_by_name(self, full_name): """ :return: List of images with given name """ return self.dkrimg.list_imgs_with_full_name(full_name) def prep_image(self, base_image): """ Tag the dockertest image to this test name """ mustpass(DockerCmd(self, "pull", [base_image]).execute()) subargs = [base_image, self.sub_stuff["image"]] tag_results = DockerCmd(self, "tag", subargs).execute() if tag_results.exit_status: raise xceptions.DockerTestNAError( "Problems during " "initialization of" " test: %s", tag_results) img = self.get_images_by_name(self.sub_stuff["image"]) self.failif(not img, "Image %s was not created." % self.sub_stuff["image"]) self.sub_stuff['image_list'] = img def initialize(self): super(tag_base, self).initialize() config.none_if_empty(self.config) self.dkrimg.gen_lower_only = self.config['gen_lower_only'] new_img_name = self.dkrimg.get_unique_name() self.sub_stuff["image"] = new_img_name base_image = DockerImage.full_name_from_defaults(self.config) self.prep_image(base_image) def complete_docker_command_line(self): """ :return: tag subargs using new_image_name """ cmd = [] if 'force_tag' in self.sub_stuff and self.sub_stuff['force_tag']: cmd.append("-f") cmd.append(self.sub_stuff["image"]) cmd.append(self.sub_stuff["new_image_name"]) self.sub_stuff["tag_cmd"] = cmd return cmd def run_once(self): super(tag_base, self).run_once() subargs = self.complete_docker_command_line() self.sub_stuff["cmdresult"] = DockerCmd(self, 'tag', subargs).execute() def postprocess(self): super(tag_base, self).postprocess() expect_pass = self.expect_pass() OutputGood(self.sub_stuff['cmdresult'], ignore_error=not expect_pass) if expect_pass: # Raise exception if problems found self.failif_ne( self.sub_stuff['cmdresult'].exit_status, 0, "Non-zero tag exit status: %s" % self.sub_stuff['cmdresult']) img = self.get_images_by_name(self.sub_stuff["new_image_name"]) # Needed for cleanup self.sub_stuff['image_list'] += img self.failif(len(img) < 1, "Failed to look up tagged image ") else: self.failif(self.sub_stuff['cmdresult'].exit_status == 0, "Was expecting tag command to fail, but it exited OK") def cleanup(self): super(tag_base, self).cleanup() # Auto-converts "yes/no" to a boolean if self.config['remove_after_test'] and 'image_list' in self.sub_stuff: for image in self.sub_stuff["image_list"]: self.logdebug("Removing image %s", image.full_name) try: self.dkrimg.remove_image_by_full_name(image.full_name) except error.CmdError, exc: err = exc.result_obj.stderr if "tagged in multiple repositories" not in err: raise self.loginfo("Successfully removed test image: %s", image.full_name) for image_name in self.sub_stuff['tmp_image_list']: image = self.get_images_by_name(image_name) if image: self.logdebug("Removing tmp image %s", image[0].full_name) self.dkrimg.remove_image_by_full_name(image[0].full_name) self.loginfo("Successfully removed test image: %s", image[0].full_name)
class tag_base(SubSubtest): """ tag base class """ def __init__(self, *args, **kwargs): super(tag_base, self).__init__(*args, **kwargs) self.dkrimg = DockerImages(self) self.sub_stuff['tmp_image_list'] = set() def get_images_by_name(self, full_name): """ :return: List of images with given name """ return self.dkrimg.list_imgs_with_full_name(full_name) def prep_image(self, base_image): """ Tag the dockertest image to this test name """ mustpass( DockerCmd(self, "pull", [base_image], verbose=False).execute()) subargs = [base_image, self.sub_stuff["image"]] tag_results = DockerCmd(self, "tag", subargs, verbose=False).execute() if tag_results.exit_status: raise xceptions.DockerTestNAError( "Problems during " "initialization of" " test: %s", tag_results) img = self.get_images_by_name(self.sub_stuff["image"]) self.failif(not img, "Image %s was not created." % self.sub_stuff["image"]) self.sub_stuff['image_list'] = img def initialize(self): super(tag_base, self).initialize() config.none_if_empty(self.config) self.dkrimg.gen_lower_only = self.config['gen_lower_only'] new_img_name = self.dkrimg.get_unique_name() self.sub_stuff["image"] = new_img_name base_image = DockerImage.full_name_from_defaults(self.config) self.prep_image(base_image) def complete_docker_command_line(self): """ :return: tag subargs using new_image_name """ force = self.config["tag_force"] cmd = [] if force: cmd.append("-f") cmd.append(self.sub_stuff["image"]) cmd.append(self.sub_stuff["new_image_name"]) self.sub_stuff["tag_cmd"] = cmd return cmd def run_once(self): super(tag_base, self).run_once() subargs = self.complete_docker_command_line() self.sub_stuff["cmdresult"] = DockerCmd(self, 'tag', subargs).execute() def postprocess(self): super(tag_base, self).postprocess() if self.config["docker_expected_result"] == "PASS": # Raise exception if problems found OutputGood(self.sub_stuff['cmdresult']) self.failif( self.sub_stuff['cmdresult'].exit_status != 0, "Non-zero tag exit status: %s" % self.sub_stuff['cmdresult']) img = self.get_images_by_name(self.sub_stuff["new_image_name"]) # Needed for cleanup self.sub_stuff['image_list'] += img self.failif(len(img) < 1, "Failed to look up tagted image ") elif self.config["docker_expected_result"] == "FAIL": chck = OutputGood(self.sub_stuff['cmdresult'], ignore_error=True) exit_code = self.sub_stuff['cmdresult'].exit_status self.failif( not chck or not exit_code, "Zero tag exit status: Command should fail due to" " wrong command arguments.") else: self.failif( True, "Improper 'docker_expected_result' value %s" % self.config["docker_expected_result"]) def cleanup(self): super(tag_base, self).cleanup() # Auto-converts "yes/no" to a boolean if self.config['remove_after_test'] and 'image_list' in self.sub_stuff: for image in self.sub_stuff["image_list"]: self.logdebug("Removing image %s", image.full_name) try: self.dkrimg.remove_image_by_full_name(image.full_name) except error.CmdError, exc: err = exc.result_obj.stderr if "tagged in multiple repositories" not in err: raise self.loginfo("Successfully removed test image: %s", image.full_name) for image in self.sub_stuff['tmp_image_list']: image = self.get_images_by_name(image) if image: self.logdebug("Removing image %s", image[0].full_name) self.dkrimg.remove_image_by_full_name(image[0].full_name) self.loginfo("Successfully removed test image: %s", image[0].full_name)