Amazing-Python-Scripts

1
import boto3
2
import boto.sqs
3
from boto.s3.key import Key
4
from moto import mock_ec2, mock_s3
5
import awstesting.awsHandler as aws
6

7

8
def add_service(service_name, region):
9
    aws_client = boto3.client(service_name, region_name=region)
10
    return aws_client
11

12

13
@mock_ec2
14
def test_create_ec2(aws_client):
15
    # demo ami
16
    ami_id = "ami-123"
17
    count = 100
18
    aws.create_ec2(aws_client, ami_id, count)
19
    instances = aws_client.describe_instances()['Reservations'][0]['Instances']
20
    for i in instances:
21
        print(i)
22
    if len(instances) == count:
23
        return "ec2 created successfully Insatnce ID = " + instances[0]['InstanceId'] + ""
24
    else:
25
        return "ec2 not created "
26

27

28
@mock_s3
29
def test_s3():
30
    print('Testing moto S3')
31

32
    # create bucket
33
    bucket_name = 'bucket1'
34
    conn = boto.connect_s3()
35
    print('Creating bucket: {}'.format(bucket_name))
36
    bucket = conn.create_bucket(bucket_name)
37

38
    # add object
39
    k = Key(bucket)
40
    key_name = 'file1'
41
    k.key = key_name
42
    k.set_contents_from_string('hello world')
43

44
    # list objects
45
    print('List of files:')
46
    for key in bucket.list():
47
        print('    {}/{}'.format(key.bucket.name, key.name))
48

49
    # get object
50
    k2 = Key(bucket)
51
    k2.key = key_name
52
    data = k2.get_contents_as_string()
53
    print('Fetched object {}/{} with content: {}'.format(bucket.name,
54
                                                         key.name, data))
55

56

57
if __name__ == "__main__":
58
    client = add_service("ec2", "us-east-1")
59
    test_create_ec2(client)
60