Skip to content

Commit

Permalink
change test decompression
Browse files Browse the repository at this point in the history
  • Loading branch information
Dav1dde committed Feb 28, 2025
1 parent 393224c commit 7eb6a30
Showing 1 changed file with 12 additions and 5 deletions.
17 changes: 12 additions & 5 deletions tests/integration/fixtures/mini_sentry.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import datetime
import copy
import gzip
import zstandard
import json
import os
import re
Expand Down Expand Up @@ -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"
Expand Down Expand Up @@ -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)
Expand Down

0 comments on commit 7eb6a30

Please sign in to comment.