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

Merge pull request #2099 from ssoulier/master

Browse Source 
fix bug in Stream init (wrong number of shards)
This commit is contained in:
Steve Pulec 2019-07-01 22:27:33 -05:00 committed by GitHub
commit 75d9707e8e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
moto/kinesis/models.py
View File

@ -123,17 +123,12 @@ class Stream(BaseModel):
self.tags = {} self.tags = {}
self.status = "ACTIVE" self.status = "ACTIVE"
if six.PY3: step = 2**128 // shard_count
izip_longest = itertools.zip_longest hash_ranges = itertools.chain(map(lambda i: (i, i * step, (i + 1) * step),
else: range(shard_count - 1)),
izip_longest = itertools.izip_longest [(shard_count - 1, (shard_count - 1) * step, 2**128)])
for index, start, end in hash_ranges:
for index, start, end in izip_longest(range(shard_count),
range(0, 2**128, 2 **
128 // shard_count),
range(2**128 // shard_count, 2 **
128, 2**128 // shard_count),
fillvalue=2**128):
shard = Shard(index, start, end) shard = Shard(index, start, end)
self.shards[shard.shard_id] = shard self.shards[shard.shard_id] = shard

4
tests/test_kinesis/test_kinesis.py
View File

@ -15,7 +15,7 @@ from moto import mock_kinesis, mock_kinesis_deprecated
def test_create_cluster(): def test_create_cluster():
conn = boto.kinesis.connect_to_region("us-west-2") conn = boto.kinesis.connect_to_region("us-west-2")
conn.create_stream("my_stream", 2) conn.create_stream("my_stream", 3)
stream_response = conn.describe_stream("my_stream") stream_response = conn.describe_stream("my_stream")
@ -27,7 +27,7 @@ def test_create_cluster():
stream["StreamStatus"].should.equal("ACTIVE") stream["StreamStatus"].should.equal("ACTIVE")
shards = stream['Shards'] shards = stream['Shards']
shards.should.have.length_of(2) shards.should.have.length_of(3)
@mock_kinesis_deprecated @mock_kinesis_deprecated