davidchristiansen/LoadBalancedFargateService-ContainerNameAndRoleOver…

…ridesImplemenation of execution tests, task role andcontainer name overrides + tests

packages/@aws-cdk/aws-ecs-patterns/lib/fargate/load-balanced-fargate-service.ts

@@ -1,4 +1,6 @@
 import ecs = require('@aws-cdk/aws-ecs');
+import iam = require('@aws-cdk/aws-iam');
+import { AliasRecord, IHostedZone } from '@aws-cdk/aws-route53';
 import { AddressRecordTarget, ARecord, IHostedZone } from '@aws-cdk/aws-route53';
 import targets = require('@aws-cdk/aws-route53-targets');
 import cdk = require('@aws-cdk/cdk');
@@ -72,6 +74,34 @@ export interface LoadBalancedFargateServiceProps extends LoadBalancedServiceBase
    * @default true
    */
   readonly createLogs?: boolean;
+
+  /**
+   * Override for the Fargate Task Definition execution role
+   *
+   * @default - No value
+   */
+  readonly executionRole?: iam.IRole;
+
+  /**
+   * Override for the Fargate Task Definition task role
+   *
+   * @default - No value
+   */
+  readonly taskRole?: iam.IRole;
+
+  /**
+   * Override value for the container name
+   *
+   * @default - No value
+   */
+  readonly containerName?: string;
+
+  /**
+   * Override value for the service name
+   *
+   * @default - No value
+   */
+  readonly serviceName?: string;
 }
 
 /**
@@ -89,12 +119,16 @@ export class LoadBalancedFargateService extends LoadBalancedServiceBase {
 
     const taskDefinition = new ecs.FargateTaskDefinition(this, 'TaskDef', {
       memoryMiB: props.memoryMiB,
-      cpu: props.cpu
+      cpu: props.cpu,
+      executionRole: props.executionRole !== undefined ? props.executionRole : undefined,
+      taskRole: props.taskRole !== undefined ? props.taskRole : undefined
     });
 
     const optIn = props.createLogs !== undefined ? props.createLogs : true;
 
-    const container = taskDefinition.addContainer('web', {
+    const containerName = props.containerName !== undefined ? props.containerName : 'web';
+
+    const container = taskDefinition.addContainer(containerName, {
       image: props.image,
       logging: optIn ? this.createAWSLogDriver(this.node.id) : undefined,
       environment: props.environment
@@ -109,7 +143,8 @@ export class LoadBalancedFargateService extends LoadBalancedServiceBase {
       cluster: props.cluster,
       desiredCount: props.desiredCount || 1,
       taskDefinition,
-      assignPublicIp
+      assignPublicIp,
+      serviceName: props.serviceName || undefined
     });
     this.service = service;