Completed the CloudWatch Events mocking module and tests.
This commit is contained in:
Justin Wiley
parent
d0def03c4c
commit
db0b494b4f
5 changed files with 367 additions and 47 deletions
|
|
@ -1,4 +1,3 @@
|
|||
import binascii
|
||||
import os
|
||||
import re
|
||||
from collections import OrderedDict
|
||||
|
|
@ -13,12 +12,12 @@ class Rule(object):
|
|||
|
||||
def __init__(self, name, **kwargs):
|
||||
self.name = name
|
||||
self.arn = kwargs['arn'] if 'arn' in kwargs else self._generate_arn(name)
|
||||
self.event_pattern = kwargs['event_pattern'] if 'event_pattern' in kwargs else None
|
||||
self.schedule_exp = kwargs['schedule_exp'] if 'schedule_exp' in kwargs else None
|
||||
self.state = kwargs['state'] if 'state' in kwargs else 'ENABLED'
|
||||
self.description = kwargs['description'] if 'description' in kwargs else None
|
||||
self.role_arn = kwargs['role_arn'] if 'role_arn' in kwargs else None
|
||||
self.arn = kwargs.get('Arn') or self._generate_arn(name)
|
||||
self.event_pattern = kwargs.get('EventPattern')
|
||||
self.schedule_exp = kwargs.get('ScheduleExpression')
|
||||
self.state = kwargs.get('State') or 'ENABLED'
|
||||
self.description = kwargs.get('Description')
|
||||
self.role_arn = kwargs.get('RoleArn')
|
||||
self.targets = {}
|
||||
|
||||
def enable(self):
|
||||
|
|
@ -28,9 +27,9 @@ class Rule(object):
|
|||
self.state = 'DISABLED'
|
||||
|
||||
def put_targets(self, targets):
|
||||
# TODO: Will need to test for valid ARNs.
|
||||
# Not testing for valid ARNs.
|
||||
for target in targets:
|
||||
self.targets[target['TargetId']] = target
|
||||
self.targets[target['Id']] = target
|
||||
|
||||
def remove_targets(self, ids):
|
||||
for target in ids:
|
||||
|
|
@ -45,7 +44,7 @@ class EventsBackend(BaseBackend):
|
|||
self.next_tokens = {}
|
||||
|
||||
def _gen_next_token(self, index):
|
||||
token = binascii.hexlify(os.urandom(16))
|
||||
token = os.urandom(128).encode('base64')
|
||||
self.next_tokens[token] = index
|
||||
return token
|
||||
|
||||
|
|
@ -54,15 +53,14 @@ class EventsBackend(BaseBackend):
|
|||
end_index = array_len
|
||||
new_next_token = None
|
||||
|
||||
if next_token is not None:
|
||||
if next_token in self.next_tokens:
|
||||
start_index = self.next_tokens[next_token]
|
||||
if next_token:
|
||||
start_index = self.next_tokens.pop(next_token, 0)
|
||||
|
||||
if limit is not None:
|
||||
new_end_index = start_index + int(limit)
|
||||
if new_end_index < end_index:
|
||||
end_index = new_end_index
|
||||
new_next_token = self._gen_next_token(end_index - 1)
|
||||
new_next_token = self._gen_next_token(end_index)
|
||||
|
||||
return start_index, end_index, new_next_token
|
||||
|
||||
|
|
@ -70,10 +68,7 @@ class EventsBackend(BaseBackend):
|
|||
return self.rules.pop(name) is not None
|
||||
|
||||
def describe_rule(self, name):
|
||||
if name in self.rules:
|
||||
return self.rules[name]
|
||||
|
||||
return None
|
||||
return self.rules.get(name)
|
||||
|
||||
def disable_rule(self, name):
|
||||
if name in self.rules:
|
||||
|
|
@ -102,8 +97,9 @@ class EventsBackend(BaseBackend):
|
|||
|
||||
for i in range(start_index, end_index):
|
||||
rule = rules_array[i]
|
||||
if target_arn in rule.targets:
|
||||
matching_rules.append(rule.name)
|
||||
for target in rule.targets:
|
||||
if rule.targets[target]['Arn'] == target_arn:
|
||||
matching_rules.append(rule.name)
|
||||
|
||||
return_obj['RuleNames'] = matching_rules
|
||||
if new_next_token is not None:
|
||||
|
|
@ -165,10 +161,22 @@ class EventsBackend(BaseBackend):
|
|||
return rule.arn
|
||||
|
||||
def put_targets(self, name, targets):
|
||||
self.rules[name].put_targets(targets)
|
||||
rule = self.rules.get(name)
|
||||
|
||||
if rule:
|
||||
rule.put_targets(targets)
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def remove_targets(self, name, ids):
|
||||
self.rules[name].remove_targets(ids)
|
||||
rule = self.rules.get(name)
|
||||
|
||||
if rule:
|
||||
rule.remove_targets(ids)
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
def test_event_pattern(self):
|
||||
raise NotImplementedError()
|
||||
|
|
|
|||
|
|
@ -1,59 +1,195 @@
|
|||
import json
|
||||
import re
|
||||
|
||||
from moto.core.responses import BaseResponse
|
||||
from moto.events import events_backend
|
||||
|
||||
|
||||
class EventsHandler(BaseResponse):
|
||||
|
||||
def error(self, type_, message='', status=400):
|
||||
return status, self.response_headers, json.dumps({'__type': type_, 'message': message})
|
||||
def _generate_rule_dict(self, rule):
|
||||
return {
|
||||
'Name': rule.name,
|
||||
'Arn': rule.arn,
|
||||
'EventPattern': rule.event_pattern,
|
||||
'State': rule.state,
|
||||
'Description': rule.description,
|
||||
'ScheduleExpression': rule.schedule_exp,
|
||||
'RoleArn': rule.role_arn
|
||||
}
|
||||
|
||||
def can_paginate(self):
|
||||
pass
|
||||
def load_body(self):
|
||||
decoded_body = self.body.decode('utf-8')
|
||||
return json.loads(decoded_body or '{}')
|
||||
|
||||
def error(self, type_, message='', status=400):
|
||||
headers = self.response_headers
|
||||
headers['status'] = status
|
||||
return json.dumps({'__type': type_, 'message': message}), headers,
|
||||
|
||||
def delete_rule(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
name = body.get('NamePrefix')
|
||||
|
||||
if not name:
|
||||
return self.error('ValidationException', 'Parameter Name is required.')
|
||||
|
||||
return '', self.response_headers
|
||||
|
||||
def describe_rule(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
name = body.get('Name')
|
||||
|
||||
if not name:
|
||||
return self.error('ValidationException', 'Parameter Name is required.')
|
||||
|
||||
rule = events_backend.describe_rule(name)
|
||||
|
||||
if not rule:
|
||||
return self.error('ResourceNotFoundException', 'Rule test does not exist.')
|
||||
|
||||
rule_dict = self._generate_rule_dict(rule)
|
||||
return json.dumps(rule_dict), self.response_headers
|
||||
|
||||
def disable_rule(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
name = body.get('Name')
|
||||
|
||||
if not name:
|
||||
return self.error('ValidationException', 'Parameter Name is required.')
|
||||
|
||||
if not events_backend.disable_rule(name):
|
||||
return self.error('ResourceNotFoundException', 'Rule ' + name + ' does not exist.')
|
||||
|
||||
return '', self.response_headers
|
||||
|
||||
def enable_rule(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
name = body.get('Name')
|
||||
|
||||
if not name:
|
||||
return self.error('ValidationException', 'Parameter Name is required.')
|
||||
|
||||
if not events_backend.enable_rule(name):
|
||||
return self.error('ResourceNotFoundException', 'Rule ' + name + ' does not exist.')
|
||||
|
||||
return '', self.response_headers
|
||||
|
||||
def generate_presigned_url(self):
|
||||
pass
|
||||
|
||||
def get_paginator(self):
|
||||
pass
|
||||
|
||||
def get_waiter(self):
|
||||
pass
|
||||
|
||||
def list_rule_names_by_target(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
target_arn = body.get('TargetArn')
|
||||
next_token = body.get('NextToken')
|
||||
limit = body.get('Limit')
|
||||
|
||||
if not target_arn:
|
||||
return self.error('ValidationException', 'Parameter TargetArn is required.')
|
||||
|
||||
rule_names = events_backend.list_rule_names_by_target(target_arn, next_token, limit)
|
||||
|
||||
return json.dumps(rule_names), self.response_headers
|
||||
|
||||
def list_rules(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
prefix = body.get('NamePrefix')
|
||||
next_token = body.get('NextToken')
|
||||
limit = body.get('Limit')
|
||||
|
||||
rules = events_backend.list_rules(prefix, next_token, limit)
|
||||
rules_obj = {'Rules': []}
|
||||
|
||||
for rule in rules['Rules']:
|
||||
rules_obj['Rules'].append(self._generate_rule_dict(rule))
|
||||
|
||||
if rules.get('NextToken'):
|
||||
rules_obj['NextToken'] = rules['NextToken']
|
||||
|
||||
return json.dumps(rules_obj), self.response_headers
|
||||
|
||||
def list_targets_by_rule(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
rule_name = body.get('Rule')
|
||||
next_token = body.get('NextToken')
|
||||
limit = body.get('Limit')
|
||||
|
||||
if not rule_name:
|
||||
return self.error('ValidationException', 'Parameter Rule is required.')
|
||||
|
||||
try:
|
||||
targets = events_backend.list_targets_by_rule(rule_name, next_token, limit)
|
||||
except KeyError:
|
||||
return self.error('ResourceNotFoundException', 'Rule ' + rule_name + ' does not exist.')
|
||||
|
||||
return json.dumps(targets), self.response_headers
|
||||
|
||||
def put_events(self):
|
||||
pass
|
||||
return '', self.response_headers
|
||||
|
||||
def put_rule(self):
|
||||
if 'Name' not in self.body:
|
||||
body = self.load_body()
|
||||
name = body.get('Name')
|
||||
event_pattern = body.get('EventPattern')
|
||||
sched_exp = body.get('ScheduleExpression')
|
||||
|
||||
if not name:
|
||||
return self.error('ValidationException', 'Parameter Name is required.')
|
||||
pass
|
||||
|
||||
if event_pattern:
|
||||
try:
|
||||
json.loads(event_pattern)
|
||||
except ValueError:
|
||||
# Not quite as informative as the real error, but it'll work for now.
|
||||
return self.error('InvalidEventPatternException', 'Event pattern is not valid.')
|
||||
|
||||
if sched_exp:
|
||||
if not (re.match('^cron\(.*\)', sched_exp) or
|
||||
re.match('^rate\(\d*\s(minute|minutes|hour|hours|day|days)\)', sched_exp)):
|
||||
return self.error('ValidationException', 'Parameter ScheduleExpression is not valid.')
|
||||
|
||||
rule_arn = events_backend.put_rule(
|
||||
name,
|
||||
ScheduleExpression=sched_exp,
|
||||
EventPattern=event_pattern,
|
||||
State=body.get('State'),
|
||||
Description=body.get('Description'),
|
||||
RoleArn=body.get('RoleArn')
|
||||
)
|
||||
|
||||
return json.dumps({'RuleArn': rule_arn}), self.response_headers
|
||||
|
||||
def put_targets(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
rule_name = body.get('Rule')
|
||||
targets = body.get('Targets')
|
||||
|
||||
if not rule_name:
|
||||
return self.error('ValidationException', 'Parameter Rule is required.')
|
||||
|
||||
if not targets:
|
||||
return self.error('ValidationException', 'Parameter Targets is required.')
|
||||
|
||||
if not events_backend.put_targets(rule_name, targets):
|
||||
return self.error('ResourceNotFoundException', 'Rule ' + rule_name + ' does not exist.')
|
||||
|
||||
return '', self.response_headers
|
||||
|
||||
def remove_targets(self):
|
||||
pass
|
||||
body = self.load_body()
|
||||
rule_name = body.get('Rule')
|
||||
ids = body.get('Ids')
|
||||
|
||||
if not rule_name:
|
||||
return self.error('ValidationException', 'Parameter Rule is required.')
|
||||
|
||||
if not ids:
|
||||
return self.error('ValidationException', 'Parameter Ids is required.')
|
||||
|
||||
if not events_backend.remove_targets(rule_name, ids):
|
||||
return self.error('ResourceNotFoundException', 'Rule ' + rule_name + ' does not exist.')
|
||||
|
||||
return '', self.response_headers
|
||||
|
||||
def test_event_pattern(self):
|
||||
pass
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@ from __future__ import unicode_literals
|
|||
from .responses import EventsHandler
|
||||
|
||||
url_bases = [
|
||||
"https?://events.(.+).amazonaws.com"
|
||||
"https://events.(.+).amazonaws.com"
|
||||
]
|
||||
|
||||
url_paths = {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue