Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion src/sentry/eventstream/snuba.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
from sentry.options.rollout import in_rollout_group
from sentry.services.eventstore.models import GroupEvent
from sentry.utils import json, metrics, snuba
from sentry.utils.eap import EAP_ITEMS_INSERT_ENDPOINT
from sentry.utils.safe import get_path
from sentry.utils.sdk import set_current_event_project

Expand Down Expand Up @@ -498,7 +499,7 @@ def _send_item(self, trace_item: TraceItem) -> None:
try:
resp = snuba._snuba_pool.urlopen(
"POST",
"/tests/entities/eap_items/insert_bytes",
EAP_ITEMS_INSERT_ENDPOINT,
fields={"item_0": trace_item.SerializeToString()},
)

Expand Down
3 changes: 2 additions & 1 deletion src/sentry/replays/lib/eap/write.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
from sentry.conf.types.kafka_definition import Topic
from sentry.replays.lib.eap.snuba_transpiler import TRACE_ITEM_TYPE_MAP, TRACE_ITEM_TYPES
from sentry.replays.lib.kafka import EAP_ITEMS_CODEC, eap_producer
from sentry.utils.eap import EAP_ITEMS_INSERT_ENDPOINT
from sentry.utils.kafka_config import get_topic_definition

Value = bool | bytes | str | int | float | Sequence["Value"] | MutableMapping[str, "Value"]
Expand Down Expand Up @@ -86,7 +87,7 @@ def write_trace_items_test_suite(trace_items: list[EAPTraceItem]) -> None:
`docker logs -f snuba-snuba-1`
"""
response = requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files={
f"item_{i}": trace_item.SerializeToString() for i, trace_item in enumerate(trace_items)
},
Expand Down
15 changes: 8 additions & 7 deletions src/sentry/testutils/cases.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,6 +151,7 @@
from sentry.users.models.useremail import UserEmail
from sentry.utils import json
from sentry.utils.auth import SsoSession
from sentry.utils.eap import EAP_ITEMS_INSERT_ENDPOINT
from sentry.utils.json import dumps_htmlsafe
from sentry.utils.not_set import NOT_SET, NotSet, default_if_not_set
from sentry.utils.samples import load_data
Expand Down Expand Up @@ -1147,7 +1148,7 @@ def store_spans(self, spans, is_eap=False):
files[f"item_{i}"] = trace_item.SerializeToString()
assert (
requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files=files,
).status_code
== 200
Expand All @@ -1164,7 +1165,7 @@ def store_spans(self, spans, is_eap=False):
def store_ourlogs(self, ourlogs):
files = {f"log_{i}": log.SerializeToString() for i, log in enumerate(ourlogs)}
response = requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files=files,
)
assert response.status_code == 200
Expand All @@ -1175,7 +1176,7 @@ def store_trace_metrics(self, trace_metrics):
for i, trace_metric in enumerate(trace_metrics)
}
response = requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files=files,
)
assert response.status_code == 200
Expand All @@ -1186,7 +1187,7 @@ def store_profile_functions(self, profile_functions):
for i, profile_function in enumerate(profile_functions)
}
response = requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files=files,
)
assert response.status_code == 200
Expand Down Expand Up @@ -2266,7 +2267,7 @@ def store_spans(self, spans, is_eap=False):
files[f"item_{i}"] = trace_item.SerializeToString()
assert (
requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files=files,
).status_code
== 200
Expand Down Expand Up @@ -3897,7 +3898,7 @@ def store_replays_eap(self, replays):

files = {f"replay_{i}": replay.SerializeToString() for i, replay in enumerate(replays)}
response = requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files=files,
)
assert response.status_code == 200
Expand Down Expand Up @@ -4044,7 +4045,7 @@ def store_uptime_results(self, uptime_results):
f"uptime_{i}": result.SerializeToString() for i, result in enumerate(uptime_results)
}
response = requests.post(
settings.SENTRY_SNUBA + "/tests/entities/eap_items/insert_bytes",
settings.SENTRY_SNUBA + EAP_ITEMS_INSERT_ENDPOINT,
files=files,
)
assert response.status_code == 200
Expand Down
1 change: 1 addition & 0 deletions src/sentry/utils/eap.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
EAP_ITEMS_INSERT_ENDPOINT = "/tests/entities/eap_items/insert_bytes"
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This utils file will be further populated by #103182

Loading