Add events target integration for sqs queue (#3815)
This commit is contained in:
Anton Grübel
GitHub
parent
4f34439170
commit
9c89c24caf
4 changed files with 239 additions and 43 deletions
|
|
@ -301,14 +301,21 @@ def test_update_rule_with_targets():
|
|||
|
||||
|
||||
@mock_events
|
||||
def test_remove_targets_errors():
|
||||
client = boto3.client("events", "us-east-1")
|
||||
def test_remove_targets_error_unknown_rule():
|
||||
# given
|
||||
client = boto3.client("events", "eu-central-1")
|
||||
|
||||
client.remove_targets.when.called_with(
|
||||
Rule="non-existent", Ids=["Id12345678"]
|
||||
).should.throw(
|
||||
client.exceptions.ResourceNotFoundException,
|
||||
"An entity that you specified does not exist",
|
||||
# when
|
||||
with pytest.raises(ClientError) as e:
|
||||
client.remove_targets(Rule="unknown", Ids=["something"])
|
||||
|
||||
# then
|
||||
ex = e.value
|
||||
ex.operation_name.should.equal("RemoveTargets")
|
||||
ex.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400)
|
||||
ex.response["Error"]["Code"].should.contain("ResourceNotFoundException")
|
||||
ex.response["Error"]["Message"].should.equal(
|
||||
"Rule unknown does not exist on EventBus default."
|
||||
)
|
||||
|
||||
|
||||
|
|
@ -328,7 +335,7 @@ def test_put_targets():
|
|||
targets_before = len(targets)
|
||||
assert targets_before == 0
|
||||
|
||||
targets_data = [{"Arn": "test_arn", "Id": "test_id"}]
|
||||
targets_data = [{"Arn": "arn:aws:s3:::test-arn", "Id": "test_id"}]
|
||||
resp = client.put_targets(Rule=rule_name, Targets=targets_data)
|
||||
assert resp["FailedEntryCount"] == 0
|
||||
assert len(resp["FailedEntries"]) == 0
|
||||
|
|
@ -337,10 +344,63 @@ def test_put_targets():
|
|||
targets_after = len(targets)
|
||||
assert targets_before + 1 == targets_after
|
||||
|
||||
assert targets[0]["Arn"] == "test_arn"
|
||||
assert targets[0]["Arn"] == "arn:aws:s3:::test-arn"
|
||||
assert targets[0]["Id"] == "test_id"
|
||||
|
||||
|
||||
@mock_events
|
||||
def test_put_targets_error_invalid_arn():
|
||||
# given
|
||||
client = boto3.client("events", "eu-central-1")
|
||||
rule_name = "test-rule"
|
||||
client.put_rule(
|
||||
Name=rule_name,
|
||||
EventPattern=json.dumps({"account": [ACCOUNT_ID]}),
|
||||
State="ENABLED",
|
||||
)
|
||||
|
||||
# when
|
||||
with pytest.raises(ClientError) as e:
|
||||
client.put_targets(
|
||||
Rule=rule_name,
|
||||
Targets=[
|
||||
{"Id": "s3", "Arn": "arn:aws:s3:::test-bucket"},
|
||||
{"Id": "s3", "Arn": "test-bucket"},
|
||||
],
|
||||
)
|
||||
|
||||
# then
|
||||
ex = e.value
|
||||
ex.operation_name.should.equal("PutTargets")
|
||||
ex.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400)
|
||||
ex.response["Error"]["Code"].should.contain("ValidationException")
|
||||
ex.response["Error"]["Message"].should.equal(
|
||||
"Parameter test-bucket is not valid. "
|
||||
"Reason: Provided Arn is not in correct format."
|
||||
)
|
||||
|
||||
|
||||
@mock_events
|
||||
def test_put_targets_error_unknown_rule():
|
||||
# given
|
||||
client = boto3.client("events", "eu-central-1")
|
||||
|
||||
# when
|
||||
with pytest.raises(ClientError) as e:
|
||||
client.put_targets(
|
||||
Rule="unknown", Targets=[{"Id": "s3", "Arn": "arn:aws:s3:::test-bucket"}]
|
||||
)
|
||||
|
||||
# then
|
||||
ex = e.value
|
||||
ex.operation_name.should.equal("PutTargets")
|
||||
ex.response["ResponseMetadata"]["HTTPStatusCode"].should.equal(400)
|
||||
ex.response["Error"]["Code"].should.contain("ResourceNotFoundException")
|
||||
ex.response["Error"]["Message"].should.equal(
|
||||
"Rule unknown does not exist on EventBus default."
|
||||
)
|
||||
|
||||
|
||||
@mock_events
|
||||
def test_permissions():
|
||||
client = boto3.client("events", "eu-central-1")
|
||||
|
|
|
|||
120
tests/test_events/test_events_targets.py
Normal file
120
tests/test_events/test_events_targets.py
Normal file
|
|
@ -0,0 +1,120 @@
|
|||
import json
|
||||
from datetime import datetime
|
||||
|
||||
import boto3
|
||||
import sure # noqa
|
||||
|
||||
from moto import mock_events, mock_sqs, mock_logs
|
||||
from moto.core import ACCOUNT_ID
|
||||
from moto.core.utils import iso_8601_datetime_without_milliseconds
|
||||
|
||||
|
||||
@mock_events
|
||||
@mock_logs
|
||||
def test_send_to_cw_log_group():
|
||||
# given
|
||||
client_events = boto3.client("events", "eu-central-1")
|
||||
client_logs = boto3.client("logs", region_name="eu-central-1")
|
||||
log_group_name = "/test-group"
|
||||
rule_name = "test-rule"
|
||||
client_logs.create_log_group(logGroupName=log_group_name)
|
||||
client_events.put_rule(
|
||||
Name=rule_name,
|
||||
EventPattern=json.dumps({"account": [ACCOUNT_ID]}),
|
||||
State="ENABLED",
|
||||
)
|
||||
client_events.put_targets(
|
||||
Rule=rule_name,
|
||||
Targets=[
|
||||
{
|
||||
"Id": "logs",
|
||||
"Arn": "arn:aws:logs:eu-central-1:{0}:log-group:{1}".format(
|
||||
ACCOUNT_ID, log_group_name
|
||||
),
|
||||
}
|
||||
],
|
||||
)
|
||||
|
||||
# when
|
||||
event_time = datetime(2021, 1, 1, 12, 23, 34)
|
||||
client_events.put_events(
|
||||
Entries=[
|
||||
{
|
||||
"Time": event_time,
|
||||
"Source": "source",
|
||||
"DetailType": "type",
|
||||
"Detail": json.dumps({"key": "value"}),
|
||||
}
|
||||
],
|
||||
)
|
||||
|
||||
# then
|
||||
response = client_logs.filter_log_events(logGroupName=log_group_name)
|
||||
response["events"].should.have.length_of(1)
|
||||
event = response["events"][0]
|
||||
event["logStreamName"].should_not.be.empty
|
||||
event["timestamp"].should.be.a(float)
|
||||
event["ingestionTime"].should.be.a(int)
|
||||
event["eventId"].should_not.be.empty
|
||||
|
||||
message = json.loads(event["message"])
|
||||
message["version"].should.equal("0")
|
||||
message["id"].should_not.be.empty
|
||||
message["detail-type"].should.equal("type")
|
||||
message["source"].should.equal("source")
|
||||
message["time"].should.equal(iso_8601_datetime_without_milliseconds(event_time))
|
||||
message["region"].should.equal("eu-central-1")
|
||||
message["resources"].should.be.empty
|
||||
message["detail"].should.equal({"key": "value"})
|
||||
|
||||
|
||||
@mock_events
|
||||
@mock_sqs
|
||||
def test_send_to_sqs_queue():
|
||||
# given
|
||||
client_events = boto3.client("events", "eu-central-1")
|
||||
client_sqs = boto3.client("sqs", region_name="eu-central-1")
|
||||
rule_name = "test-rule"
|
||||
queue_url = client_sqs.create_queue(QueueName="test-queue")["QueueUrl"]
|
||||
queue_arn = client_sqs.get_queue_attributes(
|
||||
QueueUrl=queue_url, AttributeNames=["QueueArn"]
|
||||
)["Attributes"]["QueueArn"]
|
||||
client_events.put_rule(
|
||||
Name=rule_name,
|
||||
EventPattern=json.dumps({"account": [ACCOUNT_ID]}),
|
||||
State="ENABLED",
|
||||
)
|
||||
client_events.put_targets(
|
||||
Rule=rule_name, Targets=[{"Id": "sqs", "Arn": queue_arn}],
|
||||
)
|
||||
|
||||
# when
|
||||
event_time = datetime(2021, 1, 1, 12, 23, 34)
|
||||
client_events.put_events(
|
||||
Entries=[
|
||||
{
|
||||
"Time": event_time,
|
||||
"Source": "source",
|
||||
"DetailType": "type",
|
||||
"Detail": json.dumps({"key": "value"}),
|
||||
}
|
||||
],
|
||||
)
|
||||
|
||||
# then
|
||||
response = client_sqs.receive_message(QueueUrl=queue_url)
|
||||
response["Messages"].should.have.length_of(1)
|
||||
message = response["Messages"][0]
|
||||
message["MessageId"].should_not.be.empty
|
||||
message["ReceiptHandle"].should_not.be.empty
|
||||
message["MD5OfBody"].should_not.be.empty
|
||||
|
||||
body = json.loads(message["Body"])
|
||||
body["version"].should.equal("0")
|
||||
body["id"].should_not.be.empty
|
||||
body["detail-type"].should.equal("type")
|
||||
body["source"].should.equal("source")
|
||||
body["time"].should.equal(iso_8601_datetime_without_milliseconds(event_time))
|
||||
body["region"].should.equal("eu-central-1")
|
||||
body["resources"].should.be.empty
|
||||
body["detail"].should.equal({"key": "value"})
|
||||
Loading…
Add table
Add a link
Reference in a new issue