diff --git a/src/sentry/preprod/eap/write.py b/src/sentry/preprod/eap/write.py index 919e72562d70c7..52130184b6eab1 100644 --- a/src/sentry/preprod/eap/write.py +++ b/src/sentry/preprod/eap/write.py @@ -23,6 +23,8 @@ from sentry.utils.arroyo_producer import SingletonProducer, get_arroyo_producer from sentry.utils.kafka_config import get_topic_definition +THIRTEEN_MONTHS = 396 # 13 months in days + def produce_preprod_size_metric_to_eap( size_metric: PreprodArtifactSizeMetrics, @@ -109,7 +111,7 @@ def produce_preprod_size_metric_to_eap( timestamp=proto_timestamp, trace_id=trace_id, received=received, - retention_days=90, # Default retention for preprod data + retention_days=THIRTEEN_MONTHS, # Default retention for preprod data attributes={k: anyvalue(v) for k, v in attributes.items() if v is not None}, client_sample_rate=1.0, server_sample_rate=1.0, @@ -217,7 +219,7 @@ def produce_preprod_build_distribution_to_eap( timestamp=proto_timestamp, trace_id=trace_id, received=received, - retention_days=90, + retention_days=THIRTEEN_MONTHS, attributes={k: anyvalue(v) for k, v in attributes.items() if v is not None}, client_sample_rate=1.0, server_sample_rate=1.0, diff --git a/tests/sentry/preprod/eap/test_write.py b/tests/sentry/preprod/eap/test_write.py index d77c3ddd495167..bbc7df160f721c 100644 --- a/tests/sentry/preprod/eap/test_write.py +++ b/tests/sentry/preprod/eap/test_write.py @@ -7,6 +7,7 @@ from sentry.conf.types.kafka_definition import Topic, get_topic_codec from sentry.models.commitcomparison import CommitComparison from sentry.preprod.eap.write import ( + THIRTEEN_MONTHS, produce_preprod_build_distribution_to_eap, produce_preprod_size_metric_to_eap, ) @@ -82,7 +83,7 @@ def test_write_preprod_size_metric_encodes_all_fields_correctly(self, mock_produ assert trace_item.organization_id == self.organization.id assert trace_item.project_id == self.project.id assert trace_item.item_type == TraceItemType.TRACE_ITEM_TYPE_PREPROD - assert trace_item.retention_days == 90 + assert trace_item.retention_days == THIRTEEN_MONTHS attrs = trace_item.attributes @@ -231,7 +232,7 @@ def test_write_preprod_build_distribution_encodes_all_fields_correctly(self, moc assert trace_item.organization_id == self.organization.id assert trace_item.project_id == self.project.id assert trace_item.item_type == TraceItemType.TRACE_ITEM_TYPE_PREPROD - assert trace_item.retention_days == 90 + assert trace_item.retention_days == THIRTEEN_MONTHS attrs = trace_item.attributes