From 7eb6a30b0541e2b2756f50d2204ab64b9c76e4e4 Mon Sep 17 00:00:00 2001 From: David Herberth Date: Fri, 28 Feb 2025 11:23:17 +0100 Subject: [PATCH] change test decompression --- tests/integration/fixtures/mini_sentry.py | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) diff --git a/tests/integration/fixtures/mini_sentry.py b/tests/integration/fixtures/mini_sentry.py index 9783e13cc4e..862da941839 100644 --- a/tests/integration/fixtures/mini_sentry.py +++ b/tests/integration/fixtures/mini_sentry.py @@ -1,6 +1,6 @@ import datetime import copy -import gzip +import zstandard import json import os import re @@ -318,9 +318,13 @@ def store_internal_error_event(): @app.route("/api/42/envelope/", methods=["POST"]) def store_event(): assert ( - flask_request.headers.get("Content-Encoding", "") == "gzip" + flask_request.headers.get("Content-Encoding", "") == "zstd" ), "Relay should always compress store requests" - data = gzip.decompress(flask_request.data) + + with zstandard.ZstdDecompressor().stream_reader( + flask_request.data + ) as decompressor: + data = decompressor.read() assert ( flask_request.headers.get("Content-Type") == "application/x-sentry-envelope" @@ -454,8 +458,11 @@ def global_metrics(): abort(403, "relay not registered") encoding = flask_request.headers.get("Content-Encoding", "") - assert encoding == "gzip", "Relay should always compress store requests" - data = gzip.decompress(flask_request.data) + assert encoding == "zstd", "Relay should always compress store requests" + with zstandard.ZstdDecompressor().stream_reader( + flask_request.data + ) as decompressor: + data = decompressor.read() metrics_batch = json.loads(data)["buckets"] sentry.captured_metrics.put(metrics_batch)