Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix runningTasksCount ECS container instance attribute

Browse Source 
ECS container instances have attributes of 'runningTasksCount' and 'pendingTasksCount'.
See Boto3 docs here: http://boto3.readthedocs.io/en/latest/reference/services/ecs.html#ECS.Client.describe_container_instances
REST API docs here: http://docs.aws.amazon.com/AmazonECS/latest/APIReference/API_DescribeContainerInstances.html#API_DescribeContainerInstances_ResponseSyntax
This commit is contained in:
Tom Elliff 2017-10-12 13:26:19 +01:00
parent 098dd3fa1c
commit 51afd54229
2 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
moto/ecs/models.py
View File

@ -289,7 +289,7 @@ class ContainerInstance(BaseObject):
'type': 'STRINGSET'}] 'type': 'STRINGSET'}]
self.container_instance_arn = "arn:aws:ecs:us-east-1:012345678910:container-instance/{0}".format( self.container_instance_arn = "arn:aws:ecs:us-east-1:012345678910:container-instance/{0}".format(
str(uuid.uuid1())) str(uuid.uuid1()))
self.pending_task_count = 0 self.pending_tasks_count = 0
self.remaining_resources = [ self.remaining_resources = [
{'doubleValue': 0.0, {'doubleValue': 0.0,
'integerValue': 4096, 'integerValue': 4096,
@ -314,7 +314,7 @@ class ContainerInstance(BaseObject):
'stringSetValue': [], 'stringSetValue': [],
'type': 'STRINGSET'} 'type': 'STRINGSET'}
] ]
self.running_task_count = 0 self.running_tasks_count = 0
self.version_info = { self.version_info = {
'agentVersion': "1.0.0", 'agentVersion': "1.0.0",
'agentHash': '4023248', 'agentHash': '4023248',
@ -737,7 +737,7 @@ class EC2ContainerServiceBackend(BaseBackend):
resource["stringSetValue"].remove(str(port)) resource["stringSetValue"].remove(str(port))
else: else:
resource["stringSetValue"].append(str(port)) resource["stringSetValue"].append(str(port))
container_instance.running_task_count += resource_multiplier * 1 container_instance.running_tasks_count += resource_multiplier * 1
def deregister_container_instance(self, cluster_str, container_instance_str, force): def deregister_container_instance(self, cluster_str, container_instance_str, force):
failures = [] failures = []
@ -748,11 +748,11 @@ class EC2ContainerServiceBackend(BaseBackend):
container_instance = self.container_instances[cluster_name].get(container_instance_id) container_instance = self.container_instances[cluster_name].get(container_instance_id)
if container_instance is None: if container_instance is None:
raise Exception("{0} is not a container id in the cluster") raise Exception("{0} is not a container id in the cluster")
if not force and container_instance.running_task_count > 0: if not force and container_instance.running_tasks_count > 0:
raise Exception("Found running tasks on the instance.") raise Exception("Found running tasks on the instance.")
# Currently assume that people might want to do something based around deregistered instances # Currently assume that people might want to do something based around deregistered instances
# with tasks left running on them - but nothing if no tasks were running already # with tasks left running on them - but nothing if no tasks were running already
elif force and container_instance.running_task_count > 0: elif force and container_instance.running_tasks_count > 0:
if not self.container_instances.get('orphaned'): if not self.container_instances.get('orphaned'):
self.container_instances['orphaned'] = {} self.container_instances['orphaned'] = {}
self.container_instances['orphaned'][container_instance_id] = container_instance self.container_instances['orphaned'][container_instance_id] = container_instance

2
tests/test_ecs/test_ecs_boto3.py
View File

@ -1210,6 +1210,7 @@ def test_resource_reservation_and_release():
remaining_resources['MEMORY'].should.equal(registered_resources['MEMORY'] - 400) remaining_resources['MEMORY'].should.equal(registered_resources['MEMORY'] - 400)
registered_resources['PORTS'].append('80') registered_resources['PORTS'].append('80')
remaining_resources['PORTS'].should.equal(registered_resources['PORTS']) remaining_resources['PORTS'].should.equal(registered_resources['PORTS'])
container_instance_description['runningTasksCount'].should.equal(1)
client.stop_task( client.stop_task(
cluster='test_ecs_cluster', cluster='test_ecs_cluster',
task=run_response['tasks'][0].get('taskArn'), task=run_response['tasks'][0].get('taskArn'),
@ -1223,6 +1224,7 @@ def test_resource_reservation_and_release():
remaining_resources['CPU'].should.equal(registered_resources['CPU']) remaining_resources['CPU'].should.equal(registered_resources['CPU'])
remaining_resources['MEMORY'].should.equal(registered_resources['MEMORY']) remaining_resources['MEMORY'].should.equal(registered_resources['MEMORY'])
remaining_resources['PORTS'].should.equal(registered_resources['PORTS']) remaining_resources['PORTS'].should.equal(registered_resources['PORTS'])
container_instance_description['runningTasksCount'].should.equal(0)
@mock_ecs @mock_ecs