Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement "name" filter to ec2.models.AMI

Browse Source
This commit is contained in:
Hugo Lopes Tavares 2014-10-10 15:14:46 -04:00
parent 9321a7bfe1
commit aee47395f7
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/ec2/models.py
View File

@ -748,6 +748,8 @@ class Ami(TaggedEC2Resource):
return self.id return self.id
elif filter_name == 'state': elif filter_name == 'state':
return self.state return self.state
elif filter_name == 'name':
return self.name
filter_value = super(Ami, self).get_filter_value(filter_name) filter_value = super(Ami, self).get_filter_value(filter_name)

6
tests/test_ec2/test_amis.py
View File

@ -177,9 +177,11 @@ def test_ami_filters():
amis_by_id = conn.get_all_images(filters={'image-id': imageA.id}) amis_by_id = conn.get_all_images(filters={'image-id': imageA.id})
set([ami.id for ami in amis_by_id]).should.equal(set([imageA.id])) set([ami.id for ami in amis_by_id]).should.equal(set([imageA.id]))
amis_by_id = conn.get_all_images(filters={'state': 'available'}) amis_by_state = conn.get_all_images(filters={'state': 'available'})
set([ami.id for ami in amis_by_id]).should.equal(set([imageA.id, imageB.id])) set([ami.id for ami in amis_by_state]).should.equal(set([imageA.id, imageB.id]))
amis_by_name = conn.get_all_images(filters={'name': imageA.name})
set([ami.id for ami in amis_by_name]).should.equal(set([imageA.id]))
@mock_ec2 @mock_ec2
def test_ami_filtering_via_tag(): def test_ami_filtering_via_tag():