Set the environment variable to start the agent with 'task-iamrole' capability

Author: aaithal

ecs-init/docker/docker.go

@@ -142,6 +142,7 @@ func (c *Client) getContainerConfig() *godocker.Config {
 		"ECS_UPDATE_DOWNLOAD_DIR":       config.CacheDirectory(),
 		"ECS_UPDATES_ENABLED":           "true",
 		"ECS_AVAILABLE_LOGGING_DRIVERS": `["json-file","syslog","awslogs"]`,
+		"ECS_ENABLE_TASK_IAM_ROLE":      "true",
 	}
 
 	// merge in user-supplied environment variables

ecs-init/docker/docker_test.go

@@ -225,6 +225,7 @@ func validateCommonCreateContainerOptions(opts godocker.CreateContainerOptions,
 	expectKey("ECS_UPDATE_DOWNLOAD_DIR="+config.CacheDirectory(), envVariables, t)
 	expectKey("ECS_UPDATES_ENABLED=true", envVariables, t)
 	expectKey(`ECS_AVAILABLE_LOGGING_DRIVERS=["json-file","syslog","awslogs"]`, envVariables, t)
+	expectKey("ECS_ENABLE_TASK_IAM_ROLE=true", envVariables, t)
 
 	if len(cfg.ExposedPorts) != 2 {
 		t.Errorf("Expected exactly 2 elements to be in ExposedPorts, but was %d", len(cfg.ExposedPorts))
@@ -272,7 +273,7 @@ func expectLocalhostTCPPortBindings(input map[godocker.Port][]godocker.PortBindi
 			}
 		}
 	} else {
-		t.Error("Expected 51678/tcp to be defined")
+		t.Errorf("Expected %s/tcp to be defined", port)
 	}
 }