diff --git a/tests/contrib/botocore/test.py b/tests/contrib/botocore/test.py index 130a1187075..f54c60c5ee3 100644 --- a/tests/contrib/botocore/test.py +++ b/tests/contrib/botocore/test.py @@ -1652,56 +1652,6 @@ def test_kinesis_put_records_base64_trace_injection(self): client.delete_stream(StreamName=stream_name) - @mock_kinesis - def test_kinesis_put_records_base64_exceeds_max_size(self): - client = self.session.create_client("kinesis", region_name="us-east-1") - - stream_name = "test" - client.create_stream(StreamName=stream_name, ShardCount=1) - stream = client.describe_stream(StreamName=stream_name)["StreamDescription"] - shard_id = stream["Shards"][0]["ShardId"] - - partition_key = "1234" - sample_string = json.dumps({"Hello": "x" * (1 << 20)}) - sample_string_bytes = sample_string.encode("ascii") - base64_bytes = base64.b64encode(sample_string_bytes) - data_str = base64_bytes.decode("ascii") - data = [ - {"Data": data_str, "PartitionKey": partition_key}, - {"Data": data_str, "PartitionKey": partition_key}, - ] - - Pin(service=self.TEST_SERVICE, tracer=self.tracer).onto(client) - client.put_records(StreamName=stream_name, Records=data) - - # check if the appropriate span was generated - spans = self.get_spans() - assert spans - span = spans[0] - assert len(spans) == 1 - assert span.get_tag("aws.region") == "us-east-1" - assert span.get_tag("aws.operation") == "PutRecords" - assert span.get_tag("params.MessageBody") is None - assert_is_measured(span) - assert_span_http_status_code(span, 200) - assert span.service == "test-botocore-tracing.kinesis" - assert span.resource == "kinesis.putrecords" - records = span.get_tag("params.Records") - assert records is None - - resp = client.get_shard_iterator(StreamName=stream_name, ShardId=shard_id, ShardIteratorType="TRIM_HORIZON") - shard_iterator = resp["ShardIterator"] - - # ensure headers are not present in received message - resp = client.get_records(ShardIterator=shard_iterator) - assert len(resp["Records"]) == 2 - records = resp["Records"] - for record in records: - headers = json.loads(base64.b64decode(record["Data"]).decode("ascii")) - assert "_datadog" not in headers - - client.delete_stream(StreamName=stream_name) - @unittest.skipIf(PY2, "Skipping for Python 2.7 since older moto doesn't support secretsmanager") def test_secretsmanager(self): from moto import mock_secretsmanager