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

feature emr: set_termination_protection call implementation

Browse source 
feature emr: support for instance_groups param in run_jobflow
This commit is contained in:
Shulman Evgeny 2015-07-08 10:26:58 +03:00
commit e498c95807
3 changed files with 76 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

47
tests/test_emr/test_emr.py
View file

@ -1,6 +1,8 @@
from __future__ import unicode_literals
import boto
from boto.emr.instance_group import InstanceGroup
from boto.emr.step import StreamingStep
import sure # noqa
@ -107,6 +109,27 @@ def test_create_job_flow_visible_to_all_users():
job_flow.visibletoallusers.should.equal('True')
@requires_boto_gte("2.8")
@mock_emr
def test_create_job_flow_with_instance_groups():
conn = boto.connect_emr()
instance_groups = [InstanceGroup(6, 'TASK', 'c1.medium', 'SPOT', 'spot-0.07', '0.07'),
InstanceGroup(6, 'TASK', 'c1.medium', 'SPOT', 'spot-0.07', '0.07')]
job_id = conn.run_jobflow(
name='My jobflow',
log_uri='s3://some_bucket/jobflow_logs',
steps=[],
instance_groups=instance_groups
)
job_flow = conn.describe_jobflow(job_id)
int(job_flow.instancecount).should.equal(12)
instance_group = job_flow.instancegroups[0]
int(instance_group.instancerunningcount).should.equal(6)
@mock_emr
def test_terminate_job_flow():
conn = boto.connect_emr()
@ -318,6 +341,30 @@ def test_set_visible_to_all_users():
job_flow.visibletoallusers.should.equal('False')
@requires_boto_gte("2.8")
@mock_emr
def test_set_termination_protection():
conn = boto.connect_emr()
job_id = conn.run_jobflow(
name='My jobflow',
log_uri='s3://some_bucket/jobflow_logs',
steps=[]
)
job_flow = conn.describe_jobflow(job_id)
job_flow.terminationprotected.should.equal(u'None')
conn.set_termination_protection(job_id, True)
job_flow = conn.describe_jobflow(job_id)
job_flow.terminationprotected.should.equal('true')
conn.set_termination_protection(job_id, False)
job_flow = conn.describe_jobflow(job_id)
job_flow.terminationprotected.should.equal('false')
@mock_emr
def test_list_clusters():
conn = boto.connect_emr()