Implemented Terminate, Cancel and List jobs

tests/test_batch/test_batch.py

@@ -583,7 +583,7 @@ def test_describe_task_definition():
     len(resp['jobDefinitions']).should.equal(3)
 
 
-# SLOW TEST
+# SLOW TESTS
 @expected_failure
 @mock_logs
 @mock_ec2
@@ -654,4 +654,156 @@ def test_submit_job():
     ls_name = resp['logStreams'][0]['logStreamName']
 
     resp = logs_client.get_log_events(logGroupName='/aws/batch/job', logStreamName=ls_name)
-    len(resp['events']).should.be.greater_than(5)
+    len(resp['events']).should.be.greater_than(5)
+
+
+@expected_failure
+@mock_logs
+@mock_ec2
+@mock_ecs
+@mock_iam
+@mock_batch
+def test_list_jobs():
+    ec2_client, iam_client, ecs_client, logs_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
+    )
+    arn = resp['computeEnvironmentArn']
+
+    resp = batch_client.create_job_queue(
+        jobQueueName='test_job_queue',
+        state='ENABLED',
+        priority=123,
+        computeEnvironmentOrder=[
+            {
+                'order': 123,
+                'computeEnvironment': arn
+            },
+        ]
+    )
+    queue_arn = resp['jobQueueArn']
+
+    resp = batch_client.register_job_definition(
+        jobDefinitionName='sleep10',
+        type='container',
+        containerProperties={
+            'image': 'busybox',
+            'vcpus': 1,
+            'memory': 128,
+            'command': ['sleep', '10']
+        }
+    )
+    job_def_arn = resp['jobDefinitionArn']
+
+    resp = batch_client.submit_job(
+        jobName='test1',
+        jobQueue=queue_arn,
+        jobDefinition=job_def_arn
+    )
+    job_id1 = resp['jobId']
+    resp = batch_client.submit_job(
+        jobName='test2',
+        jobQueue=queue_arn,
+        jobDefinition=job_def_arn
+    )
+    job_id2 = resp['jobId']
+
+    future = datetime.datetime.now() + datetime.timedelta(seconds=30)
+
+    resp_finished_jobs = batch_client.list_jobs(
+        jobQueue=queue_arn,
+        jobStatus='SUCCEEDED'
+    )
+
+    # Wait only as long as it takes to run the jobs
+    while datetime.datetime.now() < future:
+        resp = batch_client.describe_jobs(jobs=[job_id1, job_id2])
+
+        any_failed_jobs = any([job['status'] == 'FAILED' for job in resp['jobs']])
+        succeeded_jobs = all([job['status'] == 'SUCCEEDED' for job in resp['jobs']])
+
+        if any_failed_jobs:
+            raise RuntimeError('A Batch job failed')
+        if succeeded_jobs:
+            break
+        time.sleep(0.5)
+    else:
+        raise RuntimeError('Batch jobs timed out')
+
+    resp_finished_jobs2 = batch_client.list_jobs(
+        jobQueue=queue_arn,
+        jobStatus='SUCCEEDED'
+    )
+
+    len(resp_finished_jobs['jobSummaryList']).should.equal(0)
+    len(resp_finished_jobs2['jobSummaryList']).should.equal(2)
+
+
+@expected_failure
+@mock_logs
+@mock_ec2
+@mock_ecs
+@mock_iam
+@mock_batch
+def test_terminate_job():
+    ec2_client, iam_client, ecs_client, logs_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
+    )
+    arn = resp['computeEnvironmentArn']
+
+    resp = batch_client.create_job_queue(
+        jobQueueName='test_job_queue',
+        state='ENABLED',
+        priority=123,
+        computeEnvironmentOrder=[
+            {
+                'order': 123,
+                'computeEnvironment': arn
+            },
+        ]
+    )
+    queue_arn = resp['jobQueueArn']
+
+    resp = batch_client.register_job_definition(
+        jobDefinitionName='sleep10',
+        type='container',
+        containerProperties={
+            'image': 'busybox',
+            'vcpus': 1,
+            'memory': 128,
+            'command': ['sleep', '10']
+        }
+    )
+    job_def_arn = resp['jobDefinitionArn']
+
+    resp = batch_client.submit_job(
+        jobName='test1',
+        jobQueue=queue_arn,
+        jobDefinition=job_def_arn
+    )
+    job_id = resp['jobId']
+
+    time.sleep(2)
+
+    batch_client.terminate_job(jobId=job_id, reason='test_terminate')
+
+    time.sleep(1)
+
+    resp = batch_client.describe_jobs(jobs=[job_id])
+    resp['jobs'][0]['jobName'].should.equal('test1')
+    resp['jobs'][0]['status'].should.equal('FAILED')
+    resp['jobs'][0]['statusReason'].should.equal('test_terminate')
+