Skip to content

Commit

Permalink
Update server path
Browse files Browse the repository at this point in the history
  • Loading branch information
bsmartradio committed Feb 4, 2025
1 parent 73ef458 commit 0975590
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions alertdb/storage.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,7 +143,7 @@ def __init__(
def get_alert(self, alert_id: str) -> bytes:
logger.info("retrieving alert id=%s", alert_id)
try:
alert_key = f"/alert_archive/v1/alerts/{alert_id}.avro.gz"
alert_key = f"/v1/alerts/{alert_id}.avro.gz"
# boto3 terminology for objects, objects live in prefixes inside
# of buckets
blob = self.object_store_client.get_object(
Expand All @@ -156,7 +156,7 @@ def get_alert(self, alert_id: str) -> bytes:
def get_schema(self, schema_id: str) -> bytes:
logger.info("retrieving schema id=%s", schema_id)
try:
schema_key = f"/alert_archive/v1/schemas/{schema_id}.json"
schema_key = f"/v1/schemas/{schema_id}.json"
blob = self.object_store_client.get_object(
Bucket=self.schema_bucket, Key=schema_key
)
Expand Down
4 changes: 2 additions & 2 deletions tests/test_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,7 +62,7 @@ def delete_blob(key_id, bucket_name):
"alert-id-3": b"payload-3",
}
for alert_id, alert_payload in alerts.items():
alert_key = f"/alert_archive/v1/alerts/{alert_id}.avro.gz"
alert_key = f"/v1/alerts/{alert_id}.avro.gz"
s3.put_object(Body=alert_payload, Bucket=packet_bucket_name, Key=alert_key)
logger.info("uploading alert %s", alert_id)
cls.addClassCleanup(delete_blob, alert_key, packet_bucket_name)
Expand Down Expand Up @@ -129,7 +129,7 @@ def test_healthcheck(self):
self.assertEqual(response["ResponseMetadata"]["HTTPStatusCode"], 200)

def _get_alert(self, alert_id: str) -> requests.Response:
alert_key = f"/alert_archive/v1/alerts/{alert_id}.avro.gz"
alert_key = f"/v1/alerts/{alert_id}.avro.gz"
alert_bucket = "alertdb-server-integration-test-bucket-packets"
try:
return self.s3.get_object(Bucket=alert_bucket, Key=alert_key)
Expand Down

0 comments on commit 0975590

Please sign in to comment.