Kinesis - DescribeStream(): Implement Filter-param (#4391)

moto/kinesis/models.py

@@ -209,15 +209,17 @@ class Stream(CloudFormationModel):
         sequence_number = shard.put_record(partition_key, data, explicit_hash_key)
         return sequence_number, shard.shard_id
 
-    def to_json(self):
+    def to_json(self, shard_limit=None):
+        all_shards = list(self.shards.values())
+        requested_shards = all_shards[0 : shard_limit or len(all_shards)]
         return {
             "StreamDescription": {
                 "StreamARN": self.arn,
                 "StreamName": self.stream_name,
                 "StreamStatus": self.status,
-                "HasMoreShards": False,
+                "HasMoreShards": len(requested_shards) != len(all_shards),
                 "RetentionPeriodHours": self.retention_period_hours,
-                "Shards": [shard.to_json() for shard in self.shards.values()],
+                "Shards": [shard.to_json() for shard in requested_shards],
             }
         }
 

moto/kinesis/responses.py

@@ -26,8 +26,9 @@ class KinesisResponse(BaseResponse):
 
     def describe_stream(self):
         stream_name = self.parameters.get("StreamName")
+        limit = self.parameters.get("Limit")
         stream = self.kinesis_backend.describe_stream(stream_name)
-        return json.dumps(stream.to_json())
+        return json.dumps(stream.to_json(shard_limit=limit))
 
     def describe_stream_summary(self):
         stream_name = self.parameters.get("StreamName")

tests/test_kinesis/test_kinesis_boto3.py

@@ -5,6 +5,36 @@ from moto import mock_kinesis
 import sure  # noqa
 
 
+@mock_kinesis
+def test_describe_stream_limit_parameter():
+    client = boto3.client("kinesis", region_name="us-west-2")
+    stream_name = "my_stream"
+
+    client.create_stream(StreamName=stream_name, ShardCount=5)
+
+    without_filter = client.describe_stream(StreamName=stream_name)["StreamDescription"]
+    without_filter["Shards"].should.have.length_of(5)
+    without_filter["HasMoreShards"].should.equal(False)
+
+    with_filter = client.describe_stream(StreamName=stream_name, Limit=2)[
+        "StreamDescription"
+    ]
+    with_filter["Shards"].should.have.length_of(2)
+    with_filter["HasMoreShards"].should.equal(True)
+
+    with_filter = client.describe_stream(StreamName=stream_name, Limit=5)[
+        "StreamDescription"
+    ]
+    with_filter["Shards"].should.have.length_of(5)
+    with_filter["HasMoreShards"].should.equal(False)
+
+    with_filter = client.describe_stream(StreamName=stream_name, Limit=6)[
+        "StreamDescription"
+    ]
+    with_filter["Shards"].should.have.length_of(5)
+    with_filter["HasMoreShards"].should.equal(False)
+
+
 @mock_kinesis
 def test_split_shard():
     conn = boto3.client("kinesis", region_name="us-west-2")