darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-04 10:21:14 -05:00
Code Issues Packages Projects Releases Wiki Activity

clean up

Browse Source
This commit is contained in:
Kenneth Kehl
2025-01-13 13:21:34 -08:00
parent 752e5cada9
commit eac21788a1
2 changed files with 4 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/celery/scheduled_tasks.py
View File

@@ -261,26 +261,18 @@ def process_delivery_receipts(self):
cloudwatch.init_app(current_app) cloudwatch.init_app(current_app)
start_time = aware_utcnow() - timedelta(minutes=3) start_time = aware_utcnow() - timedelta(minutes=3)
end_time = aware_utcnow() end_time = aware_utcnow()
print(f"START TIME {start_time} END TIME {end_time}")
delivered_receipts, failed_receipts = cloudwatch.check_delivery_receipts( delivered_receipts, failed_receipts = cloudwatch.check_delivery_receipts(
start_time, end_time start_time, end_time
) )
print(f"DELIVERED {delivered_receipts} FAILED {failed_receipts}")
delivered_receipts = list(delivered_receipts) delivered_receipts = list(delivered_receipts)
print(f"DELIVERED LIST {delivered_receipts}")
for i in range(0, len(delivered_receipts), batch_size): for i in range(0, len(delivered_receipts), batch_size):
batch = delivered_receipts[i : i + batch_size] batch = delivered_receipts[i : i + batch_size]
print("UPDATING DELIVERY RECEIPTS")
dao_update_delivery_receipts(batch, True) dao_update_delivery_receipts(batch, True)
print("DEIVERY RECEIPTS UPDATED")
failed_receipts = list(failed_receipts) failed_receipts = list(failed_receipts)
for i in range(0, len(failed_receipts), batch_size): for i in range(0, len(failed_receipts), batch_size):
print("UDPATING FAILED RECEIPTS")
batch = failed_receipts[i : i + batch_size] batch = failed_receipts[i : i + batch_size]
dao_update_delivery_receipts(batch, False) dao_update_delivery_receipts(batch, False)
print("FAILED RECEITPS UPDATED")
except Exception as ex: except Exception as ex:
print(f"EXCEPTION {ex}")
retry_count = self.request.retries retry_count = self.request.retries
wait_time = 3600 * 2**retry_count wait_time = 3600 * 2**retry_count
try: try:

7
tests/app/celery/test_scheduled_tasks.py
View File

@@ -607,10 +607,11 @@ def test_process_delivery_receipts_success(mocker):
dao_update_mock = mocker.patch( dao_update_mock = mocker.patch(
"app.celery.scheduled_tasks.dao_update_delivery_receipts" "app.celery.scheduled_tasks.dao_update_delivery_receipts"
) )
cloudwatch_mock = mocker.patch( cloudwatch_mock = mocker.patch("app.celery.scheduled_tasks.AwsCloudwatchClient")
"app.celery.scheduled_tasks.AwsCloudwatchClient" cloudwatch_mock.return_value.check_delivery_receipts.return_value = (
range(2000),
range(500),
) )
cloudwatch_mock.return_value.check_delivery_receipts.return_value = (range(2000), range(500))
current_app_mock = mocker.patch("app.celery.scheduled_tasks.current_app") current_app_mock = mocker.patch("app.celery.scheduled_tasks.current_app")
current_app_mock.return_value = MagicMock() current_app_mock.return_value = MagicMock()
processor = MagicMock() processor = MagicMock()