def setUp(self):
da = DockerAgent(auto_clean_cnt=False)
self.test_dir = tempfile.mkdtemp()
self.test_img_name = 'atf_docker/test:latest'
self.test_img_name2 = 'johnklee/atf_docker_test2:latest'
self.hello_img_name = 'hello-world'
self.test_build_path = os.path.join(CUR_TESTDATA_DIR,
'docker/images/test/')
self.test_dockerfile_path = os.path.join(self.test_build_path,
'Dockerfile')
test_image_list = da.images(name=self.test_img_name)
if len(test_image_list) == 0:
logs = da.build(path=self.test_build_path,
dockerfile=self.test_dockerfile_path,
tag=self.test_img_name,
pull=True)
self.assertTrue(
len(logs.grep('Successfully tagged atf_docker/test:latest')) >
0, 'Unexpected logs:\n{}\n'.format(logs))
self.test_img = da.images(name=self.test_img_name)[0]
self.test_img_id = self.test_img['Id']
hello_image_list = da.images(name=self.hello_img_name)
if len(hello_image_list) > 0:
logs = da.remove_image(hello_image_list[0], force=True)
def test_api_remove_image(self):
build_path = os.path.join(CUR_TESTDATA_DIR, 'docker/images/test/')
dockerfile_path = os.path.join(build_path, 'Dockerfile')
da = DockerAgent()
# 0) Confirm that test image exist
test_image_list = da.images(name=self.test_img_name)
self.assertTrue(
len(test_image_list) > 0,
'Test image={} does not exist!'.format(self.test_img_name))
# 1) Remove test image
da.remove_image(test_image_list[0], force=True)
test_image_list = da.images(name=self.test_img_name)
self.assertTrue(
len(test_image_list) == 0,
'Test image={} should be removed!'.format(self.test_img_name))