def test_fetch_count_of_complaints(sample_email_notification):
create_complaint(
service=sample_email_notification.service,
notification=sample_email_notification,
created_at=datetime(2018, 6, 6, 22, 00, 00),
)
create_complaint(
service=sample_email_notification.service,
notification=sample_email_notification,
created_at=datetime(2018, 6, 6, 23, 00, 00),
)
create_complaint(
service=sample_email_notification.service,
notification=sample_email_notification,
created_at=datetime(2018, 6, 7, 00, 00, 00),
)
create_complaint(
service=sample_email_notification.service,
notification=sample_email_notification,
created_at=datetime(2018, 6, 7, 13, 00, 00),
)
create_complaint(
service=sample_email_notification.service,
notification=sample_email_notification,
created_at=datetime(2018, 6, 7, 23),
)
count_of_complaints = fetch_count_of_complaints(start_date=datetime(2018, 6, 7), end_date=datetime(2018, 6, 7))
assert count_of_complaints == 2
def get_complaint_count():
if request.args:
validate(request.args, complaint_count_request)
# If start and end date are not set, we are expecting today's stats.
today = str(datetime.utcnow().date())
start_date = datetime.strptime(request.args.get('start_date', today),
'%Y-%m-%d').date()
end_date = datetime.strptime(request.args.get('end_date', today),
'%Y-%m-%d').date()
count_of_complaints = fetch_count_of_complaints(start_date=start_date,
end_date=end_date)
return jsonify(count_of_complaints), 200
def test_fetch_count_of_complaints_returns_zero(notify_db):
count_of_complaints = fetch_count_of_complaints(
start_date=datetime(2018, 6, 7), end_date=datetime(2018, 6, 7))
assert count_of_complaints == 0
