def test_tag_image_same_name(temp_image_name): # noqa
if MOCK:
mock_docker()
t = DockerTasker()
temp_image_name.registry = "somewhere.example.com"
temp_image_name.tag = "1"
flexmock(t.d).should_receive('tag').never()
t.tag_image(temp_image_name, temp_image_name.copy())
def test_tag_image_same_name(temp_image_name): # noqa
if MOCK:
mock_docker()
t = DockerTasker()
temp_image_name.registry = "somewhere.example.com"
temp_image_name.tag = "1"
flexmock(docker.APIClient).should_receive('tag').never()
t.tag_image(temp_image_name, temp_image_name.copy())
def test_push_image(temp_image_name):
if MOCK:
mock_docker()
t = DockerTasker()
temp_image_name.registry = LOCALHOST_REGISTRY
temp_image_name.tag = "1"
t.tag_image(INPUT_IMAGE, temp_image_name)
output = t.push_image(temp_image_name, insecure=True)
assert output is not None
t.remove_image(temp_image_name)
def test_push_image(temp_image_name):
if MOCK:
mock_docker()
t = DockerTasker()
temp_image_name.registry = LOCALHOST_REGISTRY
temp_image_name.tag = "1"
t.tag_image(INPUT_IMAGE, temp_image_name)
output = t.push_image(temp_image_name, insecure=True)
assert output is not None
t.remove_image(temp_image_name)
def test_push_image(temp_image_name, should_fail):
if MOCK:
mock_docker(push_should_fail=should_fail)
t = DockerTasker()
temp_image_name.registry = LOCALHOST_REGISTRY
temp_image_name.tag = "1"
t.tag_image(INPUT_IMAGE, temp_image_name)
if should_fail:
with pytest.raises(RuntimeError) as exc:
output = t.push_image(temp_image_name, insecure=True)
assert "Failed to push image" in str(exc)
else:
output = t.push_image(temp_image_name, insecure=True)
assert output is not None
t.remove_image(temp_image_name)
def test_push_image(temp_image_name, should_fail):
if MOCK:
mock_docker(push_should_fail=should_fail)
t = DockerTasker(retry_times=0)
temp_image_name.registry = LOCALHOST_REGISTRY
temp_image_name.tag = "1"
t.tag_image(INPUT_IMAGE, temp_image_name)
if should_fail:
with pytest.raises(RetryGeneratorException) as exc:
output = t.push_image(temp_image_name, insecure=True)
assert "Failed to mock_method image" in str(exc)
assert "connection refused" in str(exc)
else:
output = t.push_image(temp_image_name, insecure=True)
assert output is not None
t.remove_image(temp_image_name)
def test_push_image(temp_image_name, should_fail):
if MOCK:
mock_docker(push_should_fail=should_fail)
t = DockerTasker(retry_times=0)
temp_image_name.registry = LOCALHOST_REGISTRY
temp_image_name.tag = "1"
t.tag_image(INPUT_IMAGE, temp_image_name)
if should_fail:
with pytest.raises(RetryGeneratorException) as exc:
output = t.push_image(temp_image_name, insecure=True)
assert "Failed to mock_method image" in str(exc)
assert "connection refused" in str(exc)
else:
output = t.push_image(temp_image_name, insecure=True)
assert output is not None
t.remove_image(temp_image_name)
def test_tag_image(temp_image_name):
if MOCK:
mock_docker()
t = DockerTasker()
temp_image_name.registry = "somewhere.example.com"
temp_image_name.tag = "1"
img = t.tag_image(INPUT_IMAGE, temp_image_name)
try:
assert t.image_exists(temp_image_name)
assert img == temp_image_name.to_str()
finally:
t.remove_image(temp_image_name)
def test_tag_image(temp_image_name): # noqa
if MOCK:
mock_docker()
t = DockerTasker()
temp_image_name.registry = "somewhere.example.com"
temp_image_name.tag = "1"
img = t.tag_image(INPUT_IMAGE, temp_image_name)
try:
assert t.image_exists(temp_image_name)
assert img == temp_image_name.to_str()
finally:
t.remove_image(temp_image_name)
class InsideBuilder(LastLogger, BuilderStateMachine):
"""
This is expected to run within container
"""
def __init__(self, source, image, **kwargs):
"""
"""
LastLogger.__init__(self)
BuilderStateMachine.__init__(self)
self.tasker = DockerTasker()
# arguments for build
self.source = source
self.base_image_id = None
self.image_id = None
self.built_image_info = None
self.image = ImageName.parse(image)
# get info about base image from dockerfile
self.df_path, self.df_dir = self.source.get_dockerfile_path()
self.base_image = ImageName.parse(DockerfileParser(self.df_path).baseimage)
logger.debug("base image specified in dockerfile = '%s'", self.base_image)
if not self.base_image.tag:
self.base_image.tag = 'latest'
def pull_base_image(self, source_registry, insecure=False):
"""
pull base image
:param source_registry: str, registry to pull from
:param insecure: bool, allow connecting to registry over plain http
:return:
"""
logger.info("pulling base image '%s' from registry '%s'", self.base_image, source_registry)
self._ensure_not_built()
base_image_with_registry = self.base_image.copy()
if source_registry:
# registry in dockerfile doesn't match provided source registry
if self.base_image.registry and self.base_image.registry != source_registry:
logger.error("registry in dockerfile doesn't match provided source registry, "
"dockerfile = '%s', provided = '%s'",
self.base_image.registry, source_registry)
raise RuntimeError(
"Registry specified in dockerfile doesn't match provided one. Dockerfile: '%s', Provided: '%s'"
% (self.base_image.registry, source_registry))
base_image_with_registry.registry = source_registry
base_image = self.tasker.pull_image(base_image_with_registry, insecure=insecure)
pulled_tags = set([base_image])
if not self.base_image.registry:
response = self.tasker.tag_image(base_image_with_registry, self.base_image, force=True)
pulled_tags.add(response)
base_image = response
logger.debug("image '%s' is available", base_image)
return pulled_tags
def build(self):
"""
build image inside current environment;
it's expected this may run within (privileged) docker container
:return: image string (e.g. fedora-python:34)
"""
logger.info("building image '%s' inside current environment", self.image)
self._ensure_not_built()
logger.debug("using dockerfile:\n%s", DockerfileParser(self.df_path).content)
logs_gen = self.tasker.build_image_from_path(
self.df_dir,
self.image,
)
logger.debug("build is submitted, waiting for it to finish")
command_result = wait_for_command(logs_gen) # wait for build to finish
logger.info("build was %ssuccesful!", 'un' if command_result.is_failed() else '')
self.is_built = True
if not command_result.is_failed():
self.built_image_info = self.get_built_image_info()
# self.base_image_id = self.built_image_info['ParentId'] # parent id is not base image!
self.image_id = self.built_image_info['Id']
build_result = BuildResult(command_result, self.image_id)
return build_result
def push_built_image(self, registry, insecure=False):
"""
push built image to provided registry
:param registry: str
:param insecure: bool, allow connecting to registry over plain http
:return: str, image
"""
logger.info("pushing built image '%s' to registry '%s'", self.image, registry)
self._ensure_is_built()
if not registry:
logger.warning("no registry specified; skipping")
return
if self.image.registry and self.image.registry != registry:
logger.error("registry in image name doesn't match provided target registry, "
"image registry = '%s', target = '%s'",
self.image.registry, registry)
raise RuntimeError(
"Registry in image name doesn't match target registry. Image: '%s', Target: '%s'"
% (self.image.registry, registry))
target_image = self.image.copy()
target_image.registry = registry
response = self.tasker.tag_and_push_image(self.image, target_image, insecure=insecure)
self.tasker.remove_image(target_image)
return response
def inspect_base_image(self):
"""
inspect base image
:return: dict
"""
logger.info("inspecting base image '%s'", self.base_image)
inspect_data = self.tasker.inspect_image(self.base_image)
return inspect_data
def inspect_built_image(self):
"""
inspect built image
:return: dict
"""
logger.info("inspecting built image '%s'", self.image_id)
self._ensure_is_built()
inspect_data = self.tasker.inspect_image(self.image_id) # dict with lots of data, see man docker-inspect
return inspect_data
def get_base_image_info(self):
"""
query docker about base image
:return dict
"""
logger.info("getting information about base image '%s'", self.base_image)
image_info = self.tasker.get_image_info_by_image_name(self.base_image)
items_count = len(image_info)
if items_count == 1:
return image_info[0]
elif items_count <= 0:
logger.error("image '%s' not found", self.base_image)
raise RuntimeError("image '%s' not found", self.base_image)
else:
logger.error("multiple (%d) images found for image '%s'", items_count, self.base_image)
raise RuntimeError("multiple (%d) images found for image '%s'" % (items_count, self.base_image))
def get_built_image_info(self):
"""
query docker about built image
:return dict
"""
logger.info("getting information about built image '%s'", self.image)
self._ensure_is_built()
image_info = self.tasker.get_image_info_by_image_name(self.image)
items_count = len(image_info)
if items_count == 1:
return image_info[0]
elif items_count <= 0:
logger.error("image '%s' not found", self.image)
raise RuntimeError("image '%s' not found" % self.image)
else:
logger.error("multiple (%d) images found for image '%s'", items_count, self.image)
raise RuntimeError("multiple (%d) images found for image '%s'" % (items_count, self.image))
