catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Add add, remove, list endpoints for SSM tags

Browse source
This commit is contained in:
James Brennan 2017-07-26 11:38:12 +00:00
commit a5089c3d69
3 changed files with 71 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
moto/ssm/models.py
View file

@ -1,5 +1,7 @@
from __future__ import unicode_literals
from collections import defaultdict
from moto.core import BaseBackend, BaseModel
from moto.ec2 import ec2_backends
@ -42,6 +44,7 @@ class SimpleSystemManagerBackend(BaseBackend):
def __init__(self):
self._parameters = {}
self._resource_tags = defaultdict(lambda: defaultdict(dict))
def delete_parameter(self, name):
try:
@ -68,6 +71,19 @@ class SimpleSystemManagerBackend(BaseBackend):
self._parameters[name] = Parameter(
name, value, type, description, keyid)
def add_tags_to_resource(self, resource_type, resource_id, tags):
for key, value in tags.items():
self._resource_tags[resource_type][resource_id][key] = value
def remove_tags_from_resource(self, resource_type, resource_id, keys):
tags = self._resource_tags[resource_type][resource_id]
for key in keys:
if key in tags:
del tags[key]
def list_tags_for_resource(self, resource_type, resource_id):
return self._resource_tags[resource_type][resource_id]
ssm_backends = {}
for region, ec2_backend in ec2_backends.items():