da07adae52
* Support for CloudFormation stack resource deletion via backend resource method delete_from_cloudformation_json() via parse_and_delete_resource(). * Correction to the inappropriate inclusion of EndingSequenceNumber in open shards. This attribute should only appear in closed shards. This regretfully prevents confirmation of consistent record counts after split/merge in unit tests. * Added parameters/decorator to CloudFormationModel method declarations to calm-down Pycharm. Co-authored-by: Joseph Weitekamp <jweite@amazon.com>
145 lines
4.1 KiB
Python
145 lines
4.1 KiB
Python
import boto3
|
|
import sure # noqa
|
|
|
|
from moto import mock_kinesis, mock_cloudformation
|
|
|
|
|
|
@mock_cloudformation
|
|
def test_kinesis_cloudformation_create_stream():
|
|
cf_conn = boto3.client("cloudformation", region_name="us-east-1")
|
|
|
|
stack_name = "MyStack"
|
|
|
|
template = '{"Resources":{"MyStream":{"Type":"AWS::Kinesis::Stream"}}}'
|
|
|
|
cf_conn.create_stack(StackName=stack_name, TemplateBody=template)
|
|
|
|
provisioned_resource = cf_conn.list_stack_resources(StackName=stack_name)[
|
|
"StackResourceSummaries"
|
|
][0]
|
|
provisioned_resource["LogicalResourceId"].should.equal("MyStream")
|
|
len(provisioned_resource["PhysicalResourceId"]).should.be.greater_than(0)
|
|
|
|
|
|
@mock_cloudformation
|
|
@mock_kinesis
|
|
def test_kinesis_cloudformation_get_attr():
|
|
cf_conn = boto3.client("cloudformation", region_name="us-east-1")
|
|
|
|
stack_name = "MyStack"
|
|
|
|
template = """
|
|
Resources:
|
|
TheStream:
|
|
Type: AWS::Kinesis::Stream
|
|
Outputs:
|
|
StreamName:
|
|
Value: !Ref TheStream
|
|
StreamArn:
|
|
Value: !GetAtt TheStream.Arn
|
|
""".strip()
|
|
|
|
cf_conn.create_stack(StackName=stack_name, TemplateBody=template)
|
|
stack_description = cf_conn.describe_stacks(StackName=stack_name)["Stacks"][0]
|
|
output_stream_name = [
|
|
output["OutputValue"]
|
|
for output in stack_description["Outputs"]
|
|
if output["OutputKey"] == "StreamName"
|
|
][0]
|
|
output_stream_arn = [
|
|
output["OutputValue"]
|
|
for output in stack_description["Outputs"]
|
|
if output["OutputKey"] == "StreamArn"
|
|
][0]
|
|
|
|
kinesis_conn = boto3.client("kinesis", region_name="us-east-1")
|
|
stream_description = kinesis_conn.describe_stream(StreamName=output_stream_name)[
|
|
"StreamDescription"
|
|
]
|
|
output_stream_arn.should.equal(stream_description["StreamARN"])
|
|
|
|
|
|
@mock_cloudformation
|
|
@mock_kinesis
|
|
def test_kinesis_cloudformation_update():
|
|
cf_conn = boto3.client("cloudformation", region_name="us-east-1")
|
|
|
|
stack_name = "MyStack"
|
|
|
|
template = """
|
|
Resources:
|
|
TheStream:
|
|
Type: AWS::Kinesis::Stream
|
|
Properties:
|
|
Name: MyStream
|
|
ShardCount: 4
|
|
""".strip()
|
|
|
|
cf_conn.create_stack(StackName=stack_name, TemplateBody=template)
|
|
stack_description = cf_conn.describe_stacks(StackName=stack_name)["Stacks"][0]
|
|
stack_description["StackName"].should.equal(stack_name)
|
|
|
|
kinesis_conn = boto3.client("kinesis", region_name="us-east-1")
|
|
stream_description = kinesis_conn.describe_stream(StreamName="MyStream")[
|
|
"StreamDescription"
|
|
]
|
|
shards_provisioned = len(
|
|
[
|
|
shard
|
|
for shard in stream_description["Shards"]
|
|
if "EndingSequenceNumber" not in shard["SequenceNumberRange"]
|
|
]
|
|
)
|
|
shards_provisioned.should.equal(4)
|
|
|
|
template = """
|
|
Resources:
|
|
TheStream:
|
|
Type: AWS::Kinesis::Stream
|
|
Properties:
|
|
ShardCount: 6
|
|
""".strip()
|
|
cf_conn.update_stack(StackName=stack_name, TemplateBody=template)
|
|
|
|
stream_description = kinesis_conn.describe_stream(StreamName="MyStream")[
|
|
"StreamDescription"
|
|
]
|
|
shards_provisioned = len(
|
|
[
|
|
shard
|
|
for shard in stream_description["Shards"]
|
|
if "EndingSequenceNumber" not in shard["SequenceNumberRange"]
|
|
]
|
|
)
|
|
shards_provisioned.should.equal(6)
|
|
|
|
|
|
@mock_cloudformation
|
|
@mock_kinesis
|
|
def test_kinesis_cloudformation_delete():
|
|
cf_conn = boto3.client("cloudformation", region_name="us-east-1")
|
|
|
|
stack_name = "MyStack"
|
|
|
|
template = """
|
|
Resources:
|
|
TheStream:
|
|
Type: AWS::Kinesis::Stream
|
|
Properties:
|
|
Name: MyStream
|
|
""".strip()
|
|
|
|
cf_conn.create_stack(StackName=stack_name, TemplateBody=template)
|
|
stack_description = cf_conn.describe_stacks(StackName=stack_name)["Stacks"][0]
|
|
stack_description["StackName"].should.equal(stack_name)
|
|
|
|
kinesis_conn = boto3.client("kinesis", region_name="us-east-1")
|
|
stream_description = kinesis_conn.describe_stream(StreamName="MyStream")[
|
|
"StreamDescription"
|
|
]
|
|
stream_description["StreamName"].should.equal("MyStream")
|
|
|
|
cf_conn.delete_stack(StackName=stack_name)
|
|
streams = kinesis_conn.list_streams()["StreamNames"]
|
|
len(streams).should.equal(0)
|