Пример #1
0
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
Пример #2
0
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