darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-01-30 06:21:50 -05:00
Code Issues Packages Projects Releases Wiki Activity

Adding test for fetch_biling_data_for_day.

Browse Source
This commit is contained in:
Rebecca Law
2018-04-25 14:24:47 +01:00
parent 16ef133aa5
commit d00614205e
3 changed files with 133 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/celery/reporting_tasks.py
View File

@@ -1,14 +1,12 @@
from datetime import datetime, timedelta, time
from datetime import datetime, timedelta
from flask import current_app
from notifications_utils.statsd_decorators import statsd
from app import notify_celery
from app.dao.fact_billing_dao import (
fetch_billing_data,
fetch_billing_data_for_day,
update_fact_billing
)
from app.utils import convert_bst_to_utc
@notify_celery.task(name="create-nightly-billing")
@@ -21,19 +19,8 @@ def create_nightly_billing(day_start=None):
for i in range(0, 3):
process_day = day_start - timedelta(days=i)
ds = convert_bst_to_utc(datetime.combine(process_day, time.min))
de = convert_bst_to_utc(datetime.combine(process_day + timedelta(days=1), time.min))
transit_data = fetch_billing_data(start_date=ds, end_date=de)
updated_records = 0
inserted_records = 0
transit_data = fetch_billing_data_for_day(process_day=process_day)
for data in transit_data:
inserted_records, updated_records = update_fact_billing(data,
inserted_records,
process_day,
updated_records)
current_app.logger.info('ft_billing {} to {}: {} rows updated, {} rows inserted'
.format(ds, de, updated_records, inserted_records))
update_fact_billing(data, process_day)