[WIP] Resource tagging (#1332)
* Ran scaffold * Made a start on get_resources. * Added more EC2 to tag list * More stuff * Added more methods * Added S3 region name * Added values test * Some final touchups
This commit is contained in:
parent
474023f9a1
commit
8ee05e22af
@ -41,6 +41,7 @@ from .swf import mock_swf, mock_swf_deprecated # flake8: noqa
|
||||
from .xray import mock_xray, mock_xray_client, XRaySegment # flake8: noqa
|
||||
from .logs import mock_logs, mock_logs_deprecated # flake8: noqa
|
||||
from .batch import mock_batch # flake8: noqa
|
||||
from .resourcegroupstaggingapi import mock_resourcegroupstaggingapi # flake8: noqa
|
||||
from .iot import mock_iot # flake8: noqa
|
||||
from .iotdata import mock_iotdata # flake8: noqa
|
||||
|
||||
|
@ -38,6 +38,7 @@ from moto.xray import xray_backends
|
||||
from moto.iot import iot_backends
|
||||
from moto.iotdata import iotdata_backends
|
||||
from moto.batch import batch_backends
|
||||
from moto.resourcegroupstaggingapi import resourcegroupstaggingapi_backends
|
||||
|
||||
|
||||
BACKENDS = {
|
||||
@ -78,6 +79,7 @@ BACKENDS = {
|
||||
'route53': route53_backends,
|
||||
'lambda': lambda_backends,
|
||||
'xray': xray_backends,
|
||||
'resourcegroupstaggingapi': resourcegroupstaggingapi_backends,
|
||||
'iot': iot_backends,
|
||||
'iot-data': iotdata_backends,
|
||||
}
|
||||
|
6
moto/resourcegroupstaggingapi/__init__.py
Normal file
6
moto/resourcegroupstaggingapi/__init__.py
Normal file
@ -0,0 +1,6 @@
|
||||
from __future__ import unicode_literals
|
||||
from .models import resourcegroupstaggingapi_backends
|
||||
from ..core.models import base_decorator
|
||||
|
||||
resourcegroupstaggingapi_backend = resourcegroupstaggingapi_backends['us-east-1']
|
||||
mock_resourcegroupstaggingapi = base_decorator(resourcegroupstaggingapi_backends)
|
511
moto/resourcegroupstaggingapi/models.py
Normal file
511
moto/resourcegroupstaggingapi/models.py
Normal file
@ -0,0 +1,511 @@
|
||||
from __future__ import unicode_literals
|
||||
import uuid
|
||||
import boto3
|
||||
import six
|
||||
from moto.core import BaseBackend
|
||||
from moto.core.exceptions import RESTError
|
||||
|
||||
from moto.s3 import s3_backends
|
||||
from moto.ec2 import ec2_backends
|
||||
from moto.elb import elb_backends
|
||||
from moto.elbv2 import elbv2_backends
|
||||
from moto.kinesis import kinesis_backends
|
||||
from moto.rds2 import rds2_backends
|
||||
from moto.glacier import glacier_backends
|
||||
from moto.redshift import redshift_backends
|
||||
from moto.emr import emr_backends
|
||||
|
||||
# Left: EC2 ElastiCache RDS ELB CloudFront WorkSpaces Lambda EMR Glacier Kinesis Redshift Route53
|
||||
# StorageGateway DynamoDB MachineLearning ACM DirectConnect DirectoryService CloudHSM
|
||||
# Inspector Elasticsearch
|
||||
|
||||
|
||||
class ResourceGroupsTaggingAPIBackend(BaseBackend):
|
||||
def __init__(self, region_name=None):
|
||||
super(ResourceGroupsTaggingAPIBackend, self).__init__()
|
||||
self.region_name = region_name
|
||||
|
||||
self._pages = {}
|
||||
# Like 'someuuid': {'gen': <generator>, 'misc': None}
|
||||
# Misc is there for peeking from a generator and it cant
|
||||
# fit in the current request. As we only store generators
|
||||
# theres not really any point to clean up
|
||||
|
||||
def reset(self):
|
||||
region_name = self.region_name
|
||||
self.__dict__ = {}
|
||||
self.__init__(region_name)
|
||||
|
||||
@property
|
||||
def s3_backend(self):
|
||||
"""
|
||||
:rtype: moto.s3.models.S3Backend
|
||||
"""
|
||||
return s3_backends['global']
|
||||
|
||||
@property
|
||||
def ec2_backend(self):
|
||||
"""
|
||||
:rtype: moto.ec2.models.EC2Backend
|
||||
"""
|
||||
return ec2_backends[self.region_name]
|
||||
|
||||
@property
|
||||
def elb_backend(self):
|
||||
"""
|
||||
:rtype: moto.elb.models.ELBBackend
|
||||
"""
|
||||
return elb_backends[self.region_name]
|
||||
|
||||
@property
|
||||
def elbv2_backend(self):
|
||||
"""
|
||||
:rtype: moto.elbv2.models.ELBv2Backend
|
||||
"""
|
||||
return elbv2_backends[self.region_name]
|
||||
|
||||
@property
|
||||
def kinesis_backend(self):
|
||||
"""
|
||||
:rtype: moto.kinesis.models.KinesisBackend
|
||||
"""
|
||||
return kinesis_backends[self.region_name]
|
||||
|
||||
@property
|
||||
def rds_backend(self):
|
||||
"""
|
||||
:rtype: moto.rds2.models.RDS2Backend
|
||||
"""
|
||||
return rds2_backends[self.region_name]
|
||||
|
||||
@property
|
||||
def glacier_backend(self):
|
||||
"""
|
||||
:rtype: moto.glacier.models.GlacierBackend
|
||||
"""
|
||||
return glacier_backends[self.region_name]
|
||||
|
||||
@property
|
||||
def emr_backend(self):
|
||||
"""
|
||||
:rtype: moto.emr.models.ElasticMapReduceBackend
|
||||
"""
|
||||
return emr_backends[self.region_name]
|
||||
|
||||
@property
|
||||
def redshift_backend(self):
|
||||
"""
|
||||
:rtype: moto.redshift.models.RedshiftBackend
|
||||
"""
|
||||
return redshift_backends[self.region_name]
|
||||
|
||||
def _get_resources_generator(self, tag_filters=None, resource_type_filters=None):
|
||||
# Look at
|
||||
# https://docs.aws.amazon.com/general/latest/gr/aws-arns-and-namespaces.html
|
||||
|
||||
# TODO move these to their respective backends
|
||||
filters = [lambda t, v: True]
|
||||
for tag_filter_dict in tag_filters:
|
||||
values = tag_filter_dict.get('Values', [])
|
||||
if len(values) == 0:
|
||||
# Check key matches
|
||||
filters.append(lambda t, v: t == tag_filter_dict['Key'])
|
||||
elif len(values) == 1:
|
||||
# Check its exactly the same as key, value
|
||||
filters.append(lambda t, v: t == tag_filter_dict['Key'] and v == values[0])
|
||||
else:
|
||||
# Check key matches and value is one of the provided values
|
||||
filters.append(lambda t, v: t == tag_filter_dict['Key'] and v in values)
|
||||
|
||||
def tag_filter(tag_list):
|
||||
result = []
|
||||
|
||||
for tag in tag_list:
|
||||
temp_result = []
|
||||
for f in filters:
|
||||
f_result = f(tag['Key'], tag['Value'])
|
||||
temp_result.append(f_result)
|
||||
result.append(all(temp_result))
|
||||
|
||||
return any(result)
|
||||
|
||||
# Do S3, resource type s3
|
||||
if not resource_type_filters or 's3' in resource_type_filters:
|
||||
for bucket in self.s3_backend.buckets.values():
|
||||
tags = []
|
||||
for tag in bucket.tags.tag_set.tags:
|
||||
tags.append({'Key': tag.key, 'Value': tag.value})
|
||||
|
||||
if not tags or not tag_filter(tags): # Skip if no tags, or invalid filter
|
||||
continue
|
||||
yield {'ResourceARN': 'arn:aws:s3:::' + bucket.name, 'Tags': tags}
|
||||
|
||||
# EC2 tags
|
||||
def get_ec2_tags(res_id):
|
||||
result = []
|
||||
for key, value in self.ec2_backend.tags.get(res_id, {}).items():
|
||||
result.append({'Key': key, 'Value': value})
|
||||
return result
|
||||
|
||||
# EC2 AMI, resource type ec2:image
|
||||
if not resource_type_filters or 'ec2' in resource_type_filters or 'ec2:image' in resource_type_filters:
|
||||
for ami in self.ec2_backend.amis.values():
|
||||
tags = get_ec2_tags(ami.id)
|
||||
|
||||
if not tags or not tag_filter(tags): # Skip if no tags, or invalid filter
|
||||
continue
|
||||
yield {'ResourceARN': 'arn:aws:ec2:{0}::image/{1}'.format(self.region_name, ami.id), 'Tags': tags}
|
||||
|
||||
# EC2 Instance, resource type ec2:instance
|
||||
if not resource_type_filters or 'ec2' in resource_type_filters or 'ec2:instance' in resource_type_filters:
|
||||
for reservation in self.ec2_backend.reservations.values():
|
||||
for instance in reservation.instances:
|
||||
tags = get_ec2_tags(instance.id)
|
||||
|
||||
if not tags or not tag_filter(tags): # Skip if no tags, or invalid filter
|
||||
continue
|
||||
yield {'ResourceARN': 'arn:aws:ec2:{0}::instance/{1}'.format(self.region_name, instance.id), 'Tags': tags}
|
||||
|
||||
# EC2 NetworkInterface, resource type ec2:network-interface
|
||||
if not resource_type_filters or 'ec2' in resource_type_filters or 'ec2:network-interface' in resource_type_filters:
|
||||
for eni in self.ec2_backend.enis.values():
|
||||
tags = get_ec2_tags(eni.id)
|
||||
|
||||
if not tags or not tag_filter(tags): # Skip if no tags, or invalid filter
|
||||
continue
|
||||
yield {'ResourceARN': 'arn:aws:ec2:{0}::network-interface/{1}'.format(self.region_name, eni.id), 'Tags': tags}
|
||||
|
||||
# TODO EC2 ReservedInstance
|
||||
|
||||
# EC2 SecurityGroup, resource type ec2:security-group
|
||||
if not resource_type_filters or 'ec2' in resource_type_filters or 'ec2:security-group' in resource_type_filters:
|
||||
for vpc in self.ec2_backend.groups.values():
|
||||
for sg in vpc.values():
|
||||
tags = get_ec2_tags(sg.id)
|
||||
|
||||
if not tags or not tag_filter(tags): # Skip if no tags, or invalid filter
|
||||
continue
|
||||
yield {'ResourceARN': 'arn:aws:ec2:{0}::security-group/{1}'.format(self.region_name, sg.id), 'Tags': tags}
|
||||
|
||||
# EC2 Snapshot, resource type ec2:snapshot
|
||||
if not resource_type_filters or 'ec2' in resource_type_filters or 'ec2:snapshot' in resource_type_filters:
|
||||
for snapshot in self.ec2_backend.snapshots.values():
|
||||
tags = get_ec2_tags(snapshot.id)
|
||||
|
||||
if not tags or not tag_filter(tags): # Skip if no tags, or invalid filter
|
||||
continue
|
||||
yield {'ResourceARN': 'arn:aws:ec2:{0}::snapshot/{1}'.format(self.region_name, snapshot.id), 'Tags': tags}
|
||||
|
||||
# TODO EC2 SpotInstanceRequest
|
||||
|
||||
# EC2 Volume, resource type ec2:volume
|
||||
if not resource_type_filters or 'ec2' in resource_type_filters or 'ec2:volume' in resource_type_filters:
|
||||
for volume in self.ec2_backend.volumes.values():
|
||||
tags = get_ec2_tags(volume.id)
|
||||
|
||||
if not tags or not tag_filter(tags): # Skip if no tags, or invalid filter
|
||||
continue
|
||||
yield {'ResourceARN': 'arn:aws:ec2:{0}::volume/{1}'.format(self.region_name, volume.id), 'Tags': tags}
|
||||
|
||||
# TODO add these to the keys and values functions / combine functions
|
||||
# ELB
|
||||
|
||||
# EMR Cluster
|
||||
|
||||
# Glacier Vault
|
||||
|
||||
# Kinesis
|
||||
|
||||
# RDS Instance
|
||||
# RDS Reserved Database Instance
|
||||
# RDS Option Group
|
||||
# RDS Parameter Group
|
||||
# RDS Security Group
|
||||
# RDS Snapshot
|
||||
# RDS Subnet Group
|
||||
# RDS Event Subscription
|
||||
|
||||
# RedShift Cluster
|
||||
# RedShift Hardware security module (HSM) client certificate
|
||||
# RedShift HSM connection
|
||||
# RedShift Parameter group
|
||||
# RedShift Snapshot
|
||||
# RedShift Subnet group
|
||||
|
||||
# VPC
|
||||
# VPC Customer Gateway
|
||||
# VPC DHCP Option Set
|
||||
# VPC Internet Gateway
|
||||
# VPC Network ACL
|
||||
# VPC Route Table
|
||||
# VPC Subnet
|
||||
# VPC Virtual Private Gateway
|
||||
# VPC VPN Connection
|
||||
|
||||
def _get_tag_keys_generator(self):
|
||||
# Look at
|
||||
# https://docs.aws.amazon.com/general/latest/gr/aws-arns-and-namespaces.html
|
||||
|
||||
# Do S3, resource type s3
|
||||
for bucket in self.s3_backend.buckets.values():
|
||||
for tag in bucket.tags.tag_set.tags:
|
||||
yield tag.key
|
||||
|
||||
# EC2 tags
|
||||
def get_ec2_keys(res_id):
|
||||
result = []
|
||||
for key in self.ec2_backend.tags.get(res_id, {}):
|
||||
result.append(key)
|
||||
return result
|
||||
|
||||
# EC2 AMI, resource type ec2:image
|
||||
for ami in self.ec2_backend.amis.values():
|
||||
for key in get_ec2_keys(ami.id):
|
||||
yield key
|
||||
|
||||
# EC2 Instance, resource type ec2:instance
|
||||
for reservation in self.ec2_backend.reservations.values():
|
||||
for instance in reservation.instances:
|
||||
for key in get_ec2_keys(instance.id):
|
||||
yield key
|
||||
|
||||
# EC2 NetworkInterface, resource type ec2:network-interface
|
||||
for eni in self.ec2_backend.enis.values():
|
||||
for key in get_ec2_keys(eni.id):
|
||||
yield key
|
||||
|
||||
# TODO EC2 ReservedInstance
|
||||
|
||||
# EC2 SecurityGroup, resource type ec2:security-group
|
||||
for vpc in self.ec2_backend.groups.values():
|
||||
for sg in vpc.values():
|
||||
for key in get_ec2_keys(sg.id):
|
||||
yield key
|
||||
|
||||
# EC2 Snapshot, resource type ec2:snapshot
|
||||
for snapshot in self.ec2_backend.snapshots.values():
|
||||
for key in get_ec2_keys(snapshot.id):
|
||||
yield key
|
||||
|
||||
# TODO EC2 SpotInstanceRequest
|
||||
|
||||
# EC2 Volume, resource type ec2:volume
|
||||
for volume in self.ec2_backend.volumes.values():
|
||||
for key in get_ec2_keys(volume.id):
|
||||
yield key
|
||||
|
||||
def _get_tag_values_generator(self, tag_key):
|
||||
# Look at
|
||||
# https://docs.aws.amazon.com/general/latest/gr/aws-arns-and-namespaces.html
|
||||
|
||||
# Do S3, resource type s3
|
||||
for bucket in self.s3_backend.buckets.values():
|
||||
for tag in bucket.tags.tag_set.tags:
|
||||
if tag.key == tag_key:
|
||||
yield tag.value
|
||||
|
||||
# EC2 tags
|
||||
def get_ec2_values(res_id):
|
||||
result = []
|
||||
for key, value in self.ec2_backend.tags.get(res_id, {}).items():
|
||||
if key == tag_key:
|
||||
result.append(value)
|
||||
return result
|
||||
|
||||
# EC2 AMI, resource type ec2:image
|
||||
for ami in self.ec2_backend.amis.values():
|
||||
for value in get_ec2_values(ami.id):
|
||||
yield value
|
||||
|
||||
# EC2 Instance, resource type ec2:instance
|
||||
for reservation in self.ec2_backend.reservations.values():
|
||||
for instance in reservation.instances:
|
||||
for value in get_ec2_values(instance.id):
|
||||
yield value
|
||||
|
||||
# EC2 NetworkInterface, resource type ec2:network-interface
|
||||
for eni in self.ec2_backend.enis.values():
|
||||
for value in get_ec2_values(eni.id):
|
||||
yield value
|
||||
|
||||
# TODO EC2 ReservedInstance
|
||||
|
||||
# EC2 SecurityGroup, resource type ec2:security-group
|
||||
for vpc in self.ec2_backend.groups.values():
|
||||
for sg in vpc.values():
|
||||
for value in get_ec2_values(sg.id):
|
||||
yield value
|
||||
|
||||
# EC2 Snapshot, resource type ec2:snapshot
|
||||
for snapshot in self.ec2_backend.snapshots.values():
|
||||
for value in get_ec2_values(snapshot.id):
|
||||
yield value
|
||||
|
||||
# TODO EC2 SpotInstanceRequest
|
||||
|
||||
# EC2 Volume, resource type ec2:volume
|
||||
for volume in self.ec2_backend.volumes.values():
|
||||
for value in get_ec2_values(volume.id):
|
||||
yield value
|
||||
|
||||
def get_resources(self, pagination_token=None,
|
||||
resources_per_page=50, tags_per_page=100,
|
||||
tag_filters=None, resource_type_filters=None):
|
||||
# Simple range checning
|
||||
if 100 >= tags_per_page >= 500:
|
||||
raise RESTError('InvalidParameterException', 'TagsPerPage must be between 100 and 500')
|
||||
if 1 >= resources_per_page >= 50:
|
||||
raise RESTError('InvalidParameterException', 'ResourcesPerPage must be between 1 and 50')
|
||||
|
||||
# If we have a token, go and find the respective generator, or error
|
||||
if pagination_token:
|
||||
if pagination_token not in self._pages:
|
||||
raise RESTError('PaginationTokenExpiredException', 'Token does not exist')
|
||||
|
||||
generator = self._pages[pagination_token]['gen']
|
||||
left_over = self._pages[pagination_token]['misc']
|
||||
else:
|
||||
generator = self._get_resources_generator(tag_filters=tag_filters,
|
||||
resource_type_filters=resource_type_filters)
|
||||
left_over = None
|
||||
|
||||
result = []
|
||||
current_tags = 0
|
||||
current_resources = 0
|
||||
if left_over:
|
||||
result.append(left_over)
|
||||
current_resources += 1
|
||||
current_tags += len(left_over['Tags'])
|
||||
|
||||
try:
|
||||
while True:
|
||||
# Generator format: [{'ResourceARN': str, 'Tags': [{'Key': str, 'Value': str]}, ...]
|
||||
next_item = six.next(generator)
|
||||
resource_tags = len(next_item['Tags'])
|
||||
|
||||
if current_resources >= resources_per_page:
|
||||
break
|
||||
if current_tags + resource_tags >= tags_per_page:
|
||||
break
|
||||
|
||||
current_resources += 1
|
||||
current_tags += resource_tags
|
||||
|
||||
result.append(next_item)
|
||||
|
||||
except StopIteration:
|
||||
# Finished generator before invalidating page limiting constraints
|
||||
return None, result
|
||||
|
||||
# Didn't hit StopIteration so there's stuff left in generator
|
||||
new_token = str(uuid.uuid4())
|
||||
self._pages[new_token] = {'gen': generator, 'misc': next_item}
|
||||
|
||||
# Token used up, might as well bin now, if you call it again your an idiot
|
||||
if pagination_token:
|
||||
del self._pages[pagination_token]
|
||||
|
||||
return new_token, result
|
||||
|
||||
def get_tag_keys(self, pagination_token=None):
|
||||
|
||||
if pagination_token:
|
||||
if pagination_token not in self._pages:
|
||||
raise RESTError('PaginationTokenExpiredException', 'Token does not exist')
|
||||
|
||||
generator = self._pages[pagination_token]['gen']
|
||||
left_over = self._pages[pagination_token]['misc']
|
||||
else:
|
||||
generator = self._get_tag_keys_generator()
|
||||
left_over = None
|
||||
|
||||
result = []
|
||||
current_tags = 0
|
||||
if left_over:
|
||||
result.append(left_over)
|
||||
current_tags += 1
|
||||
|
||||
try:
|
||||
while True:
|
||||
# Generator format: ['tag', 'tag', 'tag', ...]
|
||||
next_item = six.next(generator)
|
||||
|
||||
if current_tags + 1 >= 128:
|
||||
break
|
||||
|
||||
current_tags += 1
|
||||
|
||||
result.append(next_item)
|
||||
|
||||
except StopIteration:
|
||||
# Finished generator before invalidating page limiting constraints
|
||||
return None, result
|
||||
|
||||
# Didn't hit StopIteration so there's stuff left in generator
|
||||
new_token = str(uuid.uuid4())
|
||||
self._pages[new_token] = {'gen': generator, 'misc': next_item}
|
||||
|
||||
# Token used up, might as well bin now, if you call it again your an idiot
|
||||
if pagination_token:
|
||||
del self._pages[pagination_token]
|
||||
|
||||
return new_token, result
|
||||
|
||||
def get_tag_values(self, pagination_token, key):
|
||||
|
||||
if pagination_token:
|
||||
if pagination_token not in self._pages:
|
||||
raise RESTError('PaginationTokenExpiredException', 'Token does not exist')
|
||||
|
||||
generator = self._pages[pagination_token]['gen']
|
||||
left_over = self._pages[pagination_token]['misc']
|
||||
else:
|
||||
generator = self._get_tag_values_generator(key)
|
||||
left_over = None
|
||||
|
||||
result = []
|
||||
current_tags = 0
|
||||
if left_over:
|
||||
result.append(left_over)
|
||||
current_tags += 1
|
||||
|
||||
try:
|
||||
while True:
|
||||
# Generator format: ['value', 'value', 'value', ...]
|
||||
next_item = six.next(generator)
|
||||
|
||||
if current_tags + 1 >= 128:
|
||||
break
|
||||
|
||||
current_tags += 1
|
||||
|
||||
result.append(next_item)
|
||||
|
||||
except StopIteration:
|
||||
# Finished generator before invalidating page limiting constraints
|
||||
return None, result
|
||||
|
||||
# Didn't hit StopIteration so there's stuff left in generator
|
||||
new_token = str(uuid.uuid4())
|
||||
self._pages[new_token] = {'gen': generator, 'misc': next_item}
|
||||
|
||||
# Token used up, might as well bin now, if you call it again your an idiot
|
||||
if pagination_token:
|
||||
del self._pages[pagination_token]
|
||||
|
||||
return new_token, result
|
||||
|
||||
# These methods will be called from responses.py.
|
||||
# They should call a tag function inside of the moto module
|
||||
# that governs the resource, that way if the target module
|
||||
# changes how tags are delt with theres less to change
|
||||
|
||||
# def tag_resources(self, resource_arn_list, tags):
|
||||
# return failed_resources_map
|
||||
#
|
||||
# def untag_resources(self, resource_arn_list, tag_keys):
|
||||
# return failed_resources_map
|
||||
|
||||
|
||||
available_regions = boto3.session.Session().get_available_regions("resourcegroupstaggingapi")
|
||||
resourcegroupstaggingapi_backends = {region: ResourceGroupsTaggingAPIBackend(region) for region in available_regions}
|
97
moto/resourcegroupstaggingapi/responses.py
Normal file
97
moto/resourcegroupstaggingapi/responses.py
Normal file
@ -0,0 +1,97 @@
|
||||
from __future__ import unicode_literals
|
||||
from moto.core.responses import BaseResponse
|
||||
from .models import resourcegroupstaggingapi_backends
|
||||
import json
|
||||
|
||||
|
||||
class ResourceGroupsTaggingAPIResponse(BaseResponse):
|
||||
SERVICE_NAME = 'resourcegroupstaggingapi'
|
||||
|
||||
@property
|
||||
def backend(self):
|
||||
"""
|
||||
Backend
|
||||
:returns: Resource tagging api backend
|
||||
:rtype: moto.resourcegroupstaggingapi.models.ResourceGroupsTaggingAPIBackend
|
||||
"""
|
||||
return resourcegroupstaggingapi_backends[self.region]
|
||||
|
||||
def get_resources(self):
|
||||
pagination_token = self._get_param("PaginationToken")
|
||||
tag_filters = self._get_param("TagFilters", [])
|
||||
resources_per_page = self._get_int_param("ResourcesPerPage", 50)
|
||||
tags_per_page = self._get_int_param("TagsPerPage", 100)
|
||||
resource_type_filters = self._get_param("ResourceTypeFilters", [])
|
||||
|
||||
pagination_token, resource_tag_mapping_list = self.backend.get_resources(
|
||||
pagination_token=pagination_token,
|
||||
tag_filters=tag_filters,
|
||||
resources_per_page=resources_per_page,
|
||||
tags_per_page=tags_per_page,
|
||||
resource_type_filters=resource_type_filters,
|
||||
)
|
||||
|
||||
# Format tag response
|
||||
response = {
|
||||
'ResourceTagMappingList': resource_tag_mapping_list
|
||||
}
|
||||
if pagination_token:
|
||||
response['PaginationToken'] = pagination_token
|
||||
|
||||
return json.dumps(response)
|
||||
|
||||
def get_tag_keys(self):
|
||||
pagination_token = self._get_param("PaginationToken")
|
||||
pagination_token, tag_keys = self.backend.get_tag_keys(
|
||||
pagination_token=pagination_token,
|
||||
)
|
||||
|
||||
response = {
|
||||
'TagKeys': tag_keys
|
||||
}
|
||||
if pagination_token:
|
||||
response['PaginationToken'] = pagination_token
|
||||
|
||||
return json.dumps(response)
|
||||
|
||||
def get_tag_values(self):
|
||||
pagination_token = self._get_param("PaginationToken")
|
||||
key = self._get_param("Key")
|
||||
pagination_token, tag_values = self.backend.get_tag_values(
|
||||
pagination_token=pagination_token,
|
||||
key=key,
|
||||
)
|
||||
|
||||
response = {
|
||||
'TagValues': tag_values
|
||||
}
|
||||
if pagination_token:
|
||||
response['PaginationToken'] = pagination_token
|
||||
|
||||
return json.dumps(response)
|
||||
|
||||
# These methods are all thats left to be implemented
|
||||
# the response is already set up, all thats needed is
|
||||
# the respective model function to be implemented.
|
||||
#
|
||||
# def tag_resources(self):
|
||||
# resource_arn_list = self._get_list_prefix("ResourceARNList.member")
|
||||
# tags = self._get_param("Tags")
|
||||
# failed_resources_map = self.backend.tag_resources(
|
||||
# resource_arn_list=resource_arn_list,
|
||||
# tags=tags,
|
||||
# )
|
||||
#
|
||||
# # failed_resources_map should be {'resource': {'ErrorCode': str, 'ErrorMessage': str, 'StatusCode': int}}
|
||||
# return json.dumps({'FailedResourcesMap': failed_resources_map})
|
||||
#
|
||||
# def untag_resources(self):
|
||||
# resource_arn_list = self._get_list_prefix("ResourceARNList.member")
|
||||
# tag_keys = self._get_list_prefix("TagKeys.member")
|
||||
# failed_resources_map = self.backend.untag_resources(
|
||||
# resource_arn_list=resource_arn_list,
|
||||
# tag_keys=tag_keys,
|
||||
# )
|
||||
#
|
||||
# # failed_resources_map should be {'resource': {'ErrorCode': str, 'ErrorMessage': str, 'StatusCode': int}}
|
||||
# return json.dumps({'FailedResourcesMap': failed_resources_map})
|
10
moto/resourcegroupstaggingapi/urls.py
Normal file
10
moto/resourcegroupstaggingapi/urls.py
Normal file
@ -0,0 +1,10 @@
|
||||
from __future__ import unicode_literals
|
||||
from .responses import ResourceGroupsTaggingAPIResponse
|
||||
|
||||
url_bases = [
|
||||
"https?://tagging.(.+).amazonaws.com",
|
||||
]
|
||||
|
||||
url_paths = {
|
||||
'{0}/$': ResourceGroupsTaggingAPIResponse.dispatch,
|
||||
}
|
0
tests/test_resourcegroupstaggingapi/__init__.py
Normal file
0
tests/test_resourcegroupstaggingapi/__init__.py
Normal file
@ -0,0 +1,226 @@
|
||||
from __future__ import unicode_literals
|
||||
|
||||
import boto3
|
||||
import sure # noqa
|
||||
from moto import mock_resourcegroupstaggingapi, mock_s3, mock_ec2
|
||||
|
||||
|
||||
@mock_s3
|
||||
@mock_resourcegroupstaggingapi
|
||||
def test_get_resources_s3():
|
||||
# Tests pagination
|
||||
s3_client = boto3.client('s3', region_name='eu-central-1')
|
||||
|
||||
# Will end up having key1,key2,key3,key4
|
||||
response_keys = set()
|
||||
|
||||
# Create 4 buckets
|
||||
for i in range(1, 5):
|
||||
i_str = str(i)
|
||||
s3_client.create_bucket(Bucket='test_bucket' + i_str)
|
||||
s3_client.put_bucket_tagging(
|
||||
Bucket='test_bucket' + i_str,
|
||||
Tagging={'TagSet': [{'Key': 'key' + i_str, 'Value': 'value' + i_str}]}
|
||||
)
|
||||
response_keys.add('key' + i_str)
|
||||
|
||||
rtapi = boto3.client('resourcegroupstaggingapi', region_name='eu-central-1')
|
||||
resp = rtapi.get_resources(ResourcesPerPage=2)
|
||||
for resource in resp['ResourceTagMappingList']:
|
||||
response_keys.remove(resource['Tags'][0]['Key'])
|
||||
|
||||
response_keys.should.have.length_of(2)
|
||||
|
||||
resp = rtapi.get_resources(
|
||||
ResourcesPerPage=2,
|
||||
PaginationToken=resp['PaginationToken']
|
||||
)
|
||||
for resource in resp['ResourceTagMappingList']:
|
||||
response_keys.remove(resource['Tags'][0]['Key'])
|
||||
|
||||
response_keys.should.have.length_of(0)
|
||||
|
||||
|
||||
@mock_ec2
|
||||
@mock_resourcegroupstaggingapi
|
||||
def test_get_resources_ec2():
|
||||
client = boto3.client('ec2', region_name='eu-central-1')
|
||||
|
||||
instances = client.run_instances(
|
||||
ImageId='ami-123',
|
||||
MinCount=1,
|
||||
MaxCount=1,
|
||||
InstanceType='t2.micro',
|
||||
TagSpecifications=[
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG1',
|
||||
'Value': 'MY_VALUE1',
|
||||
},
|
||||
{
|
||||
'Key': 'MY_TAG2',
|
||||
'Value': 'MY_VALUE2',
|
||||
},
|
||||
],
|
||||
},
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG3',
|
||||
'Value': 'MY_VALUE3',
|
||||
},
|
||||
]
|
||||
},
|
||||
],
|
||||
)
|
||||
instance_id = instances['Instances'][0]['InstanceId']
|
||||
image_id = client.create_image(Name='testami', InstanceId=instance_id)['ImageId']
|
||||
|
||||
client.create_tags(
|
||||
Resources=[image_id],
|
||||
Tags=[{'Key': 'ami', 'Value': 'test'}]
|
||||
)
|
||||
|
||||
rtapi = boto3.client('resourcegroupstaggingapi', region_name='eu-central-1')
|
||||
resp = rtapi.get_resources()
|
||||
# Check we have 1 entry for Instance, 1 Entry for AMI
|
||||
resp['ResourceTagMappingList'].should.have.length_of(2)
|
||||
|
||||
# 1 Entry for AMI
|
||||
resp = rtapi.get_resources(ResourceTypeFilters=['ec2:image'])
|
||||
resp['ResourceTagMappingList'].should.have.length_of(1)
|
||||
resp['ResourceTagMappingList'][0]['ResourceARN'].should.contain('image/')
|
||||
|
||||
# As were iterating the same data, this rules out that the test above was a fluke
|
||||
resp = rtapi.get_resources(ResourceTypeFilters=['ec2:instance'])
|
||||
resp['ResourceTagMappingList'].should.have.length_of(1)
|
||||
resp['ResourceTagMappingList'][0]['ResourceARN'].should.contain('instance/')
|
||||
|
||||
# Basic test of tag filters
|
||||
resp = rtapi.get_resources(TagFilters=[{'Key': 'MY_TAG1', 'Values': ['MY_VALUE1', 'some_other_value']}])
|
||||
resp['ResourceTagMappingList'].should.have.length_of(1)
|
||||
resp['ResourceTagMappingList'][0]['ResourceARN'].should.contain('instance/')
|
||||
|
||||
|
||||
@mock_ec2
|
||||
@mock_resourcegroupstaggingapi
|
||||
def test_get_tag_keys_ec2():
|
||||
client = boto3.client('ec2', region_name='eu-central-1')
|
||||
|
||||
client.run_instances(
|
||||
ImageId='ami-123',
|
||||
MinCount=1,
|
||||
MaxCount=1,
|
||||
InstanceType='t2.micro',
|
||||
TagSpecifications=[
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG1',
|
||||
'Value': 'MY_VALUE1',
|
||||
},
|
||||
{
|
||||
'Key': 'MY_TAG2',
|
||||
'Value': 'MY_VALUE2',
|
||||
},
|
||||
],
|
||||
},
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG3',
|
||||
'Value': 'MY_VALUE3',
|
||||
},
|
||||
]
|
||||
},
|
||||
],
|
||||
)
|
||||
|
||||
rtapi = boto3.client('resourcegroupstaggingapi', region_name='eu-central-1')
|
||||
resp = rtapi.get_tag_keys()
|
||||
|
||||
resp['TagKeys'].should.contain('MY_TAG1')
|
||||
resp['TagKeys'].should.contain('MY_TAG2')
|
||||
resp['TagKeys'].should.contain('MY_TAG3')
|
||||
|
||||
# TODO test pagenation
|
||||
|
||||
|
||||
@mock_ec2
|
||||
@mock_resourcegroupstaggingapi
|
||||
def test_get_tag_values_ec2():
|
||||
client = boto3.client('ec2', region_name='eu-central-1')
|
||||
|
||||
client.run_instances(
|
||||
ImageId='ami-123',
|
||||
MinCount=1,
|
||||
MaxCount=1,
|
||||
InstanceType='t2.micro',
|
||||
TagSpecifications=[
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG1',
|
||||
'Value': 'MY_VALUE1',
|
||||
},
|
||||
{
|
||||
'Key': 'MY_TAG2',
|
||||
'Value': 'MY_VALUE2',
|
||||
},
|
||||
],
|
||||
},
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG3',
|
||||
'Value': 'MY_VALUE3',
|
||||
},
|
||||
]
|
||||
},
|
||||
],
|
||||
)
|
||||
client.run_instances(
|
||||
ImageId='ami-123',
|
||||
MinCount=1,
|
||||
MaxCount=1,
|
||||
InstanceType='t2.micro',
|
||||
TagSpecifications=[
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG1',
|
||||
'Value': 'MY_VALUE4',
|
||||
},
|
||||
{
|
||||
'Key': 'MY_TAG2',
|
||||
'Value': 'MY_VALUE5',
|
||||
},
|
||||
],
|
||||
},
|
||||
{
|
||||
'ResourceType': 'instance',
|
||||
'Tags': [
|
||||
{
|
||||
'Key': 'MY_TAG3',
|
||||
'Value': 'MY_VALUE6',
|
||||
},
|
||||
]
|
||||
},
|
||||
],
|
||||
)
|
||||
|
||||
rtapi = boto3.client('resourcegroupstaggingapi', region_name='eu-central-1')
|
||||
resp = rtapi.get_tag_values(Key='MY_TAG1')
|
||||
|
||||
resp['TagValues'].should.contain('MY_VALUE1')
|
||||
resp['TagValues'].should.contain('MY_VALUE4')
|
||||
|
||||
# TODO test pagenation
|
24
tests/test_resourcegroupstaggingapi/test_server.py
Normal file
24
tests/test_resourcegroupstaggingapi/test_server.py
Normal file
@ -0,0 +1,24 @@
|
||||
from __future__ import unicode_literals
|
||||
|
||||
import sure # noqa
|
||||
|
||||
import moto.server as server
|
||||
|
||||
'''
|
||||
Test the different server responses
|
||||
'''
|
||||
|
||||
|
||||
def test_resourcegroupstaggingapi_list():
|
||||
backend = server.create_backend_app("resourcegroupstaggingapi")
|
||||
test_client = backend.test_client()
|
||||
# do test
|
||||
|
||||
headers = {
|
||||
'X-Amz-Target': 'ResourceGroupsTaggingAPI_20170126.GetResources',
|
||||
'X-Amz-Date': '20171114T234623Z'
|
||||
}
|
||||
resp = test_client.post('/', headers=headers, data='{}')
|
||||
|
||||
assert resp.status_code == 200
|
||||
assert b'ResourceTagMappingList' in resp.data
|
Loading…
Reference in New Issue
Block a user