9a26c92e72
This is a bug fix to my initial work when implementing batch_delete_image. Deleting the last tag for a given image should delete the image from the backend. I also cleaned up the tests previously-added in the initial implementation.
1082 lines
37 KiB
Python
1082 lines
37 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import hashlib
|
|
import json
|
|
from datetime import datetime
|
|
from freezegun import freeze_time
|
|
import os
|
|
from random import random
|
|
|
|
import re
|
|
import sure # noqa
|
|
|
|
import boto3
|
|
from botocore.exceptions import ClientError, ParamValidationError
|
|
from dateutil.tz import tzlocal
|
|
|
|
from moto import mock_ecr
|
|
from nose import SkipTest
|
|
|
|
|
|
def _create_image_digest(contents=None):
|
|
if not contents:
|
|
contents = 'docker_image{0}'.format(int(random() * 10 ** 6))
|
|
return "sha256:%s" % hashlib.sha256(contents.encode('utf-8')).hexdigest()
|
|
|
|
|
|
def _create_image_manifest():
|
|
return {
|
|
"schemaVersion": 2,
|
|
"mediaType": "application/vnd.docker.distribution.manifest.v2+json",
|
|
"config":
|
|
{
|
|
"mediaType": "application/vnd.docker.container.image.v1+json",
|
|
"size": 7023,
|
|
"digest": _create_image_digest("config")
|
|
},
|
|
"layers": [
|
|
{
|
|
"mediaType": "application/vnd.docker.image.rootfs.diff.tar.gzip",
|
|
"size": 32654,
|
|
"digest": _create_image_digest("layer1")
|
|
},
|
|
{
|
|
"mediaType": "application/vnd.docker.image.rootfs.diff.tar.gzip",
|
|
"size": 16724,
|
|
"digest": _create_image_digest("layer2")
|
|
},
|
|
{
|
|
"mediaType": "application/vnd.docker.image.rootfs.diff.tar.gzip",
|
|
"size": 73109,
|
|
# randomize image digest
|
|
"digest": _create_image_digest()
|
|
}
|
|
]
|
|
}
|
|
|
|
|
|
@mock_ecr
|
|
def test_create_repository():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
response = client.create_repository(
|
|
repositoryName='test_ecr_repository'
|
|
)
|
|
response['repository']['repositoryName'].should.equal('test_ecr_repository')
|
|
response['repository']['repositoryArn'].should.equal(
|
|
'arn:aws:ecr:us-east-1:012345678910:repository/test_ecr_repository')
|
|
response['repository']['registryId'].should.equal('012345678910')
|
|
response['repository']['repositoryUri'].should.equal(
|
|
'012345678910.dkr.ecr.us-east-1.amazonaws.com/test_ecr_repository')
|
|
# response['repository']['createdAt'].should.equal(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_repositories():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository1'
|
|
)
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository0'
|
|
)
|
|
response = client.describe_repositories()
|
|
len(response['repositories']).should.equal(2)
|
|
|
|
respository_arns = ['arn:aws:ecr:us-east-1:012345678910:repository/test_repository1',
|
|
'arn:aws:ecr:us-east-1:012345678910:repository/test_repository0']
|
|
set([response['repositories'][0]['repositoryArn'],
|
|
response['repositories'][1]['repositoryArn']]).should.equal(set(respository_arns))
|
|
|
|
respository_uris = ['012345678910.dkr.ecr.us-east-1.amazonaws.com/test_repository1',
|
|
'012345678910.dkr.ecr.us-east-1.amazonaws.com/test_repository0']
|
|
set([response['repositories'][0]['repositoryUri'],
|
|
response['repositories'][1]['repositoryUri']]).should.equal(set(respository_uris))
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_repositories_1():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository1'
|
|
)
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository0'
|
|
)
|
|
response = client.describe_repositories(registryId='012345678910')
|
|
len(response['repositories']).should.equal(2)
|
|
|
|
respository_arns = ['arn:aws:ecr:us-east-1:012345678910:repository/test_repository1',
|
|
'arn:aws:ecr:us-east-1:012345678910:repository/test_repository0']
|
|
set([response['repositories'][0]['repositoryArn'],
|
|
response['repositories'][1]['repositoryArn']]).should.equal(set(respository_arns))
|
|
|
|
respository_uris = ['012345678910.dkr.ecr.us-east-1.amazonaws.com/test_repository1',
|
|
'012345678910.dkr.ecr.us-east-1.amazonaws.com/test_repository0']
|
|
set([response['repositories'][0]['repositoryUri'],
|
|
response['repositories'][1]['repositoryUri']]).should.equal(set(respository_uris))
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_repositories_2():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository1'
|
|
)
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository0'
|
|
)
|
|
response = client.describe_repositories(registryId='109876543210')
|
|
len(response['repositories']).should.equal(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_repositories_3():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository1'
|
|
)
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository0'
|
|
)
|
|
response = client.describe_repositories(repositoryNames=['test_repository1'])
|
|
len(response['repositories']).should.equal(1)
|
|
respository_arn = 'arn:aws:ecr:us-east-1:012345678910:repository/test_repository1'
|
|
response['repositories'][0]['repositoryArn'].should.equal(respository_arn)
|
|
|
|
respository_uri = '012345678910.dkr.ecr.us-east-1.amazonaws.com/test_repository1'
|
|
response['repositories'][0]['repositoryUri'].should.equal(respository_uri)
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_repositories_with_image():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
response = client.describe_repositories(repositoryNames=['test_repository'])
|
|
len(response['repositories']).should.equal(1)
|
|
|
|
|
|
@mock_ecr
|
|
def test_delete_repository():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
response = client.delete_repository(repositoryName='test_repository')
|
|
response['repository']['repositoryName'].should.equal('test_repository')
|
|
response['repository']['repositoryArn'].should.equal(
|
|
'arn:aws:ecr:us-east-1:012345678910:repository/test_repository')
|
|
response['repository']['registryId'].should.equal('012345678910')
|
|
response['repository']['repositoryUri'].should.equal(
|
|
'012345678910.dkr.ecr.us-east-1.amazonaws.com/test_repository')
|
|
# response['repository']['createdAt'].should.equal(0)
|
|
|
|
response = client.describe_repositories()
|
|
len(response['repositories']).should.equal(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_put_image():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
response = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
response['image']['imageId']['imageTag'].should.equal('latest')
|
|
response['image']['imageId']['imageDigest'].should.contain("sha")
|
|
response['image']['repositoryName'].should.equal('test_repository')
|
|
response['image']['registryId'].should.equal('012345678910')
|
|
|
|
|
|
@mock_ecr
|
|
def test_put_image_with_push_date():
|
|
if os.environ.get('TEST_SERVER_MODE', 'false').lower() == 'true':
|
|
raise SkipTest('Cant manipulate time in server mode')
|
|
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
with freeze_time('2018-08-28 00:00:00'):
|
|
image1_date = datetime.now()
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
with freeze_time('2019-05-31 00:00:00'):
|
|
image2_date = datetime.now()
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
describe_response = client.describe_images(repositoryName='test_repository')
|
|
|
|
type(describe_response['imageDetails']).should.be(list)
|
|
len(describe_response['imageDetails']).should.be(2)
|
|
|
|
set([describe_response['imageDetails'][0]['imagePushedAt'],
|
|
describe_response['imageDetails'][1]['imagePushedAt']]).should.equal(set([image1_date, image2_date]))
|
|
|
|
|
|
@mock_ecr
|
|
def test_put_image_with_multiple_tags():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
manifest = _create_image_manifest()
|
|
response = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag='v1'
|
|
)
|
|
|
|
response['image']['imageId']['imageTag'].should.equal('v1')
|
|
response['image']['imageId']['imageDigest'].should.contain("sha")
|
|
response['image']['repositoryName'].should.equal('test_repository')
|
|
response['image']['registryId'].should.equal('012345678910')
|
|
|
|
response1 = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag='latest'
|
|
)
|
|
|
|
response1['image']['imageId']['imageTag'].should.equal('latest')
|
|
response1['image']['imageId']['imageDigest'].should.contain("sha")
|
|
response1['image']['repositoryName'].should.equal('test_repository')
|
|
response1['image']['registryId'].should.equal('012345678910')
|
|
|
|
response2 = client.describe_images(repositoryName='test_repository')
|
|
type(response2['imageDetails']).should.be(list)
|
|
len(response2['imageDetails']).should.be(1)
|
|
|
|
response2['imageDetails'][0]['imageDigest'].should.contain("sha")
|
|
|
|
response2['imageDetails'][0]['registryId'].should.equal("012345678910")
|
|
|
|
response2['imageDetails'][0]['repositoryName'].should.equal("test_repository")
|
|
|
|
len(response2['imageDetails'][0]['imageTags']).should.be(2)
|
|
response2['imageDetails'][0]['imageTags'].should.be.equal(['v1', 'latest'])
|
|
|
|
|
|
@mock_ecr
|
|
def test_list_images():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository_1'
|
|
)
|
|
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository_2'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository_1',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository_1',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v1'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository_1',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v2'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository_2',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='oldest'
|
|
)
|
|
|
|
response = client.list_images(repositoryName='test_repository_1')
|
|
type(response['imageIds']).should.be(list)
|
|
len(response['imageIds']).should.be(3)
|
|
|
|
image_tags = ['latest', 'v1', 'v2']
|
|
set([response['imageIds'][0]['imageTag'],
|
|
response['imageIds'][1]['imageTag'],
|
|
response['imageIds'][2]['imageTag']]).should.equal(set(image_tags))
|
|
|
|
response = client.list_images(repositoryName='test_repository_2')
|
|
type(response['imageIds']).should.be(list)
|
|
len(response['imageIds']).should.be(1)
|
|
response['imageIds'][0]['imageTag'].should.equal('oldest')
|
|
|
|
|
|
@mock_ecr
|
|
def test_list_images_from_repository_that_doesnt_exist():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository_1'
|
|
)
|
|
|
|
# non existing repo
|
|
error_msg = re.compile(
|
|
r".*The repository with name 'repo-that-doesnt-exist' does not exist in the registry with id '123'.*",
|
|
re.MULTILINE)
|
|
client.list_images.when.called_with(
|
|
repositoryName='repo-that-doesnt-exist',
|
|
registryId='123',
|
|
).should.throw(Exception, error_msg)
|
|
|
|
# repo does not exist in specified registry
|
|
error_msg = re.compile(
|
|
r".*The repository with name 'test_repository_1' does not exist in the registry with id '222'.*",
|
|
re.MULTILINE)
|
|
client.list_images.when.called_with(
|
|
repositoryName='test_repository_1',
|
|
registryId='222',
|
|
).should.throw(Exception, error_msg)
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_images():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest())
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v1'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v2'
|
|
)
|
|
|
|
response = client.describe_images(repositoryName='test_repository')
|
|
type(response['imageDetails']).should.be(list)
|
|
len(response['imageDetails']).should.be(4)
|
|
|
|
response['imageDetails'][0]['imageDigest'].should.contain("sha")
|
|
response['imageDetails'][1]['imageDigest'].should.contain("sha")
|
|
response['imageDetails'][2]['imageDigest'].should.contain("sha")
|
|
response['imageDetails'][3]['imageDigest'].should.contain("sha")
|
|
|
|
response['imageDetails'][0]['registryId'].should.equal("012345678910")
|
|
response['imageDetails'][1]['registryId'].should.equal("012345678910")
|
|
response['imageDetails'][2]['registryId'].should.equal("012345678910")
|
|
response['imageDetails'][3]['registryId'].should.equal("012345678910")
|
|
|
|
response['imageDetails'][0]['repositoryName'].should.equal("test_repository")
|
|
response['imageDetails'][1]['repositoryName'].should.equal("test_repository")
|
|
response['imageDetails'][2]['repositoryName'].should.equal("test_repository")
|
|
response['imageDetails'][3]['repositoryName'].should.equal("test_repository")
|
|
|
|
response['imageDetails'][0].should_not.have.key('imageTags')
|
|
len(response['imageDetails'][1]['imageTags']).should.be(1)
|
|
len(response['imageDetails'][2]['imageTags']).should.be(1)
|
|
len(response['imageDetails'][3]['imageTags']).should.be(1)
|
|
|
|
image_tags = ['latest', 'v1', 'v2']
|
|
set([response['imageDetails'][1]['imageTags'][0],
|
|
response['imageDetails'][2]['imageTags'][0],
|
|
response['imageDetails'][3]['imageTags'][0]]).should.equal(set(image_tags))
|
|
|
|
response['imageDetails'][0]['imageSizeInBytes'].should.equal(52428800)
|
|
response['imageDetails'][1]['imageSizeInBytes'].should.equal(52428800)
|
|
response['imageDetails'][2]['imageSizeInBytes'].should.equal(52428800)
|
|
response['imageDetails'][3]['imageSizeInBytes'].should.equal(52428800)
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_images_by_tag():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
tag_map = {}
|
|
for tag in ['latest', 'v1', 'v2']:
|
|
put_response = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag=tag
|
|
)
|
|
tag_map[tag] = put_response['image']
|
|
|
|
for tag, put_response in tag_map.items():
|
|
response = client.describe_images(repositoryName='test_repository', imageIds=[{'imageTag': tag}])
|
|
len(response['imageDetails']).should.be(1)
|
|
image_detail = response['imageDetails'][0]
|
|
image_detail['registryId'].should.equal("012345678910")
|
|
image_detail['repositoryName'].should.equal("test_repository")
|
|
image_detail['imageTags'].should.equal([put_response['imageId']['imageTag']])
|
|
image_detail['imageDigest'].should.equal(put_response['imageId']['imageDigest'])
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_images_tags_should_not_contain_empty_tag1():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest)
|
|
)
|
|
|
|
tags = ['v1', 'v2', 'latest']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag
|
|
)
|
|
|
|
response = client.describe_images(repositoryName='test_repository', imageIds=[{'imageTag': tag}])
|
|
len(response['imageDetails']).should.be(1)
|
|
image_detail = response['imageDetails'][0]
|
|
len(image_detail['imageTags']).should.equal(3)
|
|
image_detail['imageTags'].should.be.equal(tags)
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_images_tags_should_not_contain_empty_tag2():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
tags = ['v1', 'v2']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag
|
|
)
|
|
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest)
|
|
)
|
|
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag='latest'
|
|
)
|
|
|
|
response = client.describe_images(repositoryName='test_repository', imageIds=[{'imageTag': tag}])
|
|
len(response['imageDetails']).should.be(1)
|
|
image_detail = response['imageDetails'][0]
|
|
len(image_detail['imageTags']).should.equal(3)
|
|
image_detail['imageTags'].should.be.equal(['v1', 'v2', 'latest'])
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_repository_that_doesnt_exist():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
|
|
error_msg = re.compile(
|
|
r".*The repository with name 'repo-that-doesnt-exist' does not exist in the registry with id '123'.*",
|
|
re.MULTILINE)
|
|
client.describe_repositories.when.called_with(
|
|
repositoryNames=['repo-that-doesnt-exist'],
|
|
registryId='123',
|
|
).should.throw(ClientError, error_msg)
|
|
|
|
@mock_ecr
|
|
def test_describe_image_that_doesnt_exist():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(repositoryName='test_repository')
|
|
|
|
error_msg1 = re.compile(
|
|
r".*The image with imageId {imageDigest:'null', imageTag:'testtag'} does not exist within "
|
|
r"the repository with name 'test_repository' in the registry with id '123'.*",
|
|
re.MULTILINE)
|
|
|
|
client.describe_images.when.called_with(
|
|
repositoryName='test_repository', imageIds=[{'imageTag': 'testtag'}], registryId='123',
|
|
).should.throw(ClientError, error_msg1)
|
|
|
|
error_msg2 = re.compile(
|
|
r".*The repository with name 'repo-that-doesnt-exist' does not exist in the registry with id '123'.*",
|
|
re.MULTILINE)
|
|
client.describe_images.when.called_with(
|
|
repositoryName='repo-that-doesnt-exist', imageIds=[{'imageTag': 'testtag'}], registryId='123',
|
|
).should.throw(ClientError, error_msg2)
|
|
|
|
|
|
@mock_ecr
|
|
def test_delete_repository_that_doesnt_exist():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
|
|
error_msg = re.compile(
|
|
r".*The repository with name 'repo-that-doesnt-exist' does not exist in the registry with id '123'.*",
|
|
re.MULTILINE)
|
|
|
|
client.delete_repository.when.called_with(
|
|
repositoryName='repo-that-doesnt-exist',
|
|
registryId='123').should.throw(
|
|
ClientError, error_msg)
|
|
|
|
|
|
@mock_ecr
|
|
def test_describe_images_by_digest():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
tags = ['latest', 'v1', 'v2']
|
|
digest_map = {}
|
|
for tag in tags:
|
|
put_response = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag=tag
|
|
)
|
|
digest_map[put_response['image']['imageId']['imageDigest']] = put_response['image']
|
|
|
|
for digest, put_response in digest_map.items():
|
|
response = client.describe_images(repositoryName='test_repository',
|
|
imageIds=[{'imageDigest': digest}])
|
|
len(response['imageDetails']).should.be(1)
|
|
image_detail = response['imageDetails'][0]
|
|
image_detail['registryId'].should.equal("012345678910")
|
|
image_detail['repositoryName'].should.equal("test_repository")
|
|
image_detail['imageTags'].should.equal([put_response['imageId']['imageTag']])
|
|
image_detail['imageDigest'].should.equal(digest)
|
|
|
|
|
|
@mock_ecr
|
|
def test_get_authorization_token_assume_region():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
auth_token_response = client.get_authorization_token()
|
|
|
|
auth_token_response.should.contain('authorizationData')
|
|
auth_token_response.should.contain('ResponseMetadata')
|
|
auth_token_response['authorizationData'].should.equal([
|
|
{
|
|
'authorizationToken': 'QVdTOjAxMjM0NTY3ODkxMC1hdXRoLXRva2Vu',
|
|
'proxyEndpoint': 'https://012345678910.dkr.ecr.us-east-1.amazonaws.com',
|
|
'expiresAt': datetime(2015, 1, 1, tzinfo=tzlocal())
|
|
},
|
|
])
|
|
|
|
|
|
@mock_ecr
|
|
def test_get_authorization_token_explicit_regions():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
auth_token_response = client.get_authorization_token(registryIds=['10987654321', '878787878787'])
|
|
|
|
auth_token_response.should.contain('authorizationData')
|
|
auth_token_response.should.contain('ResponseMetadata')
|
|
auth_token_response['authorizationData'].should.equal([
|
|
{
|
|
'authorizationToken': 'QVdTOjEwOTg3NjU0MzIxLWF1dGgtdG9rZW4=',
|
|
'proxyEndpoint': 'https://10987654321.dkr.ecr.us-east-1.amazonaws.com',
|
|
'expiresAt': datetime(2015, 1, 1, tzinfo=tzlocal()),
|
|
},
|
|
{
|
|
'authorizationToken': 'QVdTOjg3ODc4Nzg3ODc4Ny1hdXRoLXRva2Vu',
|
|
'proxyEndpoint': 'https://878787878787.dkr.ecr.us-east-1.amazonaws.com',
|
|
'expiresAt': datetime(2015, 1, 1, tzinfo=tzlocal())
|
|
|
|
}
|
|
])
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_get_image():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v1'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v2'
|
|
)
|
|
|
|
response = client.batch_get_image(
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageTag': 'v2'
|
|
},
|
|
],
|
|
)
|
|
|
|
type(response['images']).should.be(list)
|
|
len(response['images']).should.be(1)
|
|
|
|
response['images'][0]['imageManifest'].should.contain("vnd.docker.distribution.manifest.v2+json")
|
|
response['images'][0]['registryId'].should.equal("012345678910")
|
|
response['images'][0]['repositoryName'].should.equal("test_repository")
|
|
|
|
response['images'][0]['imageId']['imageTag'].should.equal("v2")
|
|
response['images'][0]['imageId']['imageDigest'].should.contain("sha")
|
|
|
|
type(response['failures']).should.be(list)
|
|
len(response['failures']).should.be(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_get_image_that_doesnt_exist():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v1'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v2'
|
|
)
|
|
|
|
response = client.batch_get_image(
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageTag': 'v5'
|
|
},
|
|
],
|
|
)
|
|
|
|
type(response['images']).should.be(list)
|
|
len(response['images']).should.be(0)
|
|
|
|
type(response['failures']).should.be(list)
|
|
len(response['failures']).should.be(1)
|
|
response['failures'][0]['failureReason'].should.equal("Requested image not found")
|
|
response['failures'][0]['failureCode'].should.equal("ImageNotFound")
|
|
response['failures'][0]['imageId']['imageTag'].should.equal("v5")
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_get_image_no_tags():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
_ = client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
_ = client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='latest'
|
|
)
|
|
|
|
error_msg = re.compile(
|
|
r".*Missing required parameter in input: \"imageIds\".*",
|
|
re.MULTILINE)
|
|
|
|
client.batch_get_image.when.called_with(
|
|
repositoryName='test_repository').should.throw(
|
|
ParamValidationError, error_msg)
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_by_tag():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
|
|
tags = ['v1', 'v1.0', 'latest']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag,
|
|
)
|
|
|
|
describe_response1 = client.describe_images(repositoryName='test_repository')
|
|
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageTag': 'latest'
|
|
},
|
|
],
|
|
)
|
|
|
|
describe_response2 = client.describe_images(repositoryName='test_repository')
|
|
|
|
type(describe_response1['imageDetails'][0]['imageTags']).should.be(list)
|
|
len(describe_response1['imageDetails'][0]['imageTags']).should.be(3)
|
|
|
|
type(describe_response2['imageDetails'][0]['imageTags']).should.be(list)
|
|
len(describe_response2['imageDetails'][0]['imageTags']).should.be(2)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(1)
|
|
|
|
batch_delete_response['imageIds'][0]['imageTag'].should.equal("latest")
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_delete_last_tag():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(_create_image_manifest()),
|
|
imageTag='v1',
|
|
)
|
|
|
|
describe_response1 = client.describe_images(repositoryName='test_repository')
|
|
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageTag': 'v1'
|
|
},
|
|
],
|
|
)
|
|
|
|
describe_response2 = client.describe_images(repositoryName='test_repository')
|
|
|
|
type(describe_response1['imageDetails'][0]['imageTags']).should.be(list)
|
|
len(describe_response1['imageDetails'][0]['imageTags']).should.be(1)
|
|
|
|
type(describe_response2['imageDetails']).should.be(list)
|
|
len(describe_response2['imageDetails']).should.be(0)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(1)
|
|
|
|
batch_delete_response['imageIds'][0]['imageTag'].should.equal("v1")
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_with_nonexistent_tag():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
|
|
tags = ['v1', 'v1.0', 'latest']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag,
|
|
)
|
|
|
|
describe_response = client.describe_images(repositoryName='test_repository')
|
|
|
|
missing_tag = "missing-tag"
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageTag': missing_tag
|
|
},
|
|
],
|
|
)
|
|
|
|
type(describe_response['imageDetails'][0]['imageTags']).should.be(list)
|
|
len(describe_response['imageDetails'][0]['imageTags']).should.be(3)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(0)
|
|
|
|
batch_delete_response['failures'][0]['imageId']['imageTag'].should.equal(missing_tag)
|
|
batch_delete_response['failures'][0]['failureCode'].should.equal("ImageNotFound")
|
|
batch_delete_response['failures'][0]['failureReason'].should.equal("Requested image not found")
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(1)
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_by_digest():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
|
|
tags = ['v1', 'v2', 'latest']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag
|
|
)
|
|
|
|
describe_response = client.describe_images(repositoryName='test_repository')
|
|
image_digest = describe_response['imageDetails'][0]['imageDigest']
|
|
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageDigest': image_digest
|
|
},
|
|
],
|
|
)
|
|
|
|
describe_response = client.describe_images(repositoryName='test_repository')
|
|
|
|
type(describe_response['imageDetails']).should.be(list)
|
|
len(describe_response['imageDetails']).should.be(0)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(3)
|
|
|
|
batch_delete_response['imageIds'][0]['imageDigest'].should.equal(image_digest)
|
|
batch_delete_response['imageIds'][1]['imageDigest'].should.equal(image_digest)
|
|
batch_delete_response['imageIds'][2]['imageDigest'].should.equal(image_digest)
|
|
|
|
set([
|
|
batch_delete_response['imageIds'][0]['imageTag'],
|
|
batch_delete_response['imageIds'][1]['imageTag'],
|
|
batch_delete_response['imageIds'][2]['imageTag']]).should.equal(set(tags))
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_with_invalid_digest():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
|
|
tags = ['v1', 'v2', 'latest']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag
|
|
)
|
|
|
|
invalid_image_digest = 'sha256:invalid-digest'
|
|
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageDigest': invalid_image_digest
|
|
},
|
|
],
|
|
)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(0)
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(1)
|
|
|
|
batch_delete_response['failures'][0]['imageId']['imageDigest'].should.equal(invalid_image_digest)
|
|
batch_delete_response['failures'][0]['failureCode'].should.equal("InvalidImageDigest")
|
|
batch_delete_response['failures'][0]['failureReason'].should.equal("Invalid request parameters: image digest should satisfy the regex '[a-zA-Z0-9-_+.]+:[a-fA-F0-9]+'")
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_with_missing_parameters():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
},
|
|
],
|
|
)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(0)
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(1)
|
|
|
|
batch_delete_response['failures'][0]['failureCode'].should.equal("MissingDigestAndTag")
|
|
batch_delete_response['failures'][0]['failureReason'].should.equal("Invalid request parameters: both tag and digest cannot be null")
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_with_matching_digest_and_tag():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
|
|
tags = ['v1', 'v1.0', 'latest']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag
|
|
)
|
|
|
|
describe_response = client.describe_images(repositoryName='test_repository')
|
|
image_digest = describe_response['imageDetails'][0]['imageDigest']
|
|
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageDigest': image_digest,
|
|
'imageTag': 'v1'
|
|
},
|
|
],
|
|
)
|
|
|
|
describe_response = client.describe_images(repositoryName='test_repository')
|
|
|
|
type(describe_response['imageDetails']).should.be(list)
|
|
len(describe_response['imageDetails']).should.be(0)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(3)
|
|
|
|
batch_delete_response['imageIds'][0]['imageDigest'].should.equal(image_digest)
|
|
batch_delete_response['imageIds'][1]['imageDigest'].should.equal(image_digest)
|
|
batch_delete_response['imageIds'][2]['imageDigest'].should.equal(image_digest)
|
|
|
|
set([
|
|
batch_delete_response['imageIds'][0]['imageTag'],
|
|
batch_delete_response['imageIds'][1]['imageTag'],
|
|
batch_delete_response['imageIds'][2]['imageTag']]).should.equal(set(tags))
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(0)
|
|
|
|
|
|
@mock_ecr
|
|
def test_batch_delete_image_with_mismatched_digest_and_tag():
|
|
client = boto3.client('ecr', region_name='us-east-1')
|
|
client.create_repository(
|
|
repositoryName='test_repository'
|
|
)
|
|
|
|
manifest = _create_image_manifest()
|
|
|
|
tags = ['v1', 'latest']
|
|
for tag in tags:
|
|
client.put_image(
|
|
repositoryName='test_repository',
|
|
imageManifest=json.dumps(manifest),
|
|
imageTag=tag
|
|
)
|
|
|
|
describe_response = client.describe_images(repositoryName='test_repository')
|
|
image_digest = describe_response['imageDetails'][0]['imageDigest']
|
|
|
|
batch_delete_response = client.batch_delete_image(
|
|
registryId='012345678910',
|
|
repositoryName='test_repository',
|
|
imageIds=[
|
|
{
|
|
'imageDigest': image_digest,
|
|
'imageTag': 'v2'
|
|
},
|
|
],
|
|
)
|
|
|
|
type(batch_delete_response['imageIds']).should.be(list)
|
|
len(batch_delete_response['imageIds']).should.be(0)
|
|
|
|
type(batch_delete_response['failures']).should.be(list)
|
|
len(batch_delete_response['failures']).should.be(1)
|
|
|
|
batch_delete_response['failures'][0]['imageId']['imageDigest'].should.equal(image_digest)
|
|
batch_delete_response['failures'][0]['imageId']['imageTag'].should.equal("v2")
|
|
batch_delete_response['failures'][0]['failureCode'].should.equal("ImageNotFound")
|
|
batch_delete_response['failures'][0]['failureReason'].should.equal("Requested image not found")
|