Merge pull request #902 from mvantellingen/ecs-revision

Return the revision in ecs.register_task_definition
getmoto · Apr 16, 2017 · cc123da · cc123da
2 parents de28e7f + 18ed732
commit cc123da
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/moto/ecs/models.py b/moto/ecs/models.py
@@ -89,6 +89,7 @@ class TaskDefinition(BaseObject):
 
     def __init__(self, family, revision, container_definitions, volumes=None):
         self.family = family
+        self.revision = revision
         self.arn = 'arn:aws:ecs:us-east-1:012345678910:task-definition/{0}:{1}'.format(
             family, revision)
         self.container_definitions = container_definitions

diff --git a/tests/test_ecs/test_ecs_boto3.py b/tests/test_ecs/test_ecs_boto3.py
@@ -86,6 +86,7 @@ def test_register_task_definition():
         ]
     )
     type(response['taskDefinition']).should.be(dict)
+    response['taskDefinition']['revision'].should.equal(1)
     response['taskDefinition']['taskDefinitionArn'].should.equal(
         'arn:aws:ecs:us-east-1:012345678910:task-definition/test_ecs_task:1')
     response['taskDefinition']['containerDefinitions'][