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)
