From 45de4a46eceb411bc3fff082fed47441e62d51f0 Mon Sep 17 00:00:00 2001 From: Steve Pulec Date: Wed, 26 Nov 2014 21:55:01 -0500 Subject: [PATCH] Better sequence start and end numbers. --- moto/kinesis/models.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/moto/kinesis/models.py b/moto/kinesis/models.py index 6a1b081f2..52a5e91f7 100644 --- a/moto/kinesis/models.py +++ b/moto/kinesis/models.py @@ -55,8 +55,15 @@ class Shard(object): self.records[sequence_number] = Record(partition_key, data, sequence_number) return sequence_number + def get_min_sequence_number(self): + if self.records: + return list(self.records.keys())[0] + return 0 + def get_max_sequence_number(self): - return list(self.records.keys())[-1] + if self.records: + return list(self.records.keys())[-1] + return 0 def to_json(self): return { @@ -65,8 +72,8 @@ class Shard(object): "StartingHashKey": "0" }, "SequenceNumberRange": { - "EndingSequenceNumber": "21269319989741826081360214168359141376", - "StartingSequenceNumber": "21267647932558653966460912964485513216" + "EndingSequenceNumber": self.get_max_sequence_number(), + "StartingSequenceNumber": self.get_min_sequence_number(), }, "ShardId": self.shard_id }