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

EventBridge: put_rule and list_rules should store and retrieve EventBusName property (#3472)

Browse Source 
Co-authored-by: Guillermo Arribas <garribas@atlassian.com>
This commit is contained in:
Guillermo Arribas 2020-11-18 02:36:17 +11:00 committed by GitHub
parent 5fe921c2bc
commit 62fd975da0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
moto/events/models.py
View File

@ -25,6 +25,7 @@ class Rule(CloudFormationModel):
self.state = kwargs.get("State") or "ENABLED" self.state = kwargs.get("State") or "ENABLED"
self.description = kwargs.get("Description") self.description = kwargs.get("Description")
self.role_arn = kwargs.get("RoleArn") self.role_arn = kwargs.get("RoleArn")
self.event_bus_name = kwargs.get("EventBusName", "default")
self.targets = [] self.targets = []
@property @property

3
moto/events/responses.py
View File

@ -25,6 +25,7 @@ class EventsHandler(BaseResponse):
"Description": rule.description, "Description": rule.description,
"ScheduleExpression": rule.schedule_exp, "ScheduleExpression": rule.schedule_exp,
"RoleArn": rule.role_arn, "RoleArn": rule.role_arn,
"EventBusName": rule.event_bus_name,
} }
@property @property
@ -167,6 +168,7 @@ class EventsHandler(BaseResponse):
state = self._get_param("State") state = self._get_param("State")
desc = self._get_param("Description") desc = self._get_param("Description")
role_arn = self._get_param("RoleArn") role_arn = self._get_param("RoleArn")
event_bus_name = self._get_param("EventBusName")
if not name: if not name:
return self.error("ValidationException", "Parameter Name is required.") return self.error("ValidationException", "Parameter Name is required.")
@ -199,6 +201,7 @@ class EventsHandler(BaseResponse):
State=state, State=state,
Description=desc, Description=desc,
RoleArn=role_arn, RoleArn=role_arn,
EventBusName=event_bus_name,
) )
return json.dumps({"RuleArn": rule.arn}), self.response_headers return json.dumps({"RuleArn": rule.arn}), self.response_headers

2
tests/test_events/test_events.py
View File

@ -86,6 +86,7 @@ def test_put_rule():
"Name": "my-event", "Name": "my-event",
"ScheduleExpression": "rate(5 minutes)", "ScheduleExpression": "rate(5 minutes)",
"EventPattern": '{"source": ["test-source"]}', "EventPattern": '{"source": ["test-source"]}',
"EventBusName": "test-bus",
} }
client.put_rule(**rule_data) client.put_rule(**rule_data)
@ -96,6 +97,7 @@ def test_put_rule():
rules[0]["Name"].should.equal(rule_data["Name"]) rules[0]["Name"].should.equal(rule_data["Name"])
rules[0]["ScheduleExpression"].should.equal(rule_data["ScheduleExpression"]) rules[0]["ScheduleExpression"].should.equal(rule_data["ScheduleExpression"])
rules[0]["EventPattern"].should.equal(rule_data["EventPattern"]) rules[0]["EventPattern"].should.equal(rule_data["EventPattern"])
rules[0]["EventBusName"].should.equal(rule_data["EventBusName"])
rules[0]["State"].should.equal("ENABLED") rules[0]["State"].should.equal("ENABLED")