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

First cut of container instance deregistration

Browse Source
This commit is contained in:
GuyTempleton 2017-04-13 08:54:25 +01:00
parent 30b1de507c
commit 71e8f6ef5b
No known key found for this signature in database
GPG Key ID: 64359D336E412597
2 changed files with 36 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
moto/ecs/models.py
View File

@ -716,7 +716,31 @@ class EC2ContainerServiceBackend(BaseBackend):
else:
resource["stringSetValue"].append(str(port))
def deregister_container_instance(self, cluster_str, container_instance_str):
def deregister_container_instance(self, cluster_str, container_instance_str, force):
cluster_name = cluster_str.split('/')[-1]
if cluster_name not in self.clusters:
raise Exception("{0} is not a cluster".format(cluster_name))
container_instance_id = container_instance_str.split('/')[-1]
container_instance = self.container_instances[cluster_name].get(container_instance_id)
if container_instance is None:
raise Exception("{0} is not a container id in the cluster")
if not force and container_instance.running_tasks_count > 0:
raise Exception("Found running tasks on the instance.")
# Currently assume that people might want to do something based around deregistered instances
# with tasks left running on them - but nothing if deregistration is forced or no tasks were
# running already
elif force and container_instance.running_tasks_count > 0:
if not self.container_instances.get('orphaned'):
self.container_instances['orphaned'] = {}
self.container_instances['orphaned'][container_instance_id] = container_instance
del(self.container_instances[cluster_name][container_instance_id])
self._respond_to_cluster_state_update(cluster_str)
pass
def _respond_to_cluster_state_update(self, cluster_str):
cluster_name = cluster_str.split('/')[-1]
if cluster_name not in self.clusters:
raise Exception("{0} is not a cluster".format(cluster_name))
pass

11
moto/ecs/responses.py
View File

@ -203,6 +203,17 @@ class EC2ContainerServiceResponse(BaseResponse):
'containerInstance': container_instance.response_object
})
def deregister_container_instance(self):
cluster_str = self._get_param('cluster', 'default'),
container_instance_str = self._get_param('containerInstance')
force = self._get_param('force', False)
container_instance = self.ecs_backend.deregister_container_instance(
cluster_str, container_instance_str, force
)
return json.dumps({
'containerInstance': container_instance.response_object
})
def list_container_instances(self):
cluster_str = self._get_param('cluster')
container_instance_arns = self.ecs_backend.list_container_instances(