-
Notifications
You must be signed in to change notification settings - Fork 3.7k
/
batch.test.ts
101 lines (97 loc) · 2.3 KB
/
batch.test.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import { expect, haveResource } from '@aws-cdk/assert';
import * as batch from '@aws-cdk/aws-batch';
import { ContainerImage } from '@aws-cdk/aws-ecs';
import * as events from '@aws-cdk/aws-events';
import { Stack } from '@aws-cdk/core';
import * as targets from '../../lib';
test('use aws batch job as an eventrule target', () => {
// GIVEN
const stack = new Stack();
const jobQueue = new batch.JobQueue(stack, 'MyQueue', {
computeEnvironments: [
{
computeEnvironment: new batch.ComputeEnvironment(stack, 'ComputeEnvironment', {
managed: false
}),
order: 1
}
]
});
const jobDefinition = new batch.JobDefinition(stack, 'MyJob', {
container: {
image: ContainerImage.fromRegistry('test-repo')
}
});
const rule = new events.Rule(stack, 'Rule', {
schedule: events.Schedule.expression('rate(1 min)')
});
// WHEN
rule.addTarget(new targets.BatchJob(jobQueue, jobDefinition));
// THEN
expect(stack).to(haveResource('AWS::Events::Rule', {
ScheduleExpression: "rate(1 min)",
State: "ENABLED",
Targets: [
{
Arn: {
Ref: "MyQueueE6CA6235"
},
Id: "Target0",
RoleArn: {
"Fn::GetAtt": [
"MyJobEventsRoleCF43C336",
"Arn"
]
}
}
]
}));
expect(stack).to(haveResource('AWS::IAM::Role', {
AssumeRolePolicyDocument: {
Statement: [
{
Action: "sts:AssumeRole",
Effect: "Allow",
Principal: {
Service: "batch.amazonaws.com"
}
}
],
Version: "2012-10-17"
},
ManagedPolicyArns: [
{
"Fn::Join": [
"",
[
"arn:",
{
Ref: "AWS::Partition"
},
":iam::aws:policy/service-role/AWSBatchServiceRole"
]
]
}
]
}));
expect(stack).to(haveResource('AWS::IAM::Policy', {
PolicyDocument: {
Statement: [
{
Action: "batch:SubmitJob",
Effect: "Allow",
Resource: {
Ref: "MyJob8719E923"
}
}
],
Version: "2012-10-17"
},
PolicyName: "MyJobEventsRoleDefaultPolicy7266D3A7",
Roles: [
{
Ref: "MyJobEventsRoleCF43C336"
}
]
}));
});