ansible-collections/community.general
1
0
Fork 0
mirror of https://github.com/ansible-collections/community.general.git synced 2025-10-23 04:24:00 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

[cloud] Add the ability to modify shard count to kinesis_stream module (#24805)

Browse source 
* Add the ability to modify shard count to kinesis_stream module

* Fixed an issue in kinesis_stream where update() reports not changed when it is changed

* Remove unreachable message and make the try and catch block shorter
This commit is contained in:
jacky.chen 2017-08-30 07:13:46 +10:00 • committed by Ryan Brown
commit 75998d3ca3
2 changed files with 109 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

23
test/units/modules/cloud/amazon/test_kinesis_stream.py
View file

@ -98,6 +98,9 @@ class AnsibleKinesisStreamFunctions(unittest.TestCase):
kinesis_stream.find_stream(client, 'test', check_mode=True)
)
should_return = {
'OpenShardsCount': 5,
'ClosedShardsCount': 0,
'ShardsCount': 5,
'HasMoreShards': True,
'RetentionPeriodHours': 24,
'StreamName': 'test',
@ -115,6 +118,9 @@ class AnsibleKinesisStreamFunctions(unittest.TestCase):
)
)
should_return = {
'OpenShardsCount': 5,
'ClosedShardsCount': 0,
'ShardsCount': 5,
'HasMoreShards': True,
'RetentionPeriodHours': 24,
'StreamName': 'test',
@ -230,9 +236,21 @@ class AnsibleKinesisStreamFunctions(unittest.TestCase):
)
self.assertFalse(success)
def test_update_shard_count(self):
client = boto3.client('kinesis', region_name=aws_region)
success, err_msg = (
kinesis_stream.update_shard_count(
client, 'test', 5, check_mode=True
)
)
self.assertTrue(success)
def test_update(self):
client = boto3.client('kinesis', region_name=aws_region)
current_stream = {
'OpenShardsCount': 5,
'ClosedShardsCount': 0,
'ShardsCount': 1,
'HasMoreShards': True,
'RetentionPeriodHours': 24,
'StreamName': 'test',
@ -245,7 +263,7 @@ class AnsibleKinesisStreamFunctions(unittest.TestCase):
}
success, changed, err_msg = (
kinesis_stream.update(
client, current_stream, 'test', retention_period=48,
client, current_stream, 'test', number_of_shards=2, retention_period=48,
tags=tags, check_mode=True
)
)
@ -266,6 +284,9 @@ class AnsibleKinesisStreamFunctions(unittest.TestCase):
)
)
should_return = {
'open_shards_count': 5,
'closed_shards_count': 0,
'shards_count': 5,
'has_more_shards': True,
'retention_period_hours': 24,
'stream_name': 'test',