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

Fix response data-types for kinesis operations (#4736)

Browse Source
This commit is contained in:
Tim Leslie 2022-01-04 21:47:59 +11:00 committed by GitHub
parent 307104417b
commit ee6b2bfff8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
moto/kinesis/models.py
View File

@ -86,7 +86,7 @@ class Shard(BaseModel):
self.records[sequence_number] = Record(
partition_key, data, sequence_number, explicit_hash_key
)
return sequence_number
return str(sequence_number)
def get_min_sequence_number(self):
if self.records:
@ -121,16 +121,16 @@ class Shard(BaseModel):
"StartingHashKey": str(self.starting_hash),
},
"SequenceNumberRange": {
"StartingSequenceNumber": self.get_min_sequence_number(),
"StartingSequenceNumber": str(self.get_min_sequence_number()),
},
"ShardId": self.shard_id,
}
if self.parent:
response["ParentShardId"] = self.parent
if not self.is_open:
response["SequenceNumberRange"][
"EndingSequenceNumber"
] = self.get_max_sequence_number()
response["SequenceNumberRange"]["EndingSequenceNumber"] = str(
self.get_max_sequence_number()
)
return response

8
tests/test_kinesis/test_kinesis.py
View File

@ -226,7 +226,8 @@ def test_put_records():
InvalidArgumentException
)
conn.put_record(stream_name, data, partition_key)
response = conn.put_record(stream_name, data, partition_key)
response["SequenceNumber"].should.equal("1")
response = conn.describe_stream(stream_name)
shard_id = response["StreamDescription"]["Shards"][0]["ShardId"]
@ -256,7 +257,10 @@ def test_put_records_boto3():
data = b"hello world"
partition_key = "1234"
client.put_record(StreamName=stream_name, Data=data, PartitionKey=partition_key)
response = client.put_record(
StreamName=stream_name, Data=data, PartitionKey=partition_key
)
response["SequenceNumber"].should.equal("1")
resp = client.get_shard_iterator(
StreamName=stream_name, ShardId=shard_id, ShardIteratorType="TRIM_HORIZON"