catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Finialised create compute environment + describe environments

Browse source
This commit is contained in:
Terry Cain 2017-09-29 23:29:36 +01:00
commit f95d72c37c
No known key found for this signature in database
GPG key ID: 14D90844E4E9B9F3
4 changed files with 237 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

84
tests/test_batch/test_batch.py
View file

@ -2,7 +2,7 @@ from __future__ import unicode_literals
import boto3
import sure # noqa
from moto import mock_batch, mock_iam, mock_ec2
from moto import mock_batch, mock_iam, mock_ec2, mock_ecs
DEFAULT_REGION = 'eu-central-1'
@ -11,6 +11,7 @@ DEFAULT_REGION = 'eu-central-1'
def _get_clients():
return boto3.client('ec2', region_name=DEFAULT_REGION), \
boto3.client('iam', region_name=DEFAULT_REGION), \
boto3.client('ecs', region_name=DEFAULT_REGION), \
boto3.client('batch', region_name=DEFAULT_REGION)
@ -46,10 +47,11 @@ def _setup(ec2_client, iam_client):
# Yes, yes it talks to all the things
@mock_ec2
@mock_ecs
@mock_iam
@mock_batch
def test_create_compute_environment():
ec2_client, iam_client, batch_client = _get_clients()
def test_create_managed_compute_environment():
ec2_client, iam_client, ecs_client, batch_client = _get_clients()
vpc_id, subnet_id, sg_id, iam_arn = _setup(ec2_client, iam_client)
compute_name = 'test_compute_env'
@ -59,11 +61,12 @@ def test_create_compute_environment():
state='ENABLED',
computeResources={
'type': 'EC2',
'minvCpus': 123,
'maxvCpus': 123,
'desiredvCpus': 123,
'minvCpus': 5,
'maxvCpus': 10,
'desiredvCpus': 5,
'instanceTypes': [
'some_instance_type',
't2.small',
't2.medium'
],
'imageId': 'some_image_id',
'subnets': [
@ -85,4 +88,71 @@ def test_create_compute_environment():
resp.should.contain('computeEnvironmentArn')
resp['computeEnvironmentName'].should.equal(compute_name)
# Given a t2.medium is 2 vcpu and t2.small is 1, therefore 2 mediums and 1 small should be created
resp = ec2_client.describe_instances()
resp.should.contain('Reservations')
len(resp['Reservations']).should.equal(3)
# Should have created 1 ECS cluster
resp = ecs_client.list_clusters()
resp.should.contain('clusterArns')
len(resp['clusterArns']).should.equal(1)
@mock_ec2
@mock_ecs
@mock_iam
@mock_batch
def test_create_unmanaged_compute_environment():
ec2_client, iam_client, ecs_client, batch_client = _get_clients()
vpc_id, subnet_id, sg_id, iam_arn = _setup(ec2_client, iam_client)
compute_name = 'test_compute_env'
resp = batch_client.create_compute_environment(
computeEnvironmentName=compute_name,
type='UNMANAGED',
state='ENABLED',
serviceRole=iam_arn
)
resp.should.contain('computeEnvironmentArn')
resp['computeEnvironmentName'].should.equal(compute_name)
# Its unmanaged so no instances should be created
resp = ec2_client.describe_instances()
resp.should.contain('Reservations')
len(resp['Reservations']).should.equal(0)
# Should have created 1 ECS cluster
resp = ecs_client.list_clusters()
resp.should.contain('clusterArns')
len(resp['clusterArns']).should.equal(1)
# TODO create 1000s of tests to test complex option combinations of create environment
@mock_ec2
@mock_ecs
@mock_iam
@mock_batch
def test_describe_compute_environment():
ec2_client, iam_client, ecs_client, batch_client = _get_clients()
vpc_id, subnet_id, sg_id, iam_arn = _setup(ec2_client, iam_client)
compute_name = 'test_compute_env'
batch_client.create_compute_environment(
computeEnvironmentName=compute_name,
type='UNMANAGED',
state='ENABLED',
serviceRole=iam_arn
)
resp = batch_client.describe_compute_environments()
len(resp['computeEnvironments']).should.equal(1)
resp['computeEnvironments'][0]['computeEnvironmentName'].should.equal(compute_name)
# Test filtering
resp = batch_client.describe_compute_environments(
computeEnvironments=['test1']
)
len(resp['computeEnvironments']).should.equal(0)