moto/tests/test_autoscaling/utils.py

import boto
import boto3
from boto import vpc as boto_vpc
from moto import mock_ec2, mock_ec2_deprecated


@mock_ec2
def setup_networking():
    ec2 = boto3.resource('ec2', region_name='us-east-1')
    vpc = ec2.create_vpc(CidrBlock='10.11.0.0/16')
    subnet1 = ec2.create_subnet(
        VpcId=vpc.id,
        CidrBlock='10.11.1.0/24',
        AvailabilityZone='us-east-1a')
    subnet2 = ec2.create_subnet(
        VpcId=vpc.id,
        CidrBlock='10.11.2.0/24',
        AvailabilityZone='us-east-1b')
    return {'vpc': vpc.id, 'subnet1': subnet1.id, 'subnet2': subnet2.id}

@mock_ec2_deprecated
def setup_networking_deprecated():
    conn = boto_vpc.connect_to_region('us-east-1')
    vpc = conn.create_vpc("10.11.0.0/16")
    subnet1 = conn.create_subnet(
        vpc.id,
        "10.11.1.0/24",
        availability_zone='us-east-1a')
    subnet2 = conn.create_subnet(
        vpc.id,
        "10.11.2.0/24",
        availability_zone='us-east-1b')
    return {'vpc': vpc.id, 'subnet1': subnet1.id, 'subnet2': subnet2.id}


@mock_ec2
def setup_instance_with_networking(image_id, instance_type):
    mock_data = setup_networking()
    ec2 = boto3.resource('ec2', region_name='us-east-1')
    instances = ec2.create_instances(
        ImageId=image_id,
        InstanceType=instance_type,
        MaxCount=1,
        MinCount=1,
        SubnetId=mock_data['subnet1']
    )
    mock_data['instance'] = instances[0].id
    return mock_data