def test_ls_ami(): """List AMIS from all regions.""" aws_env = AWSEnv(regions=["us-east-1", "eu-west-1"], stub=True) stub_us = aws_env.stub("ec2", region="us-east-1") stub_eu = aws_env.stub("ec2", region="eu-west-1") stub_eu.add_response( "describe_images", { "Images": [ {"ImageId": "ami-1234", "RootDeviceName": "/dev/sda1", "Tags": []} ] }, {"Filters": [], "Owners": ["self"]}, ) stub_us.add_response( "describe_images", { "Images": [ {"ImageId": "ami-5678", "RootDeviceName": "/dev/sda1", "Tags": []} ] }, {"Filters": [], "Owners": ["self"]}, ) assert len(AMI.ls()) == 2
def test_ls_ami(): """List AMIS from all regions.""" aws_env = AWSEnv(regions=['us-east-1', 'eu-west-1'], stub=True) stub_us = aws_env.stub('ec2', region='us-east-1') stub_eu = aws_env.stub('ec2', region='eu-west-1') stub_eu.add_response( 'describe_images', { 'Images': [{ 'ImageId': 'ami-1234', 'RootDeviceName': '/dev/sda1', 'Tags': [] }] }, { 'Filters': [], 'Owners': ['self'] }) stub_us.add_response( 'describe_images', { 'Images': [{ 'ImageId': 'ami-5678', 'RootDeviceName': '/dev/sda1', 'Tags': [] }] }, { 'Filters': [], 'Owners': ['self'] }) assert len(AMI.ls()) == 2