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

additional filters to get_all_vpcs

Browse Source
This commit is contained in:
rocky4570fft 2016-08-14 16:58:35 +10:00
parent ae938223d4
commit 0161422d21
2 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
moto/ec2/models.py
View File

@ -1791,6 +1791,10 @@ class VPC(TaggedEC2Resource):
return self.id
elif filter_name == 'cidr':
return self.cidr_block
elif filter_name == 'isDefault':
return self.is_default
elif filter_name == 'state':
return self.state
elif filter_name == 'dhcp-options-id':
if not self.dhcp_options:
return None

29
tests/test_ec2/test_vpcs.py
View File

@ -50,6 +50,35 @@ def test_vpc_defaults():
conn.get_all_route_tables().should.have.length_of(0)
conn.get_all_security_groups(filters={'vpc-id': [vpc.id]}).should.have.length_of(0)
@mock_ec2
def test_vpc_isdefault_filter():
conn = boto.connect_vpc('the_key', 'the_secret')
vpc = conn.create_vpc("10.0.0.0/16")
conn.get_all_vpcs(filters={'isDefault': 'true'}).should.have.length_of(1)
vpc.delete()
conn.get_all_vpcs(filters={'isDefault': 'true'}).should.have.length_of(0)
@mock_ec2
def test_multiple_vpcs_default_filter():
conn = boto.connect_vpc('the_key', 'the_secret')
conn.create_vpc("10.8.0.0/16")
conn.create_vpc("10.0.0.0/16")
conn.create_vpc("192.168.0.0/16")
conn.get_all_vpcs().should.have.length_of(3)
vpc = conn.get_all_vpcs(filters={'isDefault': 'true'})
vpc.should.have.length_of(1)
vpc[0].cidr_block.should.equal('10.8.0.0/16')
@mock_ec2
def test_vpc_state_available_filter():
conn = boto.connect_vpc('the_key', 'the_secret')
vpc = conn.create_vpc("10.0.0.0/16")
conn.create_vpc("10.1.0.0/16")
conn.get_all_vpcs(filters={'state': 'available'}).should.have.length_of(2)
vpc.delete()
conn.get_all_vpcs(filters={'state': 'available'}).should.have.length_of(1)
@mock_ec2
def test_vpc_tagging():