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

allow get_filter_value to return state of ami and test

Browse Source 
* updated Ami.get_filter_value to return the state value of the AMI
  * updated test_ec2.test_amis.test_ami_filters to test for AMIs in the state of 'available'
This commit is contained in:
Joseph Lawson 2014-09-30 16:12:23 -04:00
parent 2650d9981f
commit 866d8b4ea8
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
moto/ec2/models.py
View File

@ -604,6 +604,8 @@ class Ami(TaggedEC2Instance):
return getattr(self,filter_name) return getattr(self,filter_name)
elif filter_name == 'image-id': elif filter_name == 'image-id':
return self.id return self.id
elif filter_name == 'state':
return self.state
else: else:
ec2_backend.raise_not_implemented_error("The filter '{0}' for DescribeImages".format(filter_name)) ec2_backend.raise_not_implemented_error("The filter '{0}' for DescribeImages".format(filter_name))

3
tests/test_ec2/test_amis.py
View File

@ -132,6 +132,9 @@ 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'})
set([ami.id for ami in amis_by_id]).should.equal(set([imageA.id]))
@mock_ec2 @mock_ec2
def test_getting_missing_ami(): def test_getting_missing_ami():